Skip to content

Commit 6da94da

Browse files
vidai-msftvidai-msft
authored
Improve total amount of time spent for CI test (Azure#19424)
1 parent f2ccb38 commit 6da94da

File tree

53 files changed

+207
-392
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

53 files changed

+207
-392
lines changed

‎src/Batch/Batch.Test/ScenarioTests/Common.ps1‎

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -12,18 +12,6 @@
1212
# limitations under the License.
1313
# ----------------------------------------------------------------------------------
1414

15-
<#
16-
.SYNOPSIS
17-
Sleeps but only during recording.
18-
#>
19-
function Start-TestSleep($milliseconds)
20-
{
21-
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -ne [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback)
22-
{
23-
Start-Sleep -Milliseconds $milliseconds
24-
}
25-
}
26-
2715
function Compute-TestTimeout($seconds)
2816
{
2917
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -eq [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback)

‎src/Batch/Batch.Test/ScenarioTests/ComputeNodeTests.ps1‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ function Test-RemoveComputeNodes
4343
{
4444
throw [System.TimeoutException] "Timed out waiting for compute nodes to enter LeavingPool state"
4545
}
46-
Start-TestSleep 1000
46+
Start-TestSleep -Seconds 1
4747
$computeNodes = Get-AzBatchComputeNode -PoolId $poolId -Select $select -BatchContext $context
4848
}
4949
}
@@ -140,7 +140,7 @@ function WaitForIdleComputeNode
140140
{
141141
throw [System.TimeoutException] "Timed out waiting for idle compute node"
142142
}
143-
Start-TestSleep 5000
143+
Start-TestSleep -Seconds 5
144144
$computeNode = Get-AzBatchComputeNode -Id $computeNodeId -PoolId $poolId -BatchContext $context -Select "id,state"
145145
}
146146
}

‎src/Batch/Batch.Test/ScenarioTests/JobTests.ps1‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ function Test-DisableEnableTerminateJob
7474
Disable-AzBatchJob $jobId Terminate -BatchContext $context
7575

7676
# Sleep a bit in Record mode since the job doesn't immediately switch to Disabled.
77-
Start-TestSleep 10000
77+
Start-TestSleep -Seconds 10
7878

7979
# Verify the job was Disabled
8080
$job = Get-AzBatchJob $jobId -BatchContext $context

‎src/Compute/Compute/Extension/VmssDiskEncryption/Scripts/AzureDiskEncryption_VMSS.ps1‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ $ErrorActionPreference = “Stop”;
6666
else
6767
{
6868
Write-Host "Sleeping 10 seconds to actiavate AzureDiskEncryption-VMSS feature . Retry count : ($i)";
69-
Start-Sleep -Seconds 10;
69+
Start-TestSleep -Seconds 10
7070
}
7171
}
7272
if(!$vmssDiskEncryptionFeature -or $vmssDiskEncryptionFeature.RegistrationState -ne 'Registered')

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/ManagedCassandraOperationsTests.ps1‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ function Test-ManagedCassandraCreateUpdateGetCmdlets
8585
{
8686
break
8787
}
88-
Start-Sleep -s 1
88+
Start-TestSleep -Seconds 1
8989
}
9090
$response = Get-AzManagedCassandraCluster -ResourceId $clusterId
9191
Assert-AreEqual 3 $response.Properties.ExternalSeedNodes.Count

‎src/CosmosDB/CosmosDB.Test/ScenarioTests/RestoreTests.ps1‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ function Test-UpdateCosmosDBAccountBackupPolicyCmdLet {
345345
}
346346

347347
$updatedCosmosDBAccount = Update-AzCosmosDBAccount -ResourceGroupName $rgName -Name $cosmosDBAccountName -BackupPolicyType Continuous
348-
Start-Sleep -s 50
348+
Start-TestSleep -Seconds 50
349349

350350
$updatedCosmosDBAccount = Get-AzCosmosDBAccount -ResourceGroupName $rgName -Name $cosmosDBAccountName
351351
Assert-NotNull $updatedCosmosDBAccount.BackupPolicy.BackupPolicyMigrationState

‎src/DataFactory/DataFactoryV2.Test/ScenarioTests/RunTests.ps1‎

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -54,27 +54,21 @@ function Test-Run
5454
Get-AzDataFactoryV2PipelineRun -DataFactory $df -PipelineRunId $Run
5555

5656
# Wait run to finish before rerun
57-
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -ne [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback) {
58-
Start-Sleep -s 120
59-
}
57+
Start-TestSleep -Seconds 120
6058

6159
# Trying rerun the pipeline run
6260
$Rerun = Invoke-AzDataFactoryV2Pipeline -ResourceGroupName $rgname -PipelineName $pipelineName -DataFactoryName $dfname -ReferencePipelineRunId $Run -IsRecovery
6361
Get-AzDataFactoryV2PipelineRun -DataFactory $df -PipelineRunId $Rerun
6462

