update and record the test

Azure · Nov 11, 2024 · c958537 · c958537
1 parent 19be19a
commit c958537
Show file tree

Hide file tree

Showing 7 changed files with 1,525 additions and 6 deletions.
diff --git a/src/DevTestLabs/DevTestLabs.Test/ScenarioTests/PolicyTests.cs b/src/DevTestLabs/DevTestLabs.Test/ScenarioTests/PolicyTests.cs
@@ -59,5 +59,40 @@ public void TestAzureRmDtlAutoStartPolicy()
         {
             TestRunner.RunTestScript("Test-AzureRmDtlAutoStartPolicy");
         }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAzDtlAutoStartPolicy()
+        {
+            TestRunner.RunTestScript("Test-AzDtlAutoStartPolicy");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAzDtlAutoShutdownPolicy()
+        {
+            TestRunner.RunTestScript("Test-AzDtlAutoShutdownPolicy");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAzDtlAllowedVMSizesPolicy()
+        {
+            TestRunner.RunTestScript("Test-AzDtlAllowedVMSizesPolicy");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAzDtlVMsPerUserPolicy()
+        {
+            TestRunner.RunTestScript("Test-AzDtlVMsPerUserPolicy");
+        }
+
+        [Fact]
+        [Trait(Category.AcceptanceType, Category.CheckIn)]
+        public void TestAzDtlVMsPerLabPolicy()
+        {
+            TestRunner.RunTestScript("Test-AzDtlVMsPerLabPolicy");
+        }
     }
 }
diff --git a/src/DevTestLabs/DevTestLabs.Test/ScenarioTests/PolicyTests.ps1 b/src/DevTestLabs/DevTestLabs.Test/ScenarioTests/PolicyTests.ps1
@@ -16,7 +16,7 @@
 .SYNOPSIS
 Tests AzureRmDtlVMsPerLabPolicy
 #>
-function Test-AzDtlVMsPerLabPolicy
+function Test-AzureRmDtlVMsPerLabPolicy
 {
     $rgName = Get-ResourceGroupName
     $labName = Get-ResourceName "onesdk"
@@ -44,7 +44,7 @@ function Test-AzDtlVMsPerLabPolicy
 .SYNOPSIS
 Tests AzureRmDtlVMsPerUserPolicy
 #>
-function Test-AzDtlVMsPerUserPolicy
+function Test-Test-AzureRmDtlVMsPerUserPolicy
 {
     $rgName = Get-ResourceGroupName
     $labName = Get-ResourceName "onesdk"
@@ -70,7 +70,7 @@ function Test-AzDtlVMsPerUserPolicy
 .SYNOPSIS
 Tests AzureRmDtlAllowedVMSizesPolicy
 #>
-function Test-AzDtlAllowedVMSizesPolicy
+function Test-AzureRmDtlAllowedVMSizesPolicy
 {
     $rgName = Get-ResourceGroupName
     $labName = Get-ResourceName "onesdk"
@@ -96,14 +96,14 @@ function Test-AzDtlAllowedVMSizesPolicy
 .SYNOPSIS
 Tests AzureRmDtlAllowedVMSizesPolicy
 #>
-function Test-AzDtlAutoShutdownPolicy
+function Test-AzureRmDtlAutoShutdownPolicy
 {
     $rgName = Get-ResourceGroupName
     $labName = Get-ResourceName "onesdk"
 
     try
     {
-        $policy = Set-AzDtlAutoShutdownPolicy -Time "13:30:00" -LabName $labName -ResourceGroupName $rgName
+        $policy = Set-AzDtlAutoShutdownPolicy -Time "13:30:00" -LabName $labName -ResourceGroupName $rgName -Enable
         $readBack = Get-AzDtlAutoShutdownPolicy -LabName $labName -ResourceGroupName $rgName
 
         Assert-AreEqual "Enabled" $policy.Status
@@ -122,7 +122,7 @@ function Test-AzDtlAutoShutdownPolicy
 .SYNOPSIS
 Tests AzureRmDtlAutoStartPolicy
 #>
-function Test-AzDtlAutoStartPolicy
+function Test-AzureRmDtlAutoStartPolicy
 {
     $rgName = Get-ResourceGroupName
     $labName = Get-ResourceName "onesdk"
@@ -154,3 +154,115 @@ function Test-AzDtlAutoStartPolicy
         Clean-ResourceGroup $rgName
     }
 }
+
+<#
+.SYNOPSIS
+Tests AzDtlAutoStartPolicy
+Setup: Create lab "onesdklab0111" in group 'DTL-TestGroup'
+#>
+function Test-AzDtlAutoStartPolicy
+{
+    $rgName = 'DTL-TestGroup'
+    $labName = "onesdklab0111"
+
+    $policy = Set-AzDtlAutoStartPolicy -Time "13:30:00" -LabName $labName -ResourceGroupName $rgName -Enable
+    $readBack = Get-AzDtlAutoStartPolicy -LabName $labName -ResourceGroupName $rgName
+
+    Assert-AreEqual "Enabled" $policy.Status
+    Assert-AreEqual "1330" $policy.WeeklyRecurrence.Time
+
+    Assert-AreEqual "Enabled" $readBack.Status
+    Assert-AreEqual "1330" $readBack.WeeklyRecurrence.Time
+
+    $policy = Set-AzDtlAutoStartPolicy -Time "13:30:00" -LabName $labName -ResourceGroupName $rgName -Days Monday, Tuesday
+    $readBack = Get-AzDtlAutoStartPolicy -LabName $labName -ResourceGroupName $rgName
+
+    Assert-AreEqual "Enabled" $policy.Status
+    Assert-AreEqual "1330" $policy.WeeklyRecurrence.Time
+    Assert-AreEqualArray ([System.DayOfWeek]::Monday, [System.DayOfWeek]::Tuesday) $policy.WeeklyRecurrence.Weekdays
+
+    Assert-AreEqual "Enabled" $readBack.Status
+    Assert-AreEqual "1330" $readBack.WeeklyRecurrence.Time
+    Assert-AreEqualArray ([System.DayOfWeek]::Monday, [System.DayOfWeek]::Tuesday) $readBack.WeeklyRecurrence.Weekdays
+}
+
+<#
+.SYNOPSIS
+Tests AzDtlAutoShutdownPolicy
+Setup: Create lab "onesdklab0111" in group 'DTL-TestGroup'
+#>
+function Test-AzDtlAutoShutdownPolicy
+{
+    $rgName = 'DTL-TestGroup'
+    $labName = "onesdklab0111"
+
+    $policy = Set-AzDtlAutoShutdownPolicy -Time "13:30:00" -LabName $labName -ResourceGroupName $rgName -Enable
+    $readBack = Get-AzDtlAutoShutdownPolicy -LabName $labName -ResourceGroupName $rgName
+
+    Assert-AreEqual "Enabled" $policy.Status
+    Assert-AreEqual "1330" $policy.DailyRecurrence.Time
+
+    Assert-AreEqual "Enabled" $readBack.Status
+    Assert-AreEqual "1330" $readBack.DailyRecurrence.Time
+}
+
+<#
+.SYNOPSIS
+Tests AzDtlAllowedVMSizesPolicy
+Setup: Create lab "onesdklab0111" in group 'DTL-TestGroup'
+#>
+function Test-AzDtlAllowedVMSizesPolicy
+{
+    $rgName = 'DTL-TestGroup'
+    $labName = "onesdklab0111"
+
+    $policy = Set-AzDtlAllowedVMSizesPolicy -Enable -LabName $labName -ResourceGroupName $rgName -VmSizes Standard_A3, Standard_A0
+    $readBack = Get-AzDtlAllowedVMSizesPolicy -LabName $labName -ResourceGroupName $rgName
+
+    Assert-AreEqual "Enabled" $policy.Status
+    Assert-AreEqual '["Standard_A3","Standard_A0"]' $policy.Threshold
+
+    Assert-AreEqual "Enabled" $readBack.Status
+    Assert-AreEqual '["Standard_A3","Standard_A0"]' $readBack.Threshold
+}
+
+<#
+.SYNOPSIS
+Tests AzDtlVMsPerUserPolicy
+Setup: Create lab "onesdklab0111" in group 'DTL-TestGroup'
+#>
+function Test-AzDtlVMsPerUserPolicy
+{
+    $rgName = 'DTL-TestGroup'
+    $labName = "onesdklab0111"
+
+    $policy = Set-AzDtlVMsPerUserPolicy -MaxVMs 5 -LabName $labName -ResourceGroupName $rgName
+    $readBack = Get-AzDtlVMsPerUserPolicy -LabName $labName -ResourceGroupName $rgName
+
+    Assert-AreEqual "Enabled" $policy.Status
+    Assert-AreEqual "5" $policy.Threshold
+
+    Assert-AreEqual "Enabled" $readBack.Status
+    Assert-AreEqual "5" $readBack.Threshold
+}
+
+<#
+.SYNOPSIS
+Tests AzDtlVMsPerLabPolicy
+Setup: Create lab "onesdklab0111" in group 'DTL-TestGroup'
+#>
+function Test-AzDtlVMsPerLabPolicy
+{
+    $rgName = 'DTL-TestGroup'
+    $labName = "onesdklab0111"
+
+    # Max VMs per lab policy
+    $policy = Set-AzDtlVMsPerLabPolicy -MaxVMs 5 -LabName $labName -ResourceGroupName $rgName
+    $readBack = Get-AzDtlVMsPerLabPolicy -LabName $labName -ResourceGroupName $rgName
+
+    Assert-AreEqual "Enabled" $policy.Status
+    Assert-AreEqual "5" $policy.Threshold
+
+    Assert-AreEqual "Enabled" $readBack.Status
+    Assert-AreEqual "5" $readBack.Threshold
+}