private static ResourceManagementClient GetResourceManagementClient(MockContext context)
{
return(context.GetServiceClient <ResourceManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
private StorageMgmtNS.StorageManagementClient GetStorageManagementClient(MockContext context)
{
return(context.GetServiceClient <StorageMgmtNS.StorageManagementClient>(
TestEnvironmentFactory.GetTestEnvironment()));
}
private static NetworkManagementClient GetNetworkManagementClientRestClient(MockContext context)
{
return(context.GetServiceClient <NetworkManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
private static SiteRecoveryManagementClient GetSiteRecoveryManagementClient(MockContext context)
{
return(context.GetServiceClient <SiteRecoveryManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
protected NetworkManagementClient GetNetworkClient(MockContext context)
{
return(context.GetServiceClient <NetworkManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
private static IotCentralClient GetIotCentralClient(MockContext context)
{
return(context.GetServiceClient <IotCentralClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
private SubscriptionClient GetSubscriptionClient(MockContext context)
{
return(context.GetServiceClient <SubscriptionClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
protected GraphRbacManagementClient GetGraphClient(MockContext context)
{
GraphRbacManagementClient graphClient = context.GetServiceClient <GraphRbacManagementClient>(TestEnvironmentFactory.GetTestEnvironment());
graphClient.BaseUri = TestEnvironmentFactory.GetTestEnvironment().Endpoints.GraphUri;
graphClient.TenantID = TestEnvironmentFactory.GetTestEnvironment().Tenant;
return(graphClient);
}
private static SecurityCenterClient GetSecurityCenterClient(MockContext context)
{
return(context.GetServiceClient <SecurityCenterClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
private static PowerBIEmbeddedManagementClient GetPowerBIClient(MockContext context)
{
return(context.GetServiceClient <PowerBIEmbeddedManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
protected AksTestRunner(ITestOutputHelper output)
{
TestRunner = TestManager.CreateInstance(output)
.WithNewPsScriptFilename($"{GetType().Name}.ps1")
.WithProjectSubfolderForTests("ScenarioTests")
.WithCommonPsScripts(new[]
{
@"Common.ps1",
@"../AzureRM.Resources.ps1"
})
.WithNewRmModules(helper => new[]
{
helper.RMProfileModule,
helper.GetRMModulePath("Az.Aks.psd1")
})
.WithNewRecordMatcherArguments(
userAgentsToIgnore: new Dictionary <string, string>
{
{ "Microsoft.Azure.Management.Resources.ResourceManagementClient", "2017-05-10" },
{ "Microsoft.Azure.Management.ResourceManager.ResourceManagementClient", "2017-05-10" }
},
resourceProviders: new Dictionary <string, string>
{
{ "Microsoft.Resources", null },
{ "Microsoft.Features", null },
{ "Microsoft.Authorization", null }
}
).WithMockContextAction(() =>
{
if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Playback)
{
AzureSession.Instance.DataStore = new MemoryDataStore();
var home = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile);
var dir = Path.GetDirectoryName(new Uri(Assembly.GetExecutingAssembly().CodeBase).AbsolutePath);
var subscription = HttpMockServer.Variables["SubscriptionId"];
AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".ssh", "id_rsa.pub"), File.ReadAllText(dir + "/Fixtures/id_rsa.pub"));
var jsonOutput = @"{""" + subscription + @""":{ ""service_principal"":""foo"",""client_secret"":""bar""}}";
AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".azure", "acsServicePrincipal.json"), jsonOutput);
}
else if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Record)
{
AzureSession.Instance.DataStore = new MemoryDataStore();
var home = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile);
var dir = Path.GetDirectoryName(new Uri(Assembly.GetExecutingAssembly().CodeBase).AbsolutePath);
var subscription = HttpMockServer.Variables["SubscriptionId"];
var currentEnvironment = TestEnvironmentFactory.GetTestEnvironment();
string spn = null;
string spnSecret = null;
if (currentEnvironment.ConnectionString.KeyValuePairs.ContainsKey("ServicePrincipal"))
{
spn = currentEnvironment.ConnectionString.KeyValuePairs["ServicePrincipal"];
}
if (currentEnvironment.ConnectionString.KeyValuePairs.ContainsKey("ServicePrincipalSecret"))
{
spnSecret = currentEnvironment.ConnectionString.KeyValuePairs["ServicePrincipalSecret"];
}
AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".ssh", "id_rsa.pub"), File.ReadAllText(dir + "/Fixtures/id_rsa.pub"));
var jsonOutput = @"{""" + subscription + @""":{ ""service_principal"":""" + spn + @""",""client_secret"":""" + spnSecret + @"""}}";
AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".azure", "acsServicePrincipal.json"), jsonOutput);
}
}
)
.Build();
}
private static WebSiteManagementClient GetWebsitesManagementClient(MockContext context)
{
return(context.GetServiceClient <WebSiteManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
private static AuthorizationManagementClient GetAuthorizationManagementClient(MockContext context)
{
return(context.GetServiceClient <AuthorizationManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
public static StorageManagementClient GetStorageManagementClient(TestEnvironmentFactory factory, RecordedDelegatingHandler handler)
{
handler.IsPassThrough = true;
return TestBase.GetServiceClient<StorageManagementClient>(factory).WithHandler(handler);
}
private HyakRmNS.ResourceManagementClient GetHyakRmClient(MockContext context)
{
return(context.GetServiceClient <HyakRmNS.ResourceManagementClient>(
TestEnvironmentFactory.GetTestEnvironment()));
}
private static GraphRbacManagementClient GetGraphRBACManagementClient(MockContext context)
{
return(context.GetServiceClient <GraphRbacManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
private RecoveryServicesBackupClient GetRsBackupClient(MockContext context)
{
return(context.GetServiceClient <RecoveryServicesBackupClient>(
TestEnvironmentFactory.GetTestEnvironment()));
}
protected KeyVaultManagementClient GetKeyVaultClient(MockContext context)
{
return(context.GetServiceClient <KeyVaultManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
private static HDInsightManagementClient GetHDInsightManagementClient(MockContext context)
{
return(context.GetServiceClient <HDInsightManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
protected static CommonStorage.StorageManagementClient GetStorageManagementClient(MockContext context)
{
return(context.GetServiceClient <CommonStorage.StorageManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
public void ResetPreCreatedVault()
{
if (mode == HttpRecorderMode.Record)
{
using (MockContext context = MockContext.Start(TestUtilities.GetCallingClass(), TestUtilities.GetCurrentMethodName(1)))
{
var testFactory = new LegacyTest.CSMTestEnvironmentFactory();
var testEnv = testFactory.GetTestEnvironment();
var resourcesClient = LegacyTest.TestBase.GetServiceClient <ResourceManagementClient>(testFactory);
var mgmtClient = context.GetServiceClient <KeyVaultManagementClient>(TestEnvironmentFactory.GetTestEnvironment());
var tenantId = Guid.Parse(testEnv.AuthorizationContext.TenantId);
var policies = new AccessPolicyEntry[] { };
mgmtClient.Vaults.CreateOrUpdate(
resourceGroupName: resourceGroupName,
vaultName: preCreatedVault,
parameters: new VaultCreateOrUpdateParameters
{
Location = location,
Tags = new Dictionary <string, string> {
{ tagName, tagValue }
},
Properties = new VaultProperties
{
EnabledForDeployment = false,
Sku = new Sku {
Name = SkuName.Premium
},
TenantId = tenantId,
VaultUri = "",
AccessPolicies = policies
}
}
);
}
}
}
public void RunPowerShellTest(XunitTracingInterceptor logger, params string[] scripts)
{
var sf = new StackTrace().GetFrame(1);
var callingClassType = sf.GetMethod().ReflectedType?.ToString();
var mockName = sf.GetMethod().Name;
_helper.TracingInterceptor = logger;
var d = new Dictionary <string, string>
{
{ "Microsoft.Resources", null },
{ "Microsoft.Features", null },
{ "Microsoft.Authorization", null }
};
var providersToIgnore = new Dictionary <string, string>
{
{ "Microsoft.Azure.Management.Resources.ResourceManagementClient", "2017-05-10" },
{ "Microsoft.Azure.Management.ResourceManager.ResourceManagementClient", "2017-05-10" }
};
HttpMockServer.Matcher = new PermissiveRecordMatcherWithApiExclusion(false, d, providersToIgnore);
HttpMockServer.RecordsDirectory = Path.Combine(AppDomain.CurrentDomain.BaseDirectory, "SessionRecords");
using (var context = MockContext.Start(callingClassType, mockName))
{
SetupManagementClients(context);
var callingClassName = callingClassType?.Split(new[] { "." }, StringSplitOptions.RemoveEmptyEntries).Last();
_helper.SetupEnvironment(AzureModule.AzureResourceManager);
_helper.SetupModules(AzureModule.AzureResourceManager,
_helper.RMProfileModule,
_helper.GetRMModulePath(@"AzureRM.Aks.psd1"),
"ScenarioTests\\Common.ps1",
"ScenarioTests\\" + callingClassName + ".ps1",
"AzureRM.Resources.ps1");
if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Playback)
{
AzureSession.Instance.DataStore = new MemoryDataStore();
var home = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile);
var dir = Path.GetDirectoryName(new Uri(Assembly.GetExecutingAssembly().CodeBase).AbsolutePath);
var subscription = HttpMockServer.Variables["SubscriptionId"];
AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".ssh", "id_rsa.pub"), File.ReadAllText(dir + "/Fixtures/id_rsa.pub"));
var jsonOutput = @"{""" + subscription + @""":{ ""service_principal"":""foo"",""client_secret"":""bar""}}";
AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".azure", "acsServicePrincipal.json"), jsonOutput);
}
else if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Record)
{
AzureSession.Instance.DataStore = new MemoryDataStore();
var home = Environment.GetFolderPath(Environment.SpecialFolder.UserProfile);
var dir = Path.GetDirectoryName(new Uri(Assembly.GetExecutingAssembly().CodeBase).AbsolutePath);
var subscription = HttpMockServer.Variables["SubscriptionId"];
var currentEnvironment = TestEnvironmentFactory.GetTestEnvironment();
string spn = null;
string spnSecret = null;
if (currentEnvironment.ConnectionString.KeyValuePairs.ContainsKey("ServicePrincipal"))
{
spn = currentEnvironment.ConnectionString.KeyValuePairs["ServicePrincipal"];
}
if (currentEnvironment.ConnectionString.KeyValuePairs.ContainsKey("ServicePrincipalSecret"))
{
spnSecret = currentEnvironment.ConnectionString.KeyValuePairs["ServicePrincipalSecret"];
}
AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".ssh", "id_rsa.pub"), File.ReadAllText(dir + "/Fixtures/id_rsa.pub"));
var jsonOutput = @"{""" + subscription + @""":{ ""service_principal"":""" + spn + @""",""client_secret"":""" + spnSecret + @"""}}";
AzureSession.Instance.DataStore.WriteFile(Path.Combine(home, ".azure", "acsServicePrincipal.json"), jsonOutput);
}
_helper.RunPowerShellTest(scripts);
}
}
public void Initialize(string className)
{
if (initialized)
{
return;
}
if (mode == HttpRecorderMode.Record)
{
using (MockContext context = MockContext.Start(TestUtilities.GetCallingClass(), TestUtilities.GetCurrentMethodName(1)))
{
var testFactory = new LegacyTest.CSMTestEnvironmentFactory();
var testEnv = testFactory.GetTestEnvironment();
var resourcesClient = LegacyTest.TestBase.GetServiceClient <ResourceManagementClient>(testFactory);
var mgmtClient = context.GetServiceClient <KeyVaultManagementClient>(TestEnvironmentFactory.GetTestEnvironment());
var tenantId = testEnv.AuthorizationContext.TenantId;
//Figure out which locations are available for Key Vault
location = GetKeyVaultLocation(resourcesClient);
//Create a resource group in that location
preCreatedVault = TestUtilities.GenerateName("pshtestvault");
resourceGroupName = TestUtilities.GenerateName("pshtestrg");
resourcesClient.ResourceGroups.CreateOrUpdate(resourceGroupName, new ResourceGroup {
Location = location
});
var createResponse = CreateVault(mgmtClient, location, tenantId);
}
}
initialized = true;
}
private static ComputeManagementClient GetComputeManagementClientRestClient(MockContext context)
{
return(context.GetServiceClient <ComputeManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
protected SqlManagementClient GetSqlClient(MockContext context)
{
return(context.GetServiceClient <SqlManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
public void SetupAzureEnvironmentFromEnvironmentVariables(AzureModule mode)
{
TestEnvironment currentEnvironment = null;
if (mode == AzureModule.AzureResourceManager)
{
#if !NETSTANDARD
currentEnvironment = new CSMTestEnvironmentFactory().GetTestEnvironment();
#else
currentEnvironment = TestEnvironmentFactory.GetTestEnvironment();
#endif
}
else
{
#if !NETSTANDARD
currentEnvironment = new RDFETestEnvironmentFactory().GetTestEnvironment();
#else
throw new NotSupportedException("RDFE environment is not supported in .Net Core");
#endif
}
if (currentEnvironment.UserName == null)
{
currentEnvironment.UserName = "******";
}
SetAuthenticationFactory(mode, currentEnvironment);
AzureEnvironment environment = new AzureEnvironment {
Name = testEnvironmentName
};
Debug.Assert(currentEnvironment != null);
environment.ActiveDirectoryAuthority = currentEnvironment.Endpoints.AADAuthUri.AbsoluteUri;
environment.GalleryUrl = currentEnvironment.Endpoints.GalleryUri.AbsoluteUri;
environment.ServiceManagementUrl = currentEnvironment.BaseUri.AbsoluteUri;
environment.ResourceManagerUrl = currentEnvironment.Endpoints.ResourceManagementUri.AbsoluteUri;
environment.GraphUrl = currentEnvironment.Endpoints.GraphUri.AbsoluteUri;
environment.AzureDataLakeAnalyticsCatalogAndJobEndpointSuffix = currentEnvironment.Endpoints.DataLakeAnalyticsJobAndCatalogServiceUri.OriginalString.Replace("https://", ""); // because it is just a sufix
environment.AzureDataLakeStoreFileSystemEndpointSuffix = currentEnvironment.Endpoints.DataLakeStoreServiceUri.OriginalString.Replace("https://", ""); // because it is just a sufix
environment.StorageEndpointSuffix = AzureEnvironmentConstants.AzureStorageEndpointSuffix;
#if !NETSTANDARD
if (!ProfileClient.Profile.EnvironmentTable.ContainsKey(testEnvironmentName))
{
ProfileClient.AddOrSetEnvironment(environment);
}
#endif
if (!AzureRmProfileProvider.Instance.GetProfile <AzureRmProfile>().EnvironmentTable.ContainsKey(testEnvironmentName))
{
AzureRmProfileProvider.Instance.GetProfile <AzureRmProfile>().EnvironmentTable[testEnvironmentName] = environment;
}
if (currentEnvironment.SubscriptionId != null)
{
testSubscription = new AzureSubscription()
{
Id = currentEnvironment.SubscriptionId,
Name = testSubscriptionName,
};
testSubscription.SetEnvironment(testEnvironmentName);
testSubscription.SetAccount(currentEnvironment.UserName);
testSubscription.SetDefault();
testSubscription.SetStorageAccount(Environment.GetEnvironmentVariable("AZURE_STORAGE_ACCOUNT"));
testAccount = new AzureAccount()
{
Id = currentEnvironment.UserName,
Type = AzureAccount.AccountType.User,
};
testAccount.SetSubscriptions(currentEnvironment.SubscriptionId);
#if !NETSTANDARD
ProfileClient.Profile.SubscriptionTable[testSubscription.GetId()] = testSubscription;
ProfileClient.Profile.AccountTable[testAccount.Id] = testAccount;
ProfileClient.SetSubscriptionAsDefault(testSubscription.Name, testSubscription.GetAccount());
#endif
var testTenant = new AzureTenant()
{
Id = Guid.NewGuid().ToString()
};
if (!string.IsNullOrEmpty(currentEnvironment.Tenant))
{
Guid tenant;
if (Guid.TryParse(currentEnvironment.Tenant, out tenant))
{
testTenant.Id = currentEnvironment.Tenant;
}
}
AzureRmProfileProvider.Instance.Profile.DefaultContext = new AzureContext(testSubscription, testAccount, environment, testTenant);
}
}
protected CommonMonitor.IMonitorManagementClient GetCommonMonitorManagementClient(MockContext context)
{
return(context.GetServiceClient <CommonMonitor.MonitorManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
private static SearchManagementClient GetAzureSearchManagementClient(MockContext context)
{
return(context.GetServiceClient <SearchManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
protected IEventHubManagementClient GetEventHubManagementClient(MockContext context)
{
return(context.GetServiceClient <EventHubManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
public static NetworkResourceProviderClient GetNetworkResourceProviderClient(TestEnvironmentFactory factory, RecordedDelegatingHandler handler)
{
handler.IsPassThrough = true;
return TestBase.GetServiceClient<NetworkResourceProviderClient>(factory).WithHandler(handler);
}
protected IOperationalInsightsManagementClient GetOperationalInsightsManagementClient(MockContext context)
{
return(context.GetServiceClient <OperationalInsightsManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
public void RunPsTestWorkflow(
Func<string[]> scriptBuilder,
Action<TestEnvironmentFactory> initialize,
Action cleanup,
string callingClassType,
string mockName)
{
HttpMockServer.Matcher = new PermissiveRecordMatcher();
using (UndoContext context = UndoContext.Current)
{
context.Start(callingClassType, mockName);
this.testFactory = GetFactory();
if(initialize != null)
{
initialize(this.testFactory);
}
SetupManagementClients();
helper.SetupEnvironment(this.Module);
var callingClassName = callingClassType
.Split(new[] { "." }, StringSplitOptions.RemoveEmptyEntries)
.Last();
helper.SetupModules(
this.Module,
"Profile\\" + callingClassName + ".ps1");
try
{
if (scriptBuilder != null)
{
var psScripts = scriptBuilder();
if (psScripts != null)
{
helper.RunPowerShellTest(psScripts);
}
}
}
finally
{
if(cleanup !=null)
{
cleanup();
}
}
}
}
private static HealthcareApisManagementClient GetHealthcareApisManagementClient(MockContext context)
{
return(context.GetServiceClient <HealthcareApisManagementClient>(TestEnvironmentFactory.GetTestEnvironment()));
}