6563
# Wait run to finish before rerun
66-
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -ne [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback) {
67-
Start-Sleep -s 120
68-
}
64+
Start-TestSleep -Seconds 120
6965

7066
# Trying rerun from activity MyCopyActivity_1_0
7167
$RerunFromActivity = Invoke-AzDataFactoryV2Pipeline -ResourceGroupName $rgname -PipelineName $pipelineName -DataFactoryName $dfname -ReferencePipelineRunId $Run -IsRecovery -StartActivityName MyCopyActivity_1_0
7268
Get-AzDataFactoryV2PipelineRun -DataFactory $df -PipelineRunId $RerunFromActivity
7369

7470
# Wait run to finish before rerun
75-
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -ne [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback) {
76-
Start-Sleep -s 120
77-
}
71+
Start-TestSleep -Seconds 120
7872

7973
# Trying rerun from failed activity
8074
$RerunFromFailedActivity = Invoke-AzDataFactoryV2Pipeline -ResourceGroupName $rgname -PipelineName $pipelineName -DataFactoryName $dfname -ReferencePipelineRunId $Run -IsRecovery -StartFromFailure

‎src/DataFactory/DataFactoryV2.Test/ScenarioTests/TriggerTests.ps1‎

Lines changed: 5 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -120,9 +120,7 @@ function Test-TriggerRun
120120

121121
Assert-AreEqual 'Started' $started.RuntimeState
122122

123-
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -ne [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback) {
124-
Start-Sleep -s 150
125-
}
123+
Start-TestSleep -Seconds 150
126124

127125
$endDate = $startDate.AddYears(1)
128126
$triggerRuns = Get-AzDataFactoryV2TriggerRun -ResourceGroupName $rgname -DataFactoryName $dfname -TriggerName $triggername -TriggerRunStartedAfter $startDate -TriggerRunStartedBefore $endDate
@@ -184,9 +182,7 @@ function Test-BlobEventTriggerSubscriptions
184182
Add-AzDataFactoryV2TriggerSubscription -ResourceGroupName $rgname -DataFactoryName $dfname -Name $triggername
185183
$status = Get-AzDataFactoryV2TriggerSubscriptionStatus -ResourceGroupName $rgname -DataFactoryName $dfname -Name $triggername
186184
while ($status.Status -ne "Enabled"){
187-
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -ne [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback) {
188-
Start-Sleep -s 150
189-
}
185+
Start-TestSleep -Seconds 150
190186
$status = Get-AzDataFactoryV2TriggerSubscriptionStatus -ResourceGroupName $rgname -DataFactoryName $dfname -Name $triggername
191187
}
192188

@@ -248,9 +244,7 @@ function Test-BlobEventTriggerSubscriptionsByInputObject
248244
Add-AzDataFactoryV2TriggerSubscription $actual
249245
$status = Get-AzDataFactoryV2TriggerSubscriptionStatus $actual
250246
while ($status.Status -ne "Enabled"){
251-
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -ne [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback) {
252-
Start-Sleep -s 150
253-
}
247+
Start-TestSleep -Seconds 150
254248
$status = Get-AzDataFactoryV2TriggerSubscriptionStatus $actual
255249
}
256250

@@ -312,9 +306,7 @@ function Test-BlobEventTriggerSubscriptionsByResourceId
312306
Add-AzDataFactoryV2TriggerSubscription -ResourceId $expected.Id
313307
$status = Get-AzDataFactoryV2TriggerSubscriptionStatus -ResourceId $expected.Id
314308
while ($status.Status -ne "Enabled"){
315-
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -ne [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback) {
316-
Start-Sleep -s 150
317-
}
309+
Start-TestSleep -Seconds 150
318310
$status = Get-AzDataFactoryV2TriggerSubscriptionStatus -ResourceId $expected.Id
319311
}
320312

@@ -375,9 +367,7 @@ function Test-TriggerInvokeAndStop
375367

376368
Assert-AreEqual 'Started' $started.RuntimeState
377369

378-
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -ne [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback) {
379-
Start-Sleep -s 150
380-
}
370+
Start-TestSleep -Seconds 150
381371

382372
$endDate = $startDate.AddYears(1)
383373
$triggerRuns = Get-AzDataFactoryV2TriggerRun -ResourceGroupName $rgname -DataFactoryName $dfname -TriggerName $triggername -TriggerRunStartedAfter $startDate -TriggerRunStartedBefore $endDate

‎src/DataMigration/DataMigration.Test/ScenarioTests/Common.ps1‎

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -181,15 +181,6 @@ function New-ProjectDbInfos
181181
return $dbInfo
182182
}
183183

