Add support for custom driver to be installed for performance test.

PiperOrigin-RevId: 853400478

Author: p3rf Team

perfkitbenchmarker/data/relational_db_configs/sqlserver_ha_configs/check_sql_server_status.ps1

@@ -1,3 +1,6 @@
+param (
+    [string]$skip_cluster_check = 'False'
+ )
 
 function Get-SqlServiceStatus {
   $sqlServerService = Get-Service -Name 'MSSQLSERVER' -ErrorAction SilentlyContinue
@@ -54,7 +57,12 @@ try {
   for (($i = 0); $i -lt 24; $i++) {
 
     $serviceRunning = Get-SqlServiceStatus
-    $isClustered = Get-ClusterStatus
+    if ($skip_cluster_check -eq 'True') {
+      $isClustered = $true
+    }
+    else {
+      $isClustered = Get-ClusterStatus
+    }
     $canConnect = Get-SqlServerConnectionState -HostName $localServerName
 
     if ($serviceRunning -and $isClustered -and $canConnect) {
@@ -80,5 +88,3 @@ catch {
   Write-Host $_.Exception.Message
   throw $_.Exception.Message
 }
-
-

perfkitbenchmarker/data/relational_db_configs/sqlserver_ha_configs/set_dns_join_domain.ps1

@@ -6,6 +6,7 @@ param (
 
 $domainUser = "$perf_domain\pkbadminuser"
 $domainName  = "$perf_domain.local"
+$connectionVerified = $false
 
 $passwordSecureString = (ConvertTo-SecureString -AsPlainText $perf_password -Force)
 $nic = Get-NetIPAddress -AddressFamily IPv4 -InterfaceAlias 'E*'
@@ -17,10 +18,49 @@ Set-DnsClientServerAddress -InterfaceIndex $nic[0].ifIndex -ServerAddresses $dns
 
 $dns_server = Get-DnsClientServerAddress -InterfaceIndex  $nic[0].ifIndex -AddressFamily IPv4
 
+if (Test-Connection -ComputerName $DomainName -Count 1 -Quiet) {
+    Write-Host 'Network connectivity to $DomainName is successful.'
+}
+else {
+    Write-Host 'Cannot reach $DomainName via ICMP. Check DNS settings and network connectivity.'
+}
+
+for ($i=0; $i -lt 10; $i++) {
+  try {
+    $test_connection = Test-NetConnection -ComputerName $DomainName -Port 135
+    if ($test_connection.TcpTestSucceeded) {
+      Write-Host 'Network connectivity to $DomainName on port 135 is successful.'
+      $connectionVerified = $true
+      break
+    }
+    else {
+      Write-Host 'Cannot reach $DomainName on port 135.'
+      Start-Sleep -Seconds 30
+    }
+  }
+  catch {
+    Start-Sleep -Seconds 30
+  }
+}
+
+if (-not $connectionVerified) {
+  throw 'Cannot reach $DomainName.'
+}
+
 if ($dns_server.ServerAddresses -eq $dns_server_ip) {
-  Write-Host "DNS Set. Joining $domainName  Domain"
-  $credObject = New-Object System.Management.Automation.PSCredential ($domainUser, $passwordSecureString)
+  for ($i=0; $i -lt 3; $i++) {
+    try {
+      Write-Host "DNS Set. Joining $domainName  Domain"
+      $credObject = New-Object System.Management.Automation.PSCredential ($domainUser, $passwordSecureString)
 
-  Add-Computer -DomainName $domainName  -credential $credObject
-  Write-Host 'Done'
+      Add-Computer -DomainName $domainName  -credential $credObject -ErrorAction Stop -PassThru
+      Write-Host 'Done'
+      break
+    }
+    catch {
+      Write-Host 'Domain join test failed. Review the error details below:'
+      Write-Error $_.Exception.Message
+      Start-Sleep -Seconds 30
+    }
+  }
 }