protected void SetupManagementClients(RestTestFramework.MockContext context)
{
this.ResourceManagementClient = this.GetResourceManagementClient(context);
this.SubscriptionClient = this.GetSubscriptionClient();
this.GalleryClient = this.GetGalleryClient();
this.AuthorizationManagementClient = this.GetAuthorizationManagementClient();
this.TrafficManagerManagementClient = this.GetFeatureClient(context);
this.helper.SetupManagementClients(
this.ResourceManagementClient,
this.SubscriptionClient,
this.GalleryClient,
this.AuthorizationManagementClient,
this.TrafficManagerManagementClient);
}
protected ResourceManagementClient GetResourceManagementClient(RestTestFramework.MockContext context)
{
return(context.GetServiceClient <ResourceManagementClient>(RestTestFramework.TestEnvironmentFactory.GetTestEnvironment()));
}
private TrafficManagerManagementClient GetFeatureClient(RestTestFramework.MockContext context)
{
return(context.GetServiceClient <TrafficManagerManagementClient>(RestTestFramework.TestEnvironmentFactory.GetTestEnvironment()));
}
public void RunPsTestWorkflow(
Func <string[]> scriptBuilder,
Action <CSMTestEnvironmentFactory> initialize,
Action cleanup,
string callingClassType,
string mockName)
{
Dictionary <string, string> d = new Dictionary <string, string>();
d.Add("Microsoft.Resources", null);
d.Add("Microsoft.Features", null);
d.Add("Microsoft.Authorization", null);
var providersToIgnore = new Dictionary <string, string>();
providersToIgnore.Add("Microsoft.Azure.Management.Resources.ResourceManagementClient", "2016-02-01");
HttpMockServer.Matcher = new PermissiveRecordMatcherWithApiExclusion(true, d, providersToIgnore);
HttpMockServer.RecordsDirectory = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "SessionRecords");
using (RestTestFramework.MockContext context = RestTestFramework.MockContext.Start(callingClassType, mockName))
{
this.csmTestFactory = new CSMTestEnvironmentFactory();
if (initialize != null)
{
initialize(this.csmTestFactory);
}
this.helper.SetupEnvironment(AzureModule.AzureResourceManager);
this.SetupManagementClients(context);
string callingClassName = callingClassType
.Split(new[] { "." }, StringSplitOptions.RemoveEmptyEntries)
.Last();
this.helper.SetupModules(AzureModule.AzureResourceManager,
"ScenarioTests\\Common.ps1",
"ScenarioTests\\" + callingClassName + ".ps1",
helper.RMProfileModule,
helper.RMResourceModule,
helper.GetRMModulePath(@"AzureRM.TrafficManager.psd1"),
"AzureRM.Resources.ps1");
try
{
if (scriptBuilder != null)
{
string[] psScripts = scriptBuilder();
if (psScripts != null)
{
this.helper.RunPowerShellTest(psScripts);
}
}
}
finally
{
if (cleanup != null)
{
cleanup();
}
}
}
}
