From 09726f3a93bb925144d4c1ef8f5f35ef6e849b67 Mon Sep 17 00:00:00 2001
From: ArcturusZhang <dapzhang@microsoft.com>
Date: Wed, 24 Mar 2021 09:15:44 +0800
Subject: [PATCH] fixing test cases

---
 ...ows_virtual_machine_scale_set_extensions_test.go | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/azurerm/internal/services/compute/windows_virtual_machine_scale_set_extensions_test.go b/azurerm/internal/services/compute/windows_virtual_machine_scale_set_extensions_test.go
index 605cf338bfb2..200d1a6155d3 100644
--- a/azurerm/internal/services/compute/windows_virtual_machine_scale_set_extensions_test.go
+++ b/azurerm/internal/services/compute/windows_virtual_machine_scale_set_extensions_test.go
@@ -183,7 +183,11 @@ func TestAccWindowsVirtualMachineScaleSet_extensionsAutomaticUpgradeWithHealthEx
 				check.That(data.ResourceName).ExistsInAzure(r),
 			),
 		},
-		data.ImportStep("admin_password", "extension.0.protected_settings"),
+		data.ImportStep(
+			"admin_password",
+			"extension.0.protected_settings",
+			"enable_automatic_updates",
+		),
 	})
 }
 
@@ -695,15 +699,16 @@ resource "azurerm_windows_virtual_machine_scale_set" "test" {
   admin_username      = "adminuser"
   admin_password      = "P@ssword1234!"
   upgrade_mode        = "Automatic"
+
   automatic_os_upgrade_policy {
     disable_automatic_rollback  = true
     enable_automatic_os_upgrade = true
   }
 
   rolling_upgrade_policy {
-    max_batch_instance_percent              = 100
-    max_unhealthy_instance_percent          = 100
-    max_unhealthy_upgraded_instance_percent = 100
+    max_batch_instance_percent              = 21
+    max_unhealthy_instance_percent          = 22
+    max_unhealthy_upgraded_instance_percent = 23
     pause_time_between_batches              = "PT30S"
   }