private static AzureHDInsightCluster DisableRdpAccessToCluster(
IHDInsightCertificateCredential creds, AzureHDInsightCluster containerWithRdpAccessDisabled)
{
IManageAzureHDInsightRdpAccessCommand rdpManagementClient =
ServiceLocator.Instance.Locate<IAzureHDInsightCommandFactory>().CreateManageRdpAccess();
rdpManagementClient.CurrentSubscription = GetCurrentSubscription();
rdpManagementClient.RdpCredential = GetAzurePsCredentials();
rdpManagementClient.Name = containerWithRdpAccessDisabled.Name;
rdpManagementClient.Location = containerWithRdpAccessDisabled.Location;
rdpManagementClient.Enable = false;
rdpManagementClient.EndProcessing();
return rdpManagementClient.Output.First();
}
internal static void RevokeRdpAccessToCluster(
IHDInsightCertificateCredential connectionCredentials, AzureHDInsightCluster cluster, IRunspace runspace)
{
IPipelineResult results =
runspace.NewPipeline()
.AddCommand(CmdletConstants.RevokeAzureHDInsightRdpAccess)
.WithParameter(CmdletConstants.Location, cluster.Location)
.WithParameter(CmdletConstants.Name, cluster.Name)
.Invoke();
AzureHDInsightCluster accessRevokedCluster = GetCluster(connectionCredentials, cluster.Name, runspace);
Assert.NotNull(accessRevokedCluster);
Assert.True(string.IsNullOrEmpty(accessRevokedCluster.RdpUserName));
}
private static AzureHDInsightCluster EnableHttpAccessToCluster(
IHDInsightCertificateCredential creds, AzureHDInsightCluster containerWithHttpAccessDisabled, string httpUserName, string httpPassword)
{
IManageAzureHDInsightHttpAccessCommand httpManagementClient =
ServiceLocator.Instance.Locate<IAzureHDInsightCommandFactory>().CreateManageHttpAccess();
httpManagementClient.CurrentSubscription = GetCurrentSubscription();
httpManagementClient.Credential = GetPSCredential(httpUserName, httpPassword);
httpManagementClient.Name = containerWithHttpAccessDisabled.Name;
httpManagementClient.Location = containerWithHttpAccessDisabled.Location;
httpManagementClient.Enable = true;
httpManagementClient.EndProcessing();
return httpManagementClient.Output.First();
}
private static void ValidateGetCluster(AzureHDInsightCluster cluster)
{
Assert.AreEqual(TestCredentials.WellKnownCluster.DnsName, cluster.Name);
Assert.AreEqual(TestCredentials.WellKnownCluster.Version, cluster.Version);
WabStorageAccountConfiguration defaultStorageAccount = GetWellKnownStorageAccounts().First();
Assert.AreEqual(defaultStorageAccount.Key, cluster.DefaultStorageAccount.StorageAccountKey);
Assert.AreEqual(defaultStorageAccount.Name, cluster.DefaultStorageAccount.StorageAccountName);
Assert.AreEqual(defaultStorageAccount.Container, cluster.DefaultStorageAccount.StorageContainerName);
foreach (WabStorageAccountConfiguration account in GetWellKnownStorageAccounts().Skip(1))
{
AzureHDInsightStorageAccount deserializedAccount =
cluster.StorageAccounts.FirstOrDefault(acc => acc.StorageAccountName == account.Name);
Assert.IsNotNull(deserializedAccount, account.Name);
Assert.AreEqual(account.Key, deserializedAccount.StorageAccountKey);
}
}
private static void ValidateGetCluster(AzureHDInsightCluster cluster)
{
Assert.AreEqual(TestCredentials.WellKnownCluster.DnsName, cluster.Name);
Assert.AreEqual(TestCredentials.HadoopUserName, cluster.HttpUserName);
Assert.AreEqual(TestCredentials.AzurePassword, cluster.HttpPassword);
Assert.AreEqual(ClusterState.Running, cluster.State);
Assert.IsFalse(string.IsNullOrEmpty(cluster.ConnectionUrl));
Assert.AreEqual(VersionStatus.Compatible.ToString(), cluster.VersionStatus);
Assert.AreEqual(TestCredentials.WellKnownCluster.Version, cluster.Version);
WabStorageAccountConfiguration defaultStorageAccount = GetWellKnownStorageAccounts().First();
Assert.AreEqual(defaultStorageAccount.Key, cluster.DefaultStorageAccount.StorageAccountKey);
Assert.AreEqual(defaultStorageAccount.Name, cluster.DefaultStorageAccount.StorageAccountName);
Assert.AreEqual(defaultStorageAccount.Container, cluster.DefaultStorageAccount.StorageContainerName);
foreach (WabStorageAccountConfiguration account in GetWellKnownStorageAccounts().Skip(1))
{
AzureHDInsightStorageAccount deserializedAccount =
cluster.StorageAccounts.FirstOrDefault(acc => acc.StorageAccountName == account.Name);
Assert.IsNotNull(deserializedAccount, account.Name);
Assert.AreEqual(account.Key, deserializedAccount.StorageAccountKey);
}
}
private static void ValidateGetCluster(AzureHDInsightCluster expected, AzureHDInsightCluster actual)
{
Assert.AreEqual(expected.Name, actual.Name);
Assert.AreEqual(expected.Version, actual.Version);
Assert.AreEqual(expected.HttpUserName, actual.HttpUserName);
Assert.AreEqual(expected.HttpPassword, actual.HttpPassword);
}
private static AzureHDInsightCluster EnableRdpAccessToCluster(
IHDInsightCertificateCredential creds, AzureHDInsightCluster containerWithRdpAccessDisabled, string rdpUserName, string rdpPassword, DateTime expiry)
{
IManageAzureHDInsightRdpAccessCommand rdpManagementClient =
ServiceLocator.Instance.Locate<IAzureHDInsightCommandFactory>().CreateManageRdpAccess();
rdpManagementClient.CurrentSubscription = GetCurrentSubscription();
rdpManagementClient.RdpCredential = GetPSCredential(rdpUserName, rdpPassword);
rdpManagementClient.Name = containerWithRdpAccessDisabled.Name;
rdpManagementClient.Location = containerWithRdpAccessDisabled.Location;
rdpManagementClient.RdpAccessExpiry = expiry;
rdpManagementClient.Enable = true;
rdpManagementClient.EndProcessing();
return rdpManagementClient.Output.First();
}
