protected override void SetupManagementClients(RestTestFramework.MockContext context)
{
var sqlCSMClient = GetSqlClient();
var storageClient = GetStorageClient();
var storageV2Client = GetStorageV2Client();
var resourcesClient = GetResourcesClient();
var authorizationClient = GetAuthorizationManagementClient();
helper.SetupSomeOfManagementClients(sqlCSMClient, storageClient, storageV2Client, resourcesClient, authorizationClient);
}
protected virtual void SetupManagementClients(RestTestFramework.MockContext context)
{
var sqlCSMClient = GetSqlClient();
var storageClient = GetStorageClient();
//TODO, Remove the MockDeploymentFactory call when the test is re-recorded
var resourcesClient = MockDeploymentClientFactory.GetResourceClient(GetResourcesClient());
var authorizationClient = GetAuthorizationManagementClient();
helper.SetupSomeOfManagementClients(sqlCSMClient, storageClient, resourcesClient, authorizationClient);
}
private CognitiveServicesManagementClient GetCognitiveServicesManagementClient(RestTestFramework.MockContext context)
{
return context.GetServiceClient<CognitiveServicesManagementClient>(RestTestFramework.TestEnvironmentFactory.GetTestEnvironment());
}
private void SetupManagementClients(RestTestFramework.MockContext context)
{
ResourceManagementClient = GetResourceManagementClient();
SubscriptionClient = GetSubscriptionClient();
CognitiveServicesClient = GetCognitiveServicesManagementClient(context);
GalleryClient = GetGalleryClient();
AuthorizationManagementClient = GetAuthorizationManagementClient();
helper.SetupManagementClients(
ResourceManagementClient,
SubscriptionClient,
CognitiveServicesClient,
GalleryClient,
AuthorizationManagementClient);
}
private NetworkManagementClient GetNetworkManagementClient(RestTestFramework.MockContext context)
{
return context.GetServiceClient<NetworkManagementClient>(RestTestFramework.TestEnvironmentFactory.GetTestEnvironment());
}
private void SetupManagementClients(RestTestFramework.MockContext context)
{
this.ResourceManagementClient = this.GetResourceManagementClient();
this.SubscriptionClient = this.GetSubscriptionClient();
this.GalleryClient = this.GetGalleryClient();
this.NetworkManagementClient = this.GetNetworkManagementClient(context);
this.AuthorizationManagementClient = this.GetAuthorizationManagementClient();
helper.SetupManagementClients(
ResourceManagementClient,
SubscriptionClient,
GalleryClient,
AuthorizationManagementClient,
this.NetworkManagementClient);
}
private ComputeManagementClient GetComputeManagementClient(RestTestFramework.MockContext context)
{
return testViaCsm
? context.GetServiceClient<ComputeManagementClient>(RestTestFramework.TestEnvironmentFactory.GetTestEnvironment())
: TestBase.GetServiceClient<ComputeManagementClient>(new RDFETestEnvironmentFactory());
}
private void SetupManagementClients(RestTestFramework.MockContext context)
{
ResourceManagementClient = GetResourceManagementClient();
SubscriptionClient = GetSubscriptionClient();
StorageClient = GetStorageManagementClient();
GalleryClient = GetGalleryClient();
//var eventsClient = GetEventsClient();
NetworkManagementClient = this.GetNetworkManagementClientClient(context);
ComputeManagementClient = GetComputeManagementClient(context);
AuthorizationManagementClient = GetAuthorizationManagementClient();
// GraphClient = GetGraphClient();
helper.SetupManagementClients(
ResourceManagementClient,
SubscriptionClient,
StorageClient,
GalleryClient,
//eventsClient,
NetworkManagementClient,
ComputeManagementClient,
AuthorizationManagementClient);
// GraphClient);
}
protected GraphRbacManagementClient GetGraphClient(RestTestFramework.MockContext context)
{
var environment = RestTestFramework.TestEnvironmentFactory.GetTestEnvironment();
string tenantId = null;
if (HttpMockServer.Mode == HttpRecorderMode.Record)
{
tenantId = environment.Tenant;
UserDomain = environment.UserName.Split(new[] { "@" }, StringSplitOptions.RemoveEmptyEntries).Last();
HttpMockServer.Variables[TenantIdKey] = tenantId;
HttpMockServer.Variables[DomainKey] = UserDomain;
}
else if (HttpMockServer.Mode == HttpRecorderMode.Playback)
{
if (HttpMockServer.Variables.ContainsKey(TenantIdKey))
{
tenantId = HttpMockServer.Variables[TenantIdKey];
AzureRmProfileProvider.Instance.Profile.Context.Tenant.Id = new Guid(tenantId);
}
if (HttpMockServer.Variables.ContainsKey(DomainKey))
{
UserDomain = HttpMockServer.Variables[DomainKey];
AzureRmProfileProvider.Instance.Profile.Context.Tenant.Domain = UserDomain;
}
}
var client = context.GetGraphServiceClient<GraphRbacManagementClient>(environment);
client.TenantID = tenantId;
return client;
}
