private void SetupManagementClients(MockContext context)
{
ResourceManagementClient = GetResourceManagementClient(context);
BatchManagementClient = GetBatchManagementClient(context);
NetworkManagementClient = GetNetworkManagementClient(context);
AzureRestClient = context.GetServiceClient <AzureRestClient>(TestEnvironmentFactory.GetTestEnvironment());
_helper.SetupManagementClients(ResourceManagementClient, BatchManagementClient, NetworkManagementClient, AzureRestClient);
}
private async Task SendIsForcedTerminateUpdate(AzureRestClient client, bool isForcedTerminate)
{
var ns = XmlNamespaces.WindowsAzure;
var content = new XElement(ns + "ServiceResource",
new XElement(ns + "IsForcedTerminate", isForcedTerminate)
);
var response = await client.PutAsync(content);
await Database.DatabaseServer.Subscription.WaitForOperationCompletionAsync(response);
}
protected void SetupManagementClients(MockContext context)
{
ResourceManagementClient = GetResourceManagementClient(context);
CosmosDBManagementClient = GetCosmosDBManagementClient(context);
NetworkManagementClient = GetNetworkManagementClient(context);
AzureRestClient = GetAzureRestClient(context);
_helper.SetupManagementClients(
ResourceManagementClient,
CosmosDBManagementClient,
NetworkManagementClient,
AzureRestClient);
}
private async Task SendDelete(AzureRestClient client)
{
var response = await client.DeleteAsync();
await Database.DatabaseServer.Subscription.WaitForOperationCompletionAsync(response);
}