184-
function SleepTask($value){
185-
if ([Microsoft.Azure.Test.HttpRecorder.HttpMockServer]::Mode -ne [Microsoft.Azure.Test.HttpRecorder.HttpRecorderMode]::Playback)
186-
{
187-
Start-Sleep -s $value
188-
}else{
189-
Start-Sleep -s 0
190-
}
191-
}
192-
193184
function New-TargetSqlMiSyncConnectionInfo
194185
{
195186
$miResourceId = [Microsoft.Azure.Commands.DataMigrationConfig]::GetConfigString("MI_RESOURCE_ID")

‎src/DataMigration/DataMigration.Test/ScenarioTests/ServiceTests.ps1‎

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ function Test-ConnectToSourceSqlServer
160160

161161
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
162162
{
163-
SleepTask 15
163+
Start-TestSleep -Seconds 15
164164
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
165165
}
166166

@@ -199,7 +199,7 @@ function Test-ConnectToTargetSqlDb
199199

200200
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
201201
{
202-
SleepTask 15
202+
Start-TestSleep -Seconds 15
203203
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
204204
}
205205

@@ -239,7 +239,7 @@ function Test-GetUserTableTask
239239

240240
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
241241
{
242-
SleepTask 15
242+
Start-TestSleep -Seconds 15
243243
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
244244
}
245245

@@ -302,7 +302,7 @@ function Test-MigrateSqlSqlDB
302302

303303
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
304304
{
305-
SleepTask 15
305+
Start-TestSleep -Seconds 15
306306
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
307307
}
308308

@@ -341,7 +341,7 @@ function Test-ConnectToTargetSqlDbMi
341341

342342
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
343343
{
344-
SleepTask 15
344+
Start-TestSleep -Seconds 15
345345
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
346346
}
347347

@@ -428,7 +428,7 @@ function Test-MigrateSqlSqlDBMi
428428

429429
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
430430
{
431-
SleepTask 15
431+
Start-TestSleep -Seconds 15
432432
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
433433
}
434434

@@ -501,7 +501,7 @@ function Test-ValidateMigrationInputSqlSqlDbMi
501501

502502
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
503503
{
504-
SleepTask 15
504+
Start-TestSleep -Seconds 15
505505
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
506506
}
507507

@@ -540,7 +540,7 @@ function Test-ConnectToSourceSqlServerSync
540540

541541
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
542542
{
543-
SleepTask 15
543+
Start-TestSleep -Seconds 15
544544
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
545545
}
546546

@@ -585,7 +585,7 @@ function Test-ConnectToTargetSqlDbSync
585585

586586
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
587587
{
588-
SleepTask 15
588+
Start-TestSleep -Seconds 15
589589
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
590590
}
591591

@@ -643,7 +643,7 @@ function Test-GetUserTableSyncTask
643643

644644
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
645645
{
646-
SleepTask 15
646+
Start-TestSleep -Seconds 15
647647
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
648648
}
649649

@@ -709,7 +709,7 @@ function Test-ValidateMigrationInputSqlSqlDbSync
709709

710710
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
711711
{
712-
SleepTask 15
712+
Start-TestSleep -Seconds 15
713713
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
714714
}
715715

@@ -792,7 +792,7 @@ function Test-MigrateSqlSqlDBSync
792792
}
793793
}
794794

795-
SleepTask 15
795+
Start-TestSleep -Seconds 15
796796
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
797797
}
798798

@@ -893,7 +893,7 @@ function Test-MigrateMongoDb
893893

894894
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
895895
{
896-
SleepTask 15
896+
Start-TestSleep -Seconds 15
897897
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
898898

899899
if($task.ProjectTask.Properties.State -eq "Running") {
@@ -934,7 +934,7 @@ function Test-ConnectToTargetSqlDbMiSync
934934

935935
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
936936
{
937-
SleepTask 15
937+
Start-TestSleep -Seconds 15
938938
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
939939
}
940940

@@ -1009,7 +1009,7 @@ function Test-ValidateMigrationInputSqlSqlDbMiSync
10091009

10101010
while(($task.ProjectTask.Properties.State -eq "Running") -or ($task.ProjectTask.Properties.State -eq "Queued"))
10111011
{
1012-
SleepTask 15
1012+
Start-TestSleep -Seconds 15
10131013
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
10141014
}
10151015

@@ -1107,7 +1107,7 @@ function Test-MigrateSqlSqlDbMiSync
11071107
}
11081108
}
11091109

1110-
SleepTask 15
1110+
Start-TestSleep -Seconds 15
11111111
$task = Get-AzDataMigrationTask -ResourceGroupName $rg.ResourceGroupName -ServiceName $service.Name -ProjectName $project.Name -TaskName $taskName -Expand
11121112
}
11131113

0 commit comments

Comments
 (0)