public void TestCreateGetUpdateDeleteSqlVirtualMachine()
{
using (SqlVirtualMachineTestContext context = new SqlVirtualMachineTestContext(this))
{
// Create SQL VM
ISqlVirtualMachinesOperations sqlOperations = context.getSqlClient().SqlVirtualMachines;
Dictionary <string, SqlVirtualMachineModel> sqlVirtualMachines = new Dictionary <string, SqlVirtualMachineModel>();
SqlVirtualMachineModel sqlVM = null;
for (int i = 0; i < 3; i++)
{
sqlVM = SqlVirtualMachineTestBase.CreateSqlVirtualMachine(context);
Assert.NotNull(sqlVM);
sqlVirtualMachines[sqlVM.Id] = sqlVM;
}
// Recover
foreach (string id in sqlVirtualMachines.Keys)
{
sqlVM = sqlOperations.Get(context.resourceGroup.Name, sqlVirtualMachines[id].Name);
Assert.NotNull(sqlVM);
SqlVirtualMachineTestBase.ValidateSqlVirtualMachine(sqlVM, sqlVirtualMachines[id]);
}
// Update
string key = "test", value = "updateTag";
sqlOperations.Update(context.resourceGroup.Name, sqlVM.Name, new SqlVirtualMachineUpdate
{
Tags = new Dictionary <string, string>
{
{ key, value }
}
});
SqlVirtualMachineModel sqlVM2 = sqlOperations.Get(context.resourceGroup.Name, sqlVM.Name);
SqlVirtualMachineTestBase.ValidateSqlVirtualMachine(sqlVM, sqlVM2, sameTags: false);
Assert.Equal(1, sqlVM2.Tags.Keys.Count);
Assert.Equal(value, sqlVM2.Tags[key]);
// Delete
sqlOperations.Delete(context.resourceGroup.Name, sqlVM2.Name);
sqlVirtualMachines.Remove(sqlVM2.Id);
// List
IPage <SqlVirtualMachineModel> recovered = sqlOperations.List();
var iter = recovered.GetEnumerator();
while (iter.MoveNext())
{
sqlVM = iter.Current;
Assert.NotEqual(sqlVM.Id, sqlVM2.Id);
if (sqlVirtualMachines.ContainsKey(sqlVM.Id))
{
SqlVirtualMachineTestBase.ValidateSqlVirtualMachine(sqlVM, sqlVirtualMachines[sqlVM.Id]);
sqlVirtualMachines.Remove(sqlVM.Id);
}
}
Assert.Empty(sqlVirtualMachines.Keys);
iter.Dispose();
}
}
public static void ValidateSqlVirtualMachine(SqlVirtualMachineModel sqlVM1, SqlVirtualMachineModel sqlVM2, bool sameTags = true)
{
Assert.Equal(sqlVM1.Id, sqlVM2.Id);
Assert.Equal(sqlVM1.Name, sqlVM2.Name);
Assert.Equal(sqlVM1.Location, sqlVM2.Location);
Assert.Equal(sqlVM1.SqlManagement, sqlVM2.SqlManagement);
if (sameTags)
{
Assert.True(ValidateTags(sqlVM1, sqlVM2));
}
}
public async Task <SqlResourceDto> GetSqlVirtualMachineResourceAsync(ResourceDto _sqlResource)
{
if (_sqlResource == null || _sqlResource == default(ResourceDto))
{
return(null);
}
DefaultAzureCredential credential;
if (_sqlResourceInfoInventoryConfiguration.UseSystemAssignedManagedIdentity)
{
credential = new DefaultAzureCredential();
}
else
{
credential = new DefaultAzureCredential(new DefaultAzureCredentialOptions {
ManagedIdentityClientId = _sqlResourceInfoInventoryConfiguration.UserAssignedManagedIdentityClientId
});
}
TokenRequestContext tokenRequestContext = new TokenRequestContext(new[] { Constants.AzureResourceManagerAPIDefaultScope });
AccessToken tokenRequestResult = await credential.GetTokenAsync(tokenRequestContext);
ServiceClientCredentials serviceClientCreds = new TokenCredentials(tokenRequestResult.Token);
Microsoft.Azure.Management.SqlVirtualMachine.SqlVirtualMachineManagementClient sqlClient = new Microsoft.Azure.Management.SqlVirtualMachine.SqlVirtualMachineManagementClient(serviceClientCreds);
sqlClient.SubscriptionId = _sqlResource.subscriptionId;
var response = await sqlClient.SqlVirtualMachines.GetWithHttpMessagesAsync(_sqlResource.resourceGroup, _sqlResource.name);
SqlVirtualMachineModel server = response.Body;
if (server != null)
{
SqlResourceDto newServer = new SqlResourceDto()
{
ServerNameId = server.Id,
ServerName = server.Name,
SubscriptionId = _sqlResource.subscriptionId,
Type = _sqlResource.type
};
return(newServer);
}
return(null);
}
private static SqlVirtualMachineModel prepareMachine(SqlVirtualMachineTestContext context, SqlVirtualMachineGroup group, VirtualMachine vm, string domain, string login, string password)
{
// Create the sql virtual machine
SqlVirtualMachineModel sqlVM = CreateSqlVirtualMachine(context, virtualMachine: vm);
// Join domain
IVirtualMachineExtensionsOperations operations = context.client.computeClient.VirtualMachineExtensions;
VirtualMachineExtension joinDomain = operations.CreateOrUpdate(context.resourceGroup.Name, vm.Name, "joindomain", new VirtualMachineExtension()
{
Location = context.location,
Publisher = "Microsoft.Compute",
VirtualMachineExtensionType = "JsonADDomainExtension",
TypeHandlerVersion = "1.3",
AutoUpgradeMinorVersion = true,
Settings = new JoinDomainSettings()
{
Name = domain + ".com",
OUPath = "",
User = domain + "\\" + login,
Restart = "true",
Options = "3"
},
ProtectedSettings = new JoinDomainProtectedSettings()
{
Password = password
}
});
// Join sql virtual machine group
context.getSqlClient().SqlVirtualMachines.CreateOrUpdate(context.resourceGroup.Name, sqlVM.Name, new SqlVirtualMachineModel()
{
Location = context.location,
SqlVirtualMachineGroupResourceId = group.Id,
VirtualMachineResourceId = sqlVM.VirtualMachineResourceId,
WsfcDomainCredentials = new WsfcDomainCredentials()
{
ClusterBootstrapAccountPassword = password,
ClusterOperatorAccountPassword = password,
SqlServiceAccountPassword = password
}
});
return(sqlVM);
}
/// <summary>
/// Convert a Management.SqlVirtualMachine.Models.SqlVirtualMachineModel to AzureSqlVirtualMachineModel
/// </summary>
/// <param name="resourceGroupName">The resource group the sql virtual machine is in</param>
/// <param name="resp">The management client sql virtual machine response to convert</param>
/// <returns>The converted sql virtual machine model</returns>
private AzureSqlVMModel CreateSqlVirtualMachineModelFromResponse(SqlVirtualMachineModel resp)
{
// Extract the resource group name from the ID.
// ID is in the form:
// /subscriptions/00000000-0000-0000-0000-000000000000/resourceGroups/rgName/providers/Microsoft.SqlVirtualMachine/SqlVirtualMachine/sqlVirtualMachineName
string[] segments = resp.Id.Split('/');
AzureSqlVMModel model = new AzureSqlVMModel(segments[4])
{
Name = resp.Name,
Location = resp.Location,
Sku = resp.SqlImageSku,
Offer = resp.SqlImageOffer,
VirtualMachineId = resp.VirtualMachineResourceId,
SqlManagementType = resp.SqlManagement,
LicenseType = resp.SqlServerLicenseType,
Tags = TagsConversionHelper.CreateTagDictionary(TagsConversionHelper.CreateTagHashtable(resp.Tags), false),
ResourceId = resp.Id
};
// Retrieve group
if (!string.IsNullOrEmpty(resp.SqlVirtualMachineGroupResourceId))
{
var client = AzureSession.Instance.ClientFactory.CreateArmClient <SqlVirtualMachineManagementClient>(DefaultContext, AzureEnvironment.Endpoint.ResourceManager);
string[] groupId = resp.SqlVirtualMachineGroupResourceId.Split('/');
SqlVirtualMachineGroup group = client.SqlVirtualMachineGroups.Get(groupId[4], groupId[8]);
model.SqlVirtualMachineGroup = new AzureSqlVMGroupModel(groupId[4])
{
Name = group.Name,
Location = group.Location,
Sku = group.SqlImageSku,
Offer = group.SqlImageOffer,
ResourceId = group.Id,
Tag = TagsConversionHelper.CreateTagDictionary(TagsConversionHelper.CreateTagHashtable(group.Tags), false),
WsfcDomainProfile = group.WsfcDomainProfile
};
}
return(model);
}
public void TestListByGroupSqlVirtualMachine()
{
using (SqlVirtualMachineTestContext context = new SqlVirtualMachineTestContext(this))
{
// Create Sql VM
ISqlVirtualMachinesOperations sqlOperations = context.getSqlClient().SqlVirtualMachines;
SqlVirtualMachineModel sqlVM = SqlVirtualMachineTestBase.CreateSqlVirtualMachine(context);
Assert.NotNull(sqlVM);
// List by group
IPage <SqlVirtualMachineModel> recovered = sqlOperations.ListByResourceGroup(context.resourceGroup.Name);
var iter = recovered.GetEnumerator();
Assert.NotNull(iter);
iter.MoveNext();
SqlVirtualMachineModel sqlVM2 = iter.Current;
Assert.NotNull(sqlVM2);
SqlVirtualMachineTestBase.ValidateSqlVirtualMachine(sqlVM, sqlVM2);
Assert.False(iter.MoveNext());
iter.Dispose();
}
}
public static SqlVirtualMachineModel CreateSqlVirtualMachine(SqlVirtualMachineTestContext context, SqlVirtualMachineGroup group = null, VirtualMachine virtualMachine = null)
{
ISqlVirtualMachinesOperations sqlOperations = context.getSqlClient().SqlVirtualMachines;
VirtualMachine vm = virtualMachine == null?VirtualMachineTestBase.CreateVM(context) : virtualMachine;
SqlVirtualMachineModel sqlVM = sqlOperations.CreateOrUpdate(context.resourceGroup.Name, vm.Name, new SqlVirtualMachineModel()
{
Location = context.location,
VirtualMachineResourceId = vm.Id,
SqlServerLicenseType = SqlServerLicenseType.PAYG,
SqlManagement = SqlManagementMode.Full,
SqlImageSku = Constants.imageSku,
SqlImageOffer = Constants.imageOffer,
SqlVirtualMachineGroupResourceId = (group != null) ? group.Id : null,
ServerConfigurationsManagementSettings = new ServerConfigurationsManagementSettings()
{
SqlConnectivityUpdateSettings = new SqlConnectivityUpdateSettings()
{
SqlAuthUpdateUserName = Constants.sqlLogin,
SqlAuthUpdatePassword = Constants.adminPassword,
ConnectivityType = "Private",
Port = 1433
},
SqlStorageUpdateSettings = new SqlStorageUpdateSettings()
{
DiskCount = 1,
DiskConfigurationType = "NEW",
StartingDeviceId = 2
},
SqlWorkloadTypeUpdateSettings = new SqlWorkloadTypeUpdateSettings()
{
SqlWorkloadType = "OLTP"
}
},
AutoPatchingSettings = new AutoPatchingSettings(false)
});
sqlVM.Validate();
return(sqlVM);
}
/// <summary>
/// Creates or updates a SQL virtual machine.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='resourceGroupName'>
/// Name of the resource group that contains the resource. You can obtain this
/// value from the Azure Resource Manager API or the portal.
/// </param>
/// <param name='sqlVirtualMachineName'>
/// Name of the SQL virtual machine.
/// </param>
/// <param name='parameters'>
/// The SQL virtual machine.
/// </param>
/// <param name='cancellationToken'>
/// The cancellation token.
/// </param>
public static async Task <SqlVirtualMachineModel> BeginCreateOrUpdateAsync(this ISqlVirtualMachinesOperations operations, string resourceGroupName, string sqlVirtualMachineName, SqlVirtualMachineModel parameters, CancellationToken cancellationToken = default(CancellationToken))
{
using (var _result = await operations.BeginCreateOrUpdateWithHttpMessagesAsync(resourceGroupName, sqlVirtualMachineName, parameters, null, cancellationToken).ConfigureAwait(false))
{
return(_result.Body);
}
}
/// <summary>
/// Creates or updates a SQL virtual machine.
/// </summary>
/// <param name='operations'>
/// The operations group for this extension method.
/// </param>
/// <param name='resourceGroupName'>
/// Name of the resource group that contains the resource. You can obtain this
/// value from the Azure Resource Manager API or the portal.
/// </param>
/// <param name='sqlVirtualMachineName'>
/// Name of the SQL virtual machine.
/// </param>
/// <param name='parameters'>
/// The SQL virtual machine.
/// </param>
public static SqlVirtualMachineModel BeginCreateOrUpdate(this ISqlVirtualMachinesOperations operations, string resourceGroupName, string sqlVirtualMachineName, SqlVirtualMachineModel parameters)
{
return(operations.BeginCreateOrUpdateAsync(resourceGroupName, sqlVirtualMachineName, parameters).GetAwaiter().GetResult());
}
/// <summary>
/// Creates or updates a Sql Virtual Machine
/// </summary>
public SqlVirtualMachineModel CreateOrUpdate(string resourceGroupName, string sqlVirtualMachineName, SqlVirtualMachineModel parameters)
{
return(GetCurrentSqlClient().SqlVirtualMachines.CreateOrUpdate(resourceGroupName, sqlVirtualMachineName, parameters));
}
public static bool ValidateTags(SqlVirtualMachineModel sqlVM1, SqlVirtualMachineModel sqlVM2)
{
return(ValidateTags(sqlVM1.Tags, sqlVM2.Tags));
}
public static AvailabilityGroupListener CreateAGListener(SqlVirtualMachineTestContext context, string agListenerName, string groupName)
{
MockClient client = context.client;
string domainName = Constants.domainName;
string adminLogin = Constants.adminLogin;
string adminPassword = Constants.adminPassword;
// Create domain
NetworkSecurityGroup nsg = CreateNsg(context);
VirtualNetwork vnet = CreateVirtualNetwork(context, networkSecurityGroup: nsg);
NetworkInterface nic = CreateNetworkInterface(context, virtualNetwork: vnet, networkSecurityGroup: nsg);
VirtualMachine vm = CreateVM(context, nic: nic);
VirtualMachineExtension domain = CreateDomain(context, vm, domainName, adminLogin, adminPassword);
Assert.NotNull(domain);
// Update DNS
Subnet subnet = vnet.Subnets[0];
UpdateVnetDNS(context, vnet, nsg, nic, subnet);
// Create SqlVirtualMachineGroup
StorageAccount storageAccount = CreateStorageAccount(context);
StorageAccountListKeysResult storageAccountKeys = client.storageClient.StorageAccounts.ListKeys(context.resourceGroup.Name, storageAccount.Name);
IEnumerator <StorageAccountKey> iter = storageAccountKeys.Keys.GetEnumerator();
iter.MoveNext();
string key = iter.Current.Value;
WsfcDomainProfile profile = new WsfcDomainProfile()
{
ClusterBootstrapAccount = getUsername(adminLogin, domainName),
ClusterOperatorAccount = getUsername(adminLogin, domainName),
SqlServiceAccount = getUsername(Constants.sqlService, domainName),
StorageAccountUrl = "https://" + storageAccount.Name + ".blob.core.windows.net/",
StorageAccountPrimaryKey = key,
DomainFqdn = domainName + ".com",
OuPath = ""
};
SqlVirtualMachineGroup group = CreateSqlVirtualMachineGroup(context, storageAccount, groupName: groupName, profile: profile);
// Create availability set
AvailabilitySet availabilitySet = client.computeClient.AvailabilitySets.CreateOrUpdate(context.resourceGroup.Name, context.generateResourceName(), new AvailabilitySet()
{
Location = context.location,
PlatformFaultDomainCount = 3,
PlatformUpdateDomainCount = 2,
Sku = new Microsoft.Azure.Management.Compute.Models.Sku
{
Name = "Aligned"
}
});
// Create two sql virtual machines
NetworkInterface nic1 = CreateNetworkInterface(context, virtualNetwork: vnet);
NetworkInterface nic2 = CreateNetworkInterface(context, virtualNetwork: vnet);
VirtualMachine vm1 = CreateVM(context, nic: nic1, availabilitySet: availabilitySet);
VirtualMachine vm2 = CreateVM(context, nic: nic2, availabilitySet: availabilitySet);
SqlVirtualMachineModel sqlVM1 = prepareMachine(context, group, vm1, domainName, adminLogin, adminPassword);
SqlVirtualMachineModel sqlVM2 = prepareMachine(context, group, vm2, domainName, adminLogin, adminPassword);
// Create load balancer
LoadBalancer loadBalancer = client.networkClient.LoadBalancers.CreateOrUpdate(context.resourceGroup.Name, context.generateResourceName(), new LoadBalancer()
{
Location = context.location,
Sku = new LoadBalancerSku("Basic"),
FrontendIPConfigurations = new List <FrontendIPConfiguration> (new FrontendIPConfiguration[]
{
new FrontendIPConfiguration()
{
Name = "LoadBalancerFrontEnd",
PrivateIPAllocationMethod = "Dynamic",
Subnet = subnet
}
})
});
// Run deployment to create an availability group with the two machines created
string AgName = "AvGroup";
VirtualMachineExtension availabilityGroup = context.client.computeClient.VirtualMachineExtensions.CreateOrUpdate(context.resourceGroup.Name, vm1.Name, "agCreation", new VirtualMachineExtension(context.location, name: AgName)
{
VirtualMachineExtensionType = "CustomScriptExtension",
Publisher = "Microsoft.Compute",
TypeHandlerVersion = "1.9",
AutoUpgradeMinorVersion = true,
Settings = new CustomScriptExtensionSettings
{
FileUris = new List <string>(new string[] {
"https://agtemplatestorage.blob.core.windows.net/templates/Deploy.ps1",
"https://agtemplatestorage.blob.core.windows.net/templates/sqlvm6.sql",
"https://agtemplatestorage.blob.core.windows.net/templates/sqlvm7.sql",
"https://agtemplatestorage.blob.core.windows.net/templates/sqlvm8.sql",
"https://agtemplatestorage.blob.core.windows.net/test/sqlvm9.sql",
"https://agtemplatestorage.blob.core.windows.net/templates/sqlvm10.sql"
}),
CommandToExecute = "powershell -ExecutionPolicy Unrestricted -File Deploy.ps1 " + AgName + " " + vm1.Name + " " + vm2.Name + " " + Constants.sqlLogin + " " + adminPassword + " " + domainName + "\\" + Constants.sqlService,
ContentVersion = "1.0.0.0"
}
});
// Create availability group listener
return(context.getSqlClient().AvailabilityGroupListeners.CreateOrUpdate(context.resourceGroup.Name, group.Name, agListenerName, new AvailabilityGroupListener()
{
LoadBalancerConfigurations = new List <LoadBalancerConfiguration>(new LoadBalancerConfiguration[]
{
new LoadBalancerConfiguration()
{
PrivateIpAddress = new PrivateIPAddress(ipAddress: "10.0.0.11", subnetResourceId: subnet.Id),
ProbePort = 59999,
LoadBalancerResourceId = loadBalancer.Id,
SqlVirtualMachineInstances = new List <string>()
{
sqlVM1.Id, sqlVM2.Id
}
}
}),
AvailabilityGroupName = AgName,
Port = 1433
}));
}
