public void SetupEnvironment(AzureModule mode)
{
SetupAzureEnvironmentFromEnvironmentVariables(mode);
#if !NETSTANDARD
ProfileClient.Profile.Save();
#endif
}
private void SetAuthenticationFactory(AzureModule mode, TestEnvironment environment)
{
string jwtToken = null;
X509Certificate2 certificate = null;
if (environment.Credentials is TokenCloudCredentials)
{
jwtToken = ((TokenCloudCredentials)environment.Credentials).Token;
}
if (environment.Credentials is CertificateCloudCredentials)
{
certificate = ((CertificateCloudCredentials)environment.Credentials).ManagementCertificate;
}
if (jwtToken != null)
{
AzureSession.AuthenticationFactory = new MockTokenAuthenticationFactory(environment.UserName,
jwtToken);
}
else if (certificate != null)
{
AzureSession.AuthenticationFactory = new MockCertificateAuthenticationFactory(environment.UserName,
certificate);
}
}
public void SetupModules(AzureModule mode, params string[] testModules)
{
modules = new List<string>();
switch (mode)
{
case AzureModule.AzureProfile:
modules.Add(@"ServiceManagement\Azure\Azure.psd1");
modules.Add(@"ResourceManager\AzureResourceManager\AzureResourceManager.psd1");
break;
case AzureModule.AzureServiceManagement:
modules.Add(@"ServiceManagement\Azure\Azure.psd1");
break;
case AzureModule.AzureResourceManager:
modules.Add(@"ResourceManager\AzureResourceManager\AzureResourceManager.psd1");
break;
default:
throw new ArgumentException("Unknown command type for testing");
}
modules.Add("Assert.ps1");
modules.Add("Common.ps1");
modules.AddRange(testModules);
}
public void SetupModules(AzureModule mode, params string[] testModules)
{
modules = new List<string>();
switch (mode)
{
case AzureModule.AzureProfile:
modules.Add(@"ServiceManagement\Azure\Azure.psd1");
modules.Add(@"ResourceManager\AzureResourceManager\AzureResourceManager.psd1");
break;
case AzureModule.AzureServiceManagement:
modules.Add(@"ServiceManagement\Azure\Azure.psd1");
break;
case AzureModule.AzureResourceManager:
modules.Add(@"ResourceManager\AzureResourceManager\AzureResourceManager.psd1");
break;
default:
throw new ArgumentException("Unknown command type for testing");
}
modules.Add("Assert.ps1");
modules.Add("Common.ps1");
modules.AddRange(testModules);
}
public void SetupModulesFromCommon(AzureModule mode, params string[] modules)
{
this.modules = new List <string>();
if (mode == AzureModule.AzureProfile)
{
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ServiceManagement\Azure\Azure.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Profile\AzureRM.Profile.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Resources\AzureRM.Resources.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Resources\AzureRM.Tags.psd1"));
}
else if (mode == AzureModule.AzureServiceManagement)
{
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ServiceManagement\Azure\Azure.psd1"));
}
else if (mode == AzureModule.AzureResourceManager)
{
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Profile\AzureRM.Profile.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Resources\AzureRM.Resources.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Resources\AzureRM.Tags.psd1"));
}
else
{
throw new ArgumentException("Unknown command type for testing");
}
this.modules.Add("Assert.ps1");
this.modules.Add("Common.ps1");
this.modules.AddRange(modules);
}
public void SetupModules(AzureModule mode, params string[] testModules)
{
modules = new List <string>();
modules.Add(@"ServiceManagement\Azure\Azure.psd1");
modules.Add("Assert.ps1");
modules.Add("Common.ps1");
modules.AddRange(testModules);
}
public override void ExecuteCmdlet()
{
AzureModule moduleToImport = (AzureModule)Enum.Parse(typeof(AzureModule), Name, false);
AzureModule moduleToRemove = moduleToImport == AzureModule.AzureResourceManager ? AzureModule.AzureServiceManagement : AzureModule.AzureResourceManager;
RemoveAzureModule(FileUtilities.GetModuleName(moduleToRemove), FileUtilities.GetPSModulePathForModule(moduleToRemove));
ImportAzureModule(FileUtilities.GetModuleName(moduleToImport), FileUtilities.GetPSModulePathForModule(moduleToImport));
}
public WindowsAzurePowerShellTokenTest(AzureModule mode, params string[] modules)
: base(mode, modules)
{
_moduleMode = mode;
if (Environment.GetEnvironmentVariable(outputDirKey) != null) {
HttpMockServer.RecordsDirectory = Environment.GetEnvironmentVariable(outputDirKey);
}
}
public WindowsAzurePowerShellTokenTest(AzureModule mode, params string[] modules)
: base(mode, modules)
{
_moduleMode = mode;
if (Environment.GetEnvironmentVariable(outputDirKey) != null)
{
HttpMockServer.RecordsDirectory = Environment.GetEnvironmentVariable(outputDirKey);
}
}
public PowerShellTest(AzureModule commandMode, params string[] modules)
{
this.modules = new List <string>();
this.modules.Add(FileUtilities.GetContentFilePath(@"ServiceManagement\Azure\Azure.psd1"));
this.modules.Add("Assert.ps1");
this.modules.Add("Common.ps1");
this.modules.AddRange(modules);
PSTestTracingInterceptor.AddToContext();
}
private void SetAuthenticationFactory(AzureModule mode, TestEnvironment rdfeEnvironment, TestEnvironment csmEnvironment)
{
if (mode == AzureModule.AzureServiceManagement)
{
SetAuthenticationFactory(rdfeEnvironment);
}
else
{
SetAuthenticationFactory(csmEnvironment);
}
}
private void SetAuthenticationFactory(AzureModule mode, TestEnvironment environment)
{
if (environment.TokenInfo.ContainsKey(TokenAudience.Management))
{
var httpMessage = new HttpRequestMessage();
environment.TokenInfo[TokenAudience.Management]
.ProcessHttpRequestAsync(httpMessage, CancellationToken.None)
.ConfigureAwait(false)
.GetAwaiter()
.GetResult();
AzureSession.Instance.AuthenticationFactory = new MockTokenAuthenticationFactory(
environment.UserName,
httpMessage.Headers.Authorization.Parameter);
}
}
public static string GetModuleName(AzureModule module)
{
switch (module)
{
case AzureModule.AzureServiceManagement:
return("Azure");
case AzureModule.AzureResourceManager:
return("AzureResourceManager");
case AzureModule.AzureProfile:
return("AzureProfile");
default:
throw new ArgumentOutOfRangeException(module.ToString());
}
}
public void SetupModules(AzureModule mode, params string[] modules)
{
this.modules = new List <string>();
if (mode == AzureModule.AzureProfile)
{
this.modules.Add(Path.Combine(PackageDirectory, @"ServiceManagement\Azure\Azure.psd1"));
this.modules.Add(Path.Combine(PackageDirectory, @"ResourceManager\AzureResourceManager\AzureResourceManager.psd1"));
}
else if (mode == AzureModule.AzureServiceManagement)
{
this.modules.Add(Path.Combine(PackageDirectory, @"ServiceManagement\Azure\Azure.psd1"));
}
this.modules.Add("Assert.ps1");
this.modules.Add("Common.ps1");
this.modules.AddRange(modules);
}
public void SetupModules(AzureModule mode, params string[] modules)
{
this.modules = new List <string>();
if (mode == AzureModule.AzureServiceManagement)
{
this.modules.Add(@"ServiceManagement\Azure\Azure.psd1");
}
else if (mode == AzureModule.AzureResourceManager)
{
this.modules.Add(@"ResourceManager\AzureResourceManager\AzureResourceManager.psd1");
}
else
{
throw new ArgumentException("Unknown command type for testing");
}
this.modules.Add("Assert.ps1");
this.modules.Add("Common.ps1");
this.modules.AddRange(modules);
}
public PowerShellTest(AzureModule commandMode, params string[] modules)
{
this.modules = new List <string>();
if (commandMode == AzureModule.AzureServiceManagement)
{
this.modules.Add(FileUtilities.GetContentFilePath(@"ServiceManagement\Azure\Azure.psd1"));
}
else if (commandMode == AzureModule.AzureResourceManager)
{
this.modules.Add(FileUtilities.GetContentFilePath(@"ResourceManager\AzureResourceManager\AzureResourceManager.psd1"));
}
else
{
throw new ArgumentException("Unknown command type for testing");
}
this.modules.Add("Assert.ps1");
this.modules.Add("Common.ps1");
this.modules.AddRange(modules);
TestingTracingInterceptor.AddToContext();
}
public PowerShellTest(AzureModule commandMode, params string[] modules)
{
this.modules = new List<string>();
if (commandMode == AzureModule.AzureServiceManagement)
{
this.modules.Add(FileUtilities.GetContentFilePath(@"ServiceManagement\Azure\Azure.psd1"));
}
else if (commandMode == AzureModule.AzureResourceManager)
{
this.modules.Add(FileUtilities.GetContentFilePath(@"ResourceManager\AzureResourceManager\AzureResourceManager.psd1"));
}
else
{
throw new ArgumentException("Unknown command type for testing");
}
this.modules.Add("Assert.ps1");
this.modules.Add("Common.ps1");
this.modules.AddRange(modules);
TestingTracingInterceptor.AddToContext();
}
public void SetupEnvironment(AzureModule mode)
{
// Ignore SSL errors
System.Net.ServicePointManager.ServerCertificateValidationCallback += (se, cert, chain, sslerror) => true;
// Set RunningMocked
if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Playback)
{
TestMockSupport.RunningMocked = true;
}
WindowsAzureProfile.Instance = new WindowsAzureProfile(new MockProfileStore());
if (!WindowsAzureProfile.Instance.Environments.ContainsKey(testEnvironmentName))
{
WindowsAzureProfile.Instance.AddEnvironment(new WindowsAzureEnvironment { Name = testEnvironmentName });
}
SetupAzureEnvironmentFromEnvironmentVariables(mode);
}
private void SetAuthenticationFactory(AzureModule mode, TestEnvironment rdfeEnvironment, TestEnvironment csmEnvironment)
{
string jwtToken = null;
X509Certificate2 certificate = null;
TestEnvironment currentEnvironment = (mode == AzureModule.AzureResourceManager ? csmEnvironment : rdfeEnvironment);
if (mode == AzureModule.AzureServiceManagement)
{
if (rdfeEnvironment.Credentials is TokenCloudCredentials)
{
jwtToken = ((TokenCloudCredentials)rdfeEnvironment.Credentials).Token;
}
if (rdfeEnvironment.Credentials is CertificateCloudCredentials)
{
certificate = ((CertificateCloudCredentials)rdfeEnvironment.Credentials).ManagementCertificate;
}
}
else
{
if (csmEnvironment.Credentials is TokenCloudCredentials)
{
jwtToken = ((TokenCloudCredentials)csmEnvironment.Credentials).Token;
}
if (csmEnvironment.Credentials is CertificateCloudCredentials)
{
certificate = ((CertificateCloudCredentials)csmEnvironment.Credentials).ManagementCertificate;
}
}
if (jwtToken != null)
{
AzureSession.AuthenticationFactory = new MockTokenAuthenticationFactory(currentEnvironment.UserName,
jwtToken);
}
else if (certificate != null)
{
AzureSession.AuthenticationFactory = new MockCertificateAuthenticationFactory(currentEnvironment.UserName,
certificate);
}
}
public void SetupEnvironment(AzureModule mode)
{
// Ignore SSL errors
System.Net.ServicePointManager.ServerCertificateValidationCallback += (se, cert, chain, sslerror) => true;
// Set RunningMocked
if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Playback)
{
TestMockSupport.RunningMocked = true;
}
WindowsAzureProfile.Instance = new WindowsAzureProfile(new MockProfileStore());
if (!WindowsAzureProfile.Instance.Environments.ContainsKey(testEnvironmentName))
{
WindowsAzureProfile.Instance.AddEnvironment(new WindowsAzureEnvironment {
Name = testEnvironmentName
});
}
SetupAzureEnvironmentFromEnvironmentVariables(mode);
}
private void SetAuthenticationFactory(AzureModule mode, TestEnvironment environment)
{
#if !NETSTANDARD
if (environment.AuthorizationContext.Certificate != null)
{
AzureSession.Instance.AuthenticationFactory = new MockCertificateAuthenticationFactory(environment.UserName,
environment.AuthorizationContext.Certificate);
}
else if (environment.AuthorizationContext.TokenCredentials.ContainsKey(TokenAudience.Management))
{
var httpMessage = new HttpRequestMessage();
environment.AuthorizationContext.TokenCredentials[TokenAudience.Management]
.ProcessHttpRequestAsync(httpMessage, CancellationToken.None)
.ConfigureAwait(false)
.GetAwaiter()
.GetResult();
AzureSession.Instance.AuthenticationFactory = new MockTokenAuthenticationFactory(
environment.UserName,
httpMessage.Headers.Authorization.Parameter);
}
#else
if (environment.TokenInfo.ContainsKey(TokenAudience.Management))
{
var httpMessage = new HttpRequestMessage();
environment.TokenInfo[TokenAudience.Management]
.ProcessHttpRequestAsync(httpMessage, CancellationToken.None)
.ConfigureAwait(false)
.GetAwaiter()
.GetResult();
AzureSession.Instance.AuthenticationFactory = new MockTokenAuthenticationFactory(
environment.UserName,
httpMessage.Headers.Authorization.Parameter);
}
#endif
}
public void SetupModules(AzureModule mode, params string[] modules)
{
this.modules = new List<string>();
if (mode == AzureModule.AzureProfile)
{
this.modules.Add(Path.Combine(PackageDirectory, @"ServiceManagement\Azure\Azure.psd1"));
this.modules.Add(Path.Combine(PackageDirectory, @"ResourceManager\AzureResourceManager\AzureResourceManager.psd1"));
}
else if (mode == AzureModule.AzureServiceManagement)
{
this.modules.Add(Path.Combine(PackageDirectory, @"ServiceManagement\Azure\Azure.psd1"));
}
this.modules.Add("Assert.ps1");
this.modules.Add("Common.ps1");
this.modules.AddRange(modules);
}
public void SetupModulesFromCommon(AzureModule mode, params string[] modules)
{
this.modules = new List<string>();
if (mode == AzureModule.AzureProfile)
{
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Profile\AzureRM.Profile.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Resources\AzureRM.Resources.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Resources\AzureRM.Tags.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"Storage\Azure.Storage\Azure.Storage.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ServiceManagement\Azure\Azure.psd1"));
}
else if (mode == AzureModule.AzureServiceManagement)
{
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Profile\AzureRM.Profile.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Resources\AzureRM.Resources.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Resources\AzureRM.Tags.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"Storage\Azure.Storage\Azure.Storage.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ServiceManagement\Azure\Azure.psd1"));
}
else if (mode == AzureModule.AzureResourceManager)
{
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Profile\AzureRM.Profile.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Resources\AzureRM.Resources.psd1"));
this.modules.Add(Path.Combine(PackageDirectoryFromCommon, @"ResourceManager\AzureResourceManager\AzureRM.Resources\AzureRM.Tags.psd1"));
}
else
{
throw new ArgumentException("Unknown command type for testing");
}
this.modules.Add("Assert.ps1");
this.modules.Add("Common.ps1");
this.modules.AddRange(modules);
}
public static string GetModuleName(AzureModule module)
{
switch (module)
{
case AzureModule.AzureServiceManagement:
return "Azure";
case AzureModule.AzureResourceManager:
return "AzureResourceManager";
case AzureModule.AzureProfile:
return "AzureProfile";
default:
throw new ArgumentOutOfRangeException(module.ToString());
}
}
private void SetupAzureEnvironmentFromEnvironmentVariables(AzureModule mode)
{
TestEnvironment rdfeEnvironment = new RDFETestEnvironmentFactory().GetTestEnvironment();
TestEnvironment csmEnvironment = new CSMTestEnvironmentFactory().GetTestEnvironment();
TestEnvironment currentEnvironment = (mode == AzureModule.AzureResourceManager ? csmEnvironment : rdfeEnvironment);
if (currentEnvironment.UserName == null)
{
currentEnvironment.UserName = "******";
}
SetEndpointsToDefaults(rdfeEnvironment, csmEnvironment);
SetAuthenticationFactory(mode, rdfeEnvironment, csmEnvironment);
AzureEnvironment environment = new AzureEnvironment { Name = testEnvironmentName };
Debug.Assert(currentEnvironment != null);
environment.Endpoints[AzureEnvironment.Endpoint.ActiveDirectory] = currentEnvironment.ActiveDirectoryEndpoint.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.Gallery] = currentEnvironment.GalleryUri.AbsoluteUri;
if (csmEnvironment != null)
{
environment.Endpoints[AzureEnvironment.Endpoint.ResourceManager] = csmEnvironment.BaseUri.AbsoluteUri;
}
if (rdfeEnvironment != null)
{
environment.Endpoints[AzureEnvironment.Endpoint.ServiceManagement] = rdfeEnvironment.BaseUri.AbsoluteUri;
}
if (!client.Profile.Environments.ContainsKey(testEnvironmentName))
{
client.AddOrSetEnvironment(environment);
}
testSubscription = new AzureSubscription()
{
Id = new Guid(currentEnvironment.SubscriptionId),
Name = testSubscriptionName,
Environment = testEnvironmentName,
Account = currentEnvironment.UserName,
Properties = new Dictionary<AzureSubscription.Property,string>
{
{ AzureSubscription.Property.Default, "True"},
{ AzureSubscription.Property.StorageAccount, Environment.GetEnvironmentVariable("AZURE_STORAGE_ACCOUNT")},
}
};
testAccount = new AzureAccount()
{
Id = currentEnvironment.UserName,
Type = AzureAccount.AccountType.User,
Properties = new Dictionary<AzureAccount.Property, string>
{
{ AzureAccount.Property.Subscriptions, currentEnvironment.SubscriptionId},
}
};
client.Profile.Subscriptions[testSubscription.Id] = testSubscription;
client.Profile.Accounts[testAccount.Id] = testAccount;
client.SetSubscriptionAsCurrent(testSubscription.Name, testSubscription.Account);
}
public void SetupModules(AzureModule mode, params string[] modules)
{
this.modules = new List<string>();
if (mode == AzureModule.AzureProfile)
{
this.modules.Add(@"ServiceManagement\Azure\Azure.psd1");
this.modules.Add(@"ResourceManager\AzureResourceManager\AzureResourceManager.psd1");
}
else if (mode == AzureModule.AzureServiceManagement)
{
this.modules.Add(@"ServiceManagement\Azure\Azure.psd1");
}
else if (mode == AzureModule.AzureResourceManager)
{
this.modules.Add(@"ResourceManager\AzureResourceManager\AzureResourceManager.psd1");
}
else
{
throw new ArgumentException("Unknown command type for testing");
}
this.modules.Add("Assert.ps1");
this.modules.Add("Common.ps1");
this.modules.AddRange(modules);
}
public ProfileTestController()
{
Module = AzureModule.AzureResourceManager;
helper = new EnvironmentSetupHelper();
}
private void SetupAzureEnvironmentFromEnvironmentVariables(AzureModule mode)
{
TestEnvironment rdfeEnvironment = new RDFETestEnvironmentFactory().GetTestEnvironment();
TestEnvironment csmEnvironment = new CSMTestEnvironmentFactory().GetTestEnvironment();
TestEnvironment currentEnvironment = (mode == AzureModule.AzureResourceManager
? csmEnvironment
: rdfeEnvironment);
string jwtToken;
if (mode == AzureModule.AzureResourceManager)
{
jwtToken = csmEnvironment.Credentials != null ?
((TokenCloudCredentials)csmEnvironment.Credentials).Token : null;
}
else if (mode == AzureModule.AzureServiceManagement)
{
jwtToken = rdfeEnvironment.Credentials != null ?
((TokenCloudCredentials)rdfeEnvironment.Credentials).Token : null;
}
else
{
throw new ArgumentException("Invalid module mode.");
}
SetEndpointsToDefaults(rdfeEnvironment, csmEnvironment);
WindowsAzureProfile.Instance.TokenProvider = new FakeAccessTokenProvider(jwtToken, csmEnvironment.UserName);
WindowsAzureProfile.Instance.CurrentEnvironment = WindowsAzureProfile.Instance.Environments[testEnvironmentName];
WindowsAzureProfile.Instance.CurrentEnvironment.ActiveDirectoryEndpoint =
currentEnvironment.ActiveDirectoryEndpoint.AbsoluteUri;
WindowsAzureProfile.Instance.CurrentEnvironment.GalleryEndpoint =
currentEnvironment.GalleryUri.AbsoluteUri;
WindowsAzureProfile.Instance.CurrentEnvironment.ResourceManagerEndpoint =
csmEnvironment.BaseUri.AbsoluteUri;
WindowsAzureProfile.Instance.CurrentEnvironment.ServiceEndpoint =
rdfeEnvironment.BaseUri.AbsoluteUri;
if (currentEnvironment.UserName == null)
{
currentEnvironment.UserName = "******";
}
testSubscription = new WindowsAzureSubscription(false, false)
{
SubscriptionId = currentEnvironment.SubscriptionId,
ActiveDirectoryEndpoint =
WindowsAzureProfile.Instance.CurrentEnvironment.ActiveDirectoryEndpoint,
ActiveDirectoryUserId = currentEnvironment.UserName,
SubscriptionName = currentEnvironment.SubscriptionId,
ServiceEndpoint = new Uri(WindowsAzureProfile.Instance.CurrentEnvironment.ServiceEndpoint),
ResourceManagerEndpoint = new Uri(WindowsAzureProfile.Instance.CurrentEnvironment.ResourceManagerEndpoint),
TokenProvider = WindowsAzureProfile.Instance.TokenProvider,
GalleryEndpoint = new Uri(WindowsAzureProfile.Instance.CurrentEnvironment.GalleryEndpoint),
SqlDatabaseDnsSuffix = WindowsAzureProfile.Instance.CurrentEnvironment.SqlDatabaseDnsSuffix,
CurrentStorageAccountName = Environment.GetEnvironmentVariable("AZURE_STORAGE_ACCOUNT"),
IsDefault = true
};
testEnvironment = currentEnvironment;
if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Playback)
{
testSubscription.SetAccessToken(new FakeAccessToken
{
AccessToken = "123",
UserId = testEnvironment.UserName
});
}
else
{
testSubscription.SetAccessToken(WindowsAzureProfile.Instance.TokenProvider.GetNewToken(WindowsAzureProfile.Instance.CurrentEnvironment));
}
WindowsAzureProfile.Instance.AddSubscription(testSubscription);
WindowsAzureProfile.Instance.Save();
}
public void SetupEnvironment(AzureModule mode)
{
SetupAzureEnvironmentFromEnvironmentVariables(mode);
ProfileClient.Profile.Save();
}
public void SetupEnvironment(AzureModule mode)
{
SetupAzureEnvironmentFromEnvironmentVariables(mode);
}
public ProfileTestController(AzureModule module)
{
Module = module;
helper = new EnvironmentSetupHelper();
}
public ProfileTestController()
{
Module = AzureModule.AzureResourceManager;
helper = new EnvironmentSetupHelper();
}
private void SetAuthenticationFactory(AzureModule mode, TestEnvironment rdfeEnvironment, TestEnvironment csmEnvironment)
{
if (mode == AzureModule.AzureServiceManagement)
{
SetAuthenticationFactory(rdfeEnvironment);
}
else
{
SetAuthenticationFactory(csmEnvironment);
}
}
private void SetupAzureEnvironmentFromEnvironmentVariables(AzureModule mode)
{
TestEnvironment rdfeEnvironment = new RDFETestEnvironmentFactory().GetTestEnvironment();
TestEnvironment csmEnvironment = new CSMTestEnvironmentFactory().GetTestEnvironment();
TestEnvironment currentEnvironment = (mode == AzureModule.AzureResourceManager ? csmEnvironment : rdfeEnvironment);
string jwtToken;
if (mode == AzureModule.AzureResourceManager)
{
jwtToken = csmEnvironment.Credentials != null ?
((TokenCloudCredentials)csmEnvironment.Credentials).Token : null;
}
else if (mode == AzureModule.AzureServiceManagement)
{
jwtToken = rdfeEnvironment.Credentials != null ?
((TokenCloudCredentials)rdfeEnvironment.Credentials).Token : null;
}
else
{
throw new ArgumentException("Invalid module mode.");
}
SetEndpointsToDefaults(rdfeEnvironment, csmEnvironment);
/*
WindowsAzureProfile.Instance.TokenProvider = new FakeAccessTokenProvider(
jwtToken,
csmEnvironment.UserName,
csmEnvironment.AuthorizationContext == null ? null : csmEnvironment.AuthorizationContext.TenatId);
*/
if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Playback)
{
AzureSession.AuthenticationFactory = new MockAuthenticationFactory();
}
else
{
AzureSession.AuthenticationFactory = new MockAuthenticationFactory(currentEnvironment.UserName, jwtToken);
}
AzureEnvironment environment = new AzureEnvironment { Name = testEnvironmentName };
Debug.Assert(currentEnvironment != null);
environment.Endpoints[AzureEnvironment.Endpoint.ActiveDirectory] = currentEnvironment.ActiveDirectoryEndpoint.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.Gallery] = currentEnvironment.GalleryUri.AbsoluteUri;
if (csmEnvironment != null)
{
environment.Endpoints[AzureEnvironment.Endpoint.ResourceManager] = csmEnvironment.BaseUri.AbsoluteUri;
}
if (rdfeEnvironment != null)
{
environment.Endpoints[AzureEnvironment.Endpoint.ServiceManagement] = rdfeEnvironment.BaseUri.AbsoluteUri;
}
if (currentEnvironment.UserName == null)
{
currentEnvironment.UserName = "******";
}
if (!client.Profile.Environments.ContainsKey(testEnvironmentName))
{
client.AddOrSetEnvironment(environment);
}
testSubscription = new AzureSubscription()
{
Id = new Guid(currentEnvironment.SubscriptionId),
Name = testSubscriptionName,
Environment = testEnvironmentName,
Account = currentEnvironment.UserName,
Properties = new Dictionary<AzureSubscription.Property,string>
{
{ AzureSubscription.Property.Default, "True"},
{ AzureSubscription.Property.StorageAccount, Environment.GetEnvironmentVariable("AZURE_STORAGE_ACCOUNT")},
}
};
testAccount = new AzureAccount()
{
Id = currentEnvironment.UserName,
Type = AzureAccount.AccountType.User,
Properties = new Dictionary<AzureAccount.Property, string>
{
{ AzureAccount.Property.Subscriptions, currentEnvironment.SubscriptionId},
}
};
client.Profile.Subscriptions[testSubscription.Id] = testSubscription;
client.Profile.Accounts[testAccount.Id] = testAccount;
client.SetSubscriptionAsCurrent(testSubscription.Name, testSubscription.Account);
}
public ProfileTestController(AzureModule module)
{
Module = module;
helper = new EnvironmentSetupHelper();
}
private void SetupAzureEnvironmentFromEnvironmentVariables(AzureModule mode)
{
TestEnvironment rdfeEnvironment = new RDFETestEnvironmentFactory().GetTestEnvironment();
TestEnvironment csmEnvironment = new CSMTestEnvironmentFactory().GetTestEnvironment();
TestEnvironment currentEnvironment = (mode == AzureModule.AzureResourceManager ? csmEnvironment : rdfeEnvironment);
if (currentEnvironment.UserName == null)
{
currentEnvironment.UserName = "******";
}
SetAuthenticationFactory(mode, rdfeEnvironment, csmEnvironment);
AzureEnvironment environment = new AzureEnvironment {
Name = testEnvironmentName
};
Debug.Assert(currentEnvironment != null);
environment.Endpoints[AzureEnvironment.Endpoint.ActiveDirectory] = currentEnvironment.Endpoints.AADAuthUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.Gallery] = currentEnvironment.Endpoints.GalleryUri.AbsoluteUri;
if (csmEnvironment != null)
{
environment.Endpoints[AzureEnvironment.Endpoint.ResourceManager] = csmEnvironment.BaseUri.AbsoluteUri;
}
if (rdfeEnvironment != null)
{
environment.Endpoints[AzureEnvironment.Endpoint.ServiceManagement] = rdfeEnvironment.BaseUri.AbsoluteUri;
}
if (!ProfileClient.Profile.Environments.ContainsKey(testEnvironmentName))
{
ProfileClient.AddOrSetEnvironment(environment);
}
if (currentEnvironment.SubscriptionId != null)
{
testSubscription = new AzureSubscription()
{
Id = new Guid(currentEnvironment.SubscriptionId),
Name = testSubscriptionName,
Environment = testEnvironmentName,
Account = currentEnvironment.UserName,
Properties = new Dictionary <AzureSubscription.Property, string>
{
{ AzureSubscription.Property.Default, "True" },
{
AzureSubscription.Property.StorageAccount,
Environment.GetEnvironmentVariable("AZURE_STORAGE_ACCOUNT")
},
}
};
testAccount = new AzureAccount()
{
Id = currentEnvironment.UserName,
Type = AzureAccount.AccountType.User,
Properties = new Dictionary <AzureAccount.Property, string>
{
{ AzureAccount.Property.Subscriptions, currentEnvironment.SubscriptionId },
}
};
ProfileClient.Profile.Subscriptions[testSubscription.Id] = testSubscription;
ProfileClient.Profile.Accounts[testAccount.Id] = testAccount;
ProfileClient.SetSubscriptionAsDefault(testSubscription.Name, testSubscription.Account);
}
}
private void SetAuthenticationFactory(AzureModule mode, TestEnvironment rdfeEnvironment, TestEnvironment csmEnvironment)
{
string jwtToken = null;
X509Certificate2 certificate = null;
TestEnvironment currentEnvironment = (mode == AzureModule.AzureResourceManager ? csmEnvironment : rdfeEnvironment);
if (mode == AzureModule.AzureServiceManagement)
{
if (rdfeEnvironment.Credentials is TokenCloudCredentials)
{
jwtToken = ((TokenCloudCredentials)rdfeEnvironment.Credentials).Token;
}
if (rdfeEnvironment.Credentials is CertificateCloudCredentials)
{
certificate = ((CertificateCloudCredentials)rdfeEnvironment.Credentials).ManagementCertificate;
}
}
else
{
if (csmEnvironment.Credentials is TokenCloudCredentials)
{
jwtToken = ((TokenCloudCredentials)csmEnvironment.Credentials).Token;
}
if (csmEnvironment.Credentials is CertificateCloudCredentials)
{
certificate = ((CertificateCloudCredentials)csmEnvironment.Credentials).ManagementCertificate;
}
}
if (jwtToken != null)
{
AzureSession.AuthenticationFactory = new MockTokenAuthenticationFactory(currentEnvironment.UserName,
jwtToken);
}
else if (certificate != null)
{
AzureSession.AuthenticationFactory = new MockCertificateAuthenticationFactory(currentEnvironment.UserName,
certificate);
}
}
protected CredentialTestBase(AzureModule commandMode)
{
helper = new CredentialTestHelper();
this.commandMode = commandMode;
}
public void SetupEnvironment(AzureModule mode)
{
SetupAzureEnvironmentFromEnvironmentVariables(mode);
client.Profile.Save();
}
private void SetupAzureEnvironmentFromEnvironmentVariables(AzureModule mode)
{
TestEnvironment rdfeEnvironment = new RDFETestEnvironmentFactory().GetTestEnvironment();
TestEnvironment csmEnvironment = new CSMTestEnvironmentFactory().GetTestEnvironment();
TestEnvironment currentEnvironment = (mode == AzureModule.AzureResourceManager ? csmEnvironment : rdfeEnvironment);
if (currentEnvironment.UserName == null)
{
currentEnvironment.UserName = "******";
}
SetAuthenticationFactory(mode, rdfeEnvironment, csmEnvironment);
AzureEnvironment environment = new AzureEnvironment {
Name = testEnvironmentName
};
Debug.Assert(currentEnvironment != null);
environment.ActiveDirectoryAuthority = currentEnvironment.Endpoints.AADAuthUri.AbsoluteUri;
environment.GalleryUrl = currentEnvironment.Endpoints.GalleryUri.AbsoluteUri;
if (csmEnvironment != null)
{
environment.ResourceManagerUrl = csmEnvironment.BaseUri.AbsoluteUri;
}
if (rdfeEnvironment != null)
{
environment.ServiceManagementUrl = rdfeEnvironment.BaseUri.AbsoluteUri;
}
if (!ProfileClient.Profile.EnvironmentTable.ContainsKey(testEnvironmentName))
{
ProfileClient.AddOrSetEnvironment(environment);
}
if (currentEnvironment.SubscriptionId != null)
{
testSubscription = new AzureSubscription()
{
Id = currentEnvironment.SubscriptionId,
Name = testSubscriptionName,
};
testSubscription.SetEnvironment(testEnvironmentName);
testSubscription.SetAccount(currentEnvironment.UserName);
testSubscription.SetStorageAccount(Environment.GetEnvironmentVariable("AZURE_STORAGE_ACCOUNT"));
testSubscription.SetDefault();
testAccount = new AzureAccount()
{
Id = currentEnvironment.UserName,
Type = AzureAccount.AccountType.User,
};
testAccount.SetSubscriptions(currentEnvironment.SubscriptionId);
ProfileClient.Profile.SubscriptionTable[testSubscription.GetId()] = testSubscription;
ProfileClient.Profile.AccountTable[testAccount.Id] = testAccount;
ProfileClient.SetSubscriptionAsDefault(testSubscription.Name, testSubscription.GetAccount());
}
}
private 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
#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 CredentialTestBase(AzureModule commandMode)
{
helper = new CredentialTestHelper();
this.commandMode = commandMode;
}
/// <summary>
/// Gets the root installation path for the given Azure module.
/// </summary>
/// <param name="module" >The module name</param>
/// <returns>The module full path</returns>
public static string GetPSModulePathForModule(AzureModule module)
{
return GetContentFilePath(GetInstallPath(), GetModuleFolderName(module));
}
private void SetAuthenticationFactory(AzureModule mode, TestEnvironment environment)
{
string jwtToken = null;
X509Certificate2 certificate = null;
if (environment.Credentials is TokenCloudCredentials)
{
jwtToken = ((TokenCloudCredentials)environment.Credentials).Token;
}
if (environment.Credentials is CertificateCloudCredentials)
{
certificate = ((CertificateCloudCredentials)environment.Credentials).ManagementCertificate;
}
if (jwtToken != null)
{
AzureSession.AuthenticationFactory = new MockTokenAuthenticationFactory(environment.UserName,
jwtToken);
}
else if (certificate != null)
{
AzureSession.AuthenticationFactory = new MockCertificateAuthenticationFactory(environment.UserName,
certificate);
}
}
public static string GetModuleFolderName(AzureModule module)
{
return module.ToString().Replace("Azure", "");
}
private void SetupAzureEnvironmentFromEnvironmentVariables(AzureModule mode)
{
TestEnvironment rdfeEnvironment = new RDFETestEnvironmentFactory().GetTestEnvironment();
TestEnvironment csmEnvironment = new CSMTestEnvironmentFactory().GetTestEnvironment();
TestEnvironment currentEnvironment = (mode == AzureModule.AzureResourceManager
? csmEnvironment
: rdfeEnvironment);
string jwtToken;
if (mode == AzureModule.AzureResourceManager)
{
jwtToken = csmEnvironment.Credentials != null ?
((TokenCloudCredentials)csmEnvironment.Credentials).Token : null;
}
else if (mode == AzureModule.AzureServiceManagement)
{
jwtToken = rdfeEnvironment.Credentials != null ?
((TokenCloudCredentials)rdfeEnvironment.Credentials).Token : null;
}
else
{
throw new ArgumentException("Invalid module mode.");
}
SetEndpointsToDefaults(rdfeEnvironment, csmEnvironment);
WindowsAzureProfile.Instance.TokenProvider = new FakeAccessTokenProvider(jwtToken, csmEnvironment.UserName);
WindowsAzureProfile.Instance.CurrentEnvironment = WindowsAzureProfile.Instance.Environments[testEnvironmentName];
WindowsAzureProfile.Instance.CurrentEnvironment.ActiveDirectoryEndpoint =
currentEnvironment.ActiveDirectoryEndpoint.AbsoluteUri;
WindowsAzureProfile.Instance.CurrentEnvironment.GalleryEndpoint =
currentEnvironment.GalleryUri.AbsoluteUri;
WindowsAzureProfile.Instance.CurrentEnvironment.ResourceManagerEndpoint =
csmEnvironment.BaseUri.AbsoluteUri;
WindowsAzureProfile.Instance.CurrentEnvironment.ServiceEndpoint =
rdfeEnvironment.BaseUri.AbsoluteUri;
if (currentEnvironment.UserName == null)
{
currentEnvironment.UserName = "******";
}
testSubscription = new WindowsAzureSubscription(false, false)
{
SubscriptionId = currentEnvironment.SubscriptionId,
ActiveDirectoryEndpoint =
WindowsAzureProfile.Instance.CurrentEnvironment.ActiveDirectoryEndpoint,
ActiveDirectoryUserId = currentEnvironment.UserName,
SubscriptionName = currentEnvironment.SubscriptionId,
ServiceEndpoint = new Uri(WindowsAzureProfile.Instance.CurrentEnvironment.ServiceEndpoint),
ResourceManagerEndpoint = new Uri(WindowsAzureProfile.Instance.CurrentEnvironment.ResourceManagerEndpoint),
TokenProvider = WindowsAzureProfile.Instance.TokenProvider,
GalleryEndpoint = new Uri(WindowsAzureProfile.Instance.CurrentEnvironment.GalleryEndpoint),
SqlDatabaseDnsSuffix = WindowsAzureProfile.Instance.CurrentEnvironment.SqlDatabaseDnsSuffix,
CurrentStorageAccountName = Environment.GetEnvironmentVariable("AZURE_STORAGE_ACCOUNT"),
IsDefault = true
};
testEnvironment = currentEnvironment;
if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Playback)
{
testSubscription.SetAccessToken(new FakeAccessToken
{
AccessToken = "123",
UserId = testEnvironment.UserName
});
}
else
{
testSubscription.SetAccessToken(WindowsAzureProfile.Instance.TokenProvider.GetNewToken(WindowsAzureProfile.Instance.CurrentEnvironment));
}
WindowsAzureProfile.Instance.AddSubscription(testSubscription);
WindowsAzureProfile.Instance.Save();
}
/// <summary>
/// Gets the root installation path for the given Azure module.
/// </summary>
/// <param name="module" >The module name</param>
/// <returns>The module full path</returns>
public static string GetPSModulePathForModule(AzureModule module)
{
return(GetContentFilePath(GetInstallPath(), GetModuleFolderName(module)));
}
private void SetupAzureEnvironmentFromEnvironmentVariables(AzureModule mode)
{
TestEnvironment currentEnvironment = null;
if (mode == AzureModule.AzureResourceManager)
{
currentEnvironment = new CSMTestEnvironmentFactory().GetTestEnvironment();
}
else
{
currentEnvironment = new RDFETestEnvironmentFactory().GetTestEnvironment();
}
if (currentEnvironment.UserName == null)
{
currentEnvironment.UserName = "******";
}
SetAuthenticationFactory(mode, currentEnvironment);
AzureEnvironment environment = new AzureEnvironment { Name = testEnvironmentName };
Debug.Assert(currentEnvironment != null);
environment.Endpoints[AzureEnvironment.Endpoint.ActiveDirectory] = currentEnvironment.Endpoints.AADAuthUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.Gallery] = currentEnvironment.Endpoints.GalleryUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.ServiceManagement] = currentEnvironment.BaseUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.ResourceManager] = currentEnvironment.Endpoints.ResourceManagementUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.Graph] = currentEnvironment.Endpoints.GraphUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.AzureDataLakeAnalyticsCatalogAndJobEndpointSuffix] = currentEnvironment.Endpoints.DataLakeAnalyticsJobAndCatalogServiceUri.OriginalString.Replace("https://", ""); // because it is just a sufix
environment.Endpoints[AzureEnvironment.Endpoint.AzureDataLakeStoreFileSystemEndpointSuffix] = currentEnvironment.Endpoints.DataLakeStoreServiceUri.OriginalString.Replace("https://", ""); // because it is just a sufix
if (!ProfileClient.Profile.Environments.ContainsKey(testEnvironmentName))
{
ProfileClient.AddOrSetEnvironment(environment);
}
if (!AzureRmProfileProvider.Instance.Profile.Environments.ContainsKey(testEnvironmentName))
{
AzureRmProfileProvider.Instance.Profile.Environments[testEnvironmentName] = environment;
}
if (currentEnvironment.SubscriptionId != null)
{
testSubscription = new AzureSubscription()
{
Id = new Guid(currentEnvironment.SubscriptionId),
Name = testSubscriptionName,
Environment = testEnvironmentName,
Account = currentEnvironment.UserName,
Properties = new Dictionary<AzureSubscription.Property, string>
{
{AzureSubscription.Property.Default, "True"},
{
AzureSubscription.Property.StorageAccount,
Environment.GetEnvironmentVariable("AZURE_STORAGE_ACCOUNT")
},
}
};
testAccount = new AzureAccount()
{
Id = currentEnvironment.UserName,
Type = AzureAccount.AccountType.User,
Properties = new Dictionary<AzureAccount.Property, string>
{
{AzureAccount.Property.Subscriptions, currentEnvironment.SubscriptionId},
}
};
ProfileClient.Profile.Subscriptions[testSubscription.Id] = testSubscription;
ProfileClient.Profile.Accounts[testAccount.Id] = testAccount;
ProfileClient.SetSubscriptionAsDefault(testSubscription.Name, testSubscription.Account);
var testTenant = new AzureTenant() { Id = Guid.NewGuid() };
if (!string.IsNullOrEmpty(currentEnvironment.Tenant))
{
Guid tenant;
if (Guid.TryParse(currentEnvironment.Tenant, out tenant))
{
testTenant.Id = tenant;
}
}
AzureRmProfileProvider.Instance.Profile.Context = new AzureContext(testSubscription, testAccount, environment, testTenant);
}
}
public static string GetModuleFolderName(AzureModule module)
{
return(module.ToString().Replace("Azure", ""));
}
private void SetupAzureEnvironmentFromEnvironmentVariables(AzureModule mode)
{
TestEnvironment rdfeEnvironment = new RDFETestEnvironmentFactory().GetTestEnvironment();
TestEnvironment csmEnvironment = new CSMTestEnvironmentFactory().GetTestEnvironment();
TestEnvironment currentEnvironment = (mode == AzureModule.AzureResourceManager ? csmEnvironment : rdfeEnvironment);
string jwtToken;
if (mode == AzureModule.AzureResourceManager)
{
jwtToken = csmEnvironment.Credentials != null ?
((TokenCloudCredentials)csmEnvironment.Credentials).Token : null;
}
else if (mode == AzureModule.AzureServiceManagement)
{
jwtToken = rdfeEnvironment.Credentials != null ?
((TokenCloudCredentials)rdfeEnvironment.Credentials).Token : null;
}
else
{
throw new ArgumentException("Invalid module mode.");
}
SetEndpointsToDefaults(rdfeEnvironment, csmEnvironment);
/*
* WindowsAzureProfile.Instance.TokenProvider = new FakeAccessTokenProvider(
* jwtToken,
* csmEnvironment.UserName,
* csmEnvironment.AuthorizationContext == null ? null : csmEnvironment.AuthorizationContext.TenatId);
*/
if (HttpMockServer.GetCurrentMode() == HttpRecorderMode.Playback)
{
AzureSession.AuthenticationFactory = new MockAuthenticationFactory();
}
else
{
AzureSession.AuthenticationFactory = new MockAuthenticationFactory(currentEnvironment.UserName, jwtToken);
}
AzureEnvironment environment = new AzureEnvironment {
Name = testEnvironmentName
};
Debug.Assert(currentEnvironment != null);
environment.Endpoints[AzureEnvironment.Endpoint.ActiveDirectory] = currentEnvironment.ActiveDirectoryEndpoint.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.Gallery] = currentEnvironment.GalleryUri.AbsoluteUri;
if (csmEnvironment != null)
{
environment.Endpoints[AzureEnvironment.Endpoint.ResourceManager] = csmEnvironment.BaseUri.AbsoluteUri;
}
if (rdfeEnvironment != null)
{
environment.Endpoints[AzureEnvironment.Endpoint.ServiceManagement] = rdfeEnvironment.BaseUri.AbsoluteUri;
}
if (currentEnvironment.UserName == null)
{
currentEnvironment.UserName = "******";
}
if (!client.Profile.Environments.ContainsKey(testEnvironmentName))
{
client.AddOrSetEnvironment(environment);
}
testSubscription = new AzureSubscription()
{
Id = new Guid(currentEnvironment.SubscriptionId),
Name = testSubscriptionName,
Environment = testEnvironmentName,
Account = currentEnvironment.UserName,
Properties = new Dictionary <AzureSubscription.Property, string>
{
{ AzureSubscription.Property.Default, "True" },
{ AzureSubscription.Property.StorageAccount, Environment.GetEnvironmentVariable("AZURE_STORAGE_ACCOUNT") },
}
};
testAccount = new AzureAccount()
{
Id = currentEnvironment.UserName,
Type = AzureAccount.AccountType.User,
Properties = new Dictionary <AzureAccount.Property, string>
{
{ AzureAccount.Property.Subscriptions, currentEnvironment.SubscriptionId },
}
};
client.Profile.Subscriptions[testSubscription.Id] = testSubscription;
client.Profile.Accounts[testAccount.Id] = testAccount;
client.SetSubscriptionAsCurrent(testSubscription.Name, testSubscription.Account);
}
private void SetupAzureEnvironmentFromEnvironmentVariables(AzureModule mode)
{
TestEnvironment currentEnvironment = null;
if (mode == AzureModule.AzureResourceManager)
{
currentEnvironment = new CSMTestEnvironmentFactory().GetTestEnvironment();
}
else
{
currentEnvironment = new RDFETestEnvironmentFactory().GetTestEnvironment();
}
if (currentEnvironment.UserName == null)
{
currentEnvironment.UserName = "******";
}
SetAuthenticationFactory(mode, currentEnvironment);
AzureEnvironment environment = new AzureEnvironment {
Name = testEnvironmentName
};
Debug.Assert(currentEnvironment != null);
environment.Endpoints[AzureEnvironment.Endpoint.ActiveDirectory] = currentEnvironment.Endpoints.AADAuthUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.Gallery] = currentEnvironment.Endpoints.GalleryUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.ServiceManagement] = currentEnvironment.BaseUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.ResourceManager] = currentEnvironment.Endpoints.ResourceManagementUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.Graph] = currentEnvironment.Endpoints.GraphUri.AbsoluteUri;
environment.Endpoints[AzureEnvironment.Endpoint.AzureDataLakeAnalyticsCatalogAndJobEndpointSuffix] = currentEnvironment.Endpoints.DataLakeAnalyticsJobAndCatalogServiceUri.OriginalString.Replace("https://", ""); // because it is just a sufix
environment.Endpoints[AzureEnvironment.Endpoint.AzureDataLakeStoreFileSystemEndpointSuffix] = currentEnvironment.Endpoints.DataLakeStoreServiceUri.OriginalString.Replace("https://", ""); // because it is just a sufix
if (!ProfileClient.Profile.Environments.ContainsKey(testEnvironmentName))
{
ProfileClient.AddOrSetEnvironment(environment);
}
if (!AzureRmProfileProvider.Instance.Profile.Environments.ContainsKey(testEnvironmentName))
{
AzureRmProfileProvider.Instance.Profile.Environments[testEnvironmentName] = environment;
}
if (currentEnvironment.SubscriptionId != null)
{
testSubscription = new AzureSubscription()
{
Id = new Guid(currentEnvironment.SubscriptionId),
Name = testSubscriptionName,
Environment = testEnvironmentName,
Account = currentEnvironment.UserName,
Properties = new Dictionary <AzureSubscription.Property, string>
{
{ AzureSubscription.Property.Default, "True" },
{
AzureSubscription.Property.StorageAccount,
Environment.GetEnvironmentVariable("AZURE_STORAGE_ACCOUNT")
},
}
};
testAccount = new AzureAccount()
{
Id = currentEnvironment.UserName,
Type = AzureAccount.AccountType.User,
Properties = new Dictionary <AzureAccount.Property, string>
{
{ AzureAccount.Property.Subscriptions, currentEnvironment.SubscriptionId },
}
};
ProfileClient.Profile.Subscriptions[testSubscription.Id] = testSubscription;
ProfileClient.Profile.Accounts[testAccount.Id] = testAccount;
ProfileClient.SetSubscriptionAsDefault(testSubscription.Name, testSubscription.Account);
var testTenant = new AzureTenant()
{
Id = Guid.NewGuid()
};
if (!string.IsNullOrEmpty(currentEnvironment.Tenant))
{
Guid tenant;
if (Guid.TryParse(currentEnvironment.Tenant, out tenant))
{
testTenant.Id = tenant;
}
}
AzureRmProfileProvider.Instance.Profile.Context = new AzureContext(testSubscription, testAccount, environment, testTenant);
}
}
private void SetAuthenticationFactory(AzureModule mode, TestEnvironment environment)
{
if(environment.AuthorizationContext.Certificate != null)
{
AzureSession.AuthenticationFactory = new MockCertificateAuthenticationFactory(environment.UserName,
environment.AuthorizationContext.Certificate);
}
else if(environment.AuthorizationContext.TokenCredentials.ContainsKey(TokenAudience.Management))
{
var httpMessage = new HttpRequestMessage();
environment.AuthorizationContext.TokenCredentials[TokenAudience.Management]
.ProcessHttpRequestAsync(httpMessage, CancellationToken.None)
.ConfigureAwait(false)
.GetAwaiter()
.GetResult();
AzureSession.AuthenticationFactory = new MockTokenAuthenticationFactory(
environment.UserName,
httpMessage.Headers.Authorization.Parameter);
}
}
