azurerm_security_center_server_vulnerability_assessment and `azurer…

…m_security_center_server_vulnerability_assessment_virtual_machine` - adding pricing tier in test config Relates-to: hashicorp#14736
leesutcliffe · Nov 24, 2022 · 9843e2e · 9843e2e
1 parent 36c34e3
commit 9843e2e
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/.../services/securitycenter/security_center_server_vulnerability_assessment_resource_test.go b/.../services/securitycenter/security_center_server_vulnerability_assessment_resource_test.go
@@ -27,6 +27,10 @@ func TestAccServerVulnerabilityAssessment_basic(t *testing.T) {
 			),
 		},
 		data.ImportStep(),
+		{
+			// reset pricing to free
+			Config: SecurityCenterSubscriptionPricingResource{}.tier("Free", "VirtualMachines"),
+		},
 	})
 }
 
@@ -42,6 +46,10 @@ func TestAccServerVulnerabilityAssessment_requiresImport(t *testing.T) {
 			),
 		},
 		data.RequiresImportErrorStep(r.requiresImport),
+		{
+			// reset pricing to free
+			Config: SecurityCenterSubscriptionPricingResource{}.tier("Free", "VirtualMachines"),
+		},
 	})
 }
 
@@ -124,8 +132,15 @@ resource "azurerm_linux_virtual_machine" "test" {
   network_interface_ids = [azurerm_network_interface.test.id]
 }
 
+resource "azurerm_security_center_subscription_pricing" "test" {
+  tier          = "Standard"
+  resource_type = "VirtualMachines"
+  subplan       = "P2"
+}
+
 resource "azurerm_security_center_server_vulnerability_assessment" "test" {
   virtual_machine_id = azurerm_linux_virtual_machine.test.id
+  depends_on         = [azurerm_security_center_subscription_pricing.test]
 }
 `, data.RandomInteger, data.Locations.Primary)
 }

diff --git a/...tycenter/security_center_server_vulnerability_assessment_virtual_machine_resource_test.go b/...tycenter/security_center_server_vulnerability_assessment_virtual_machine_resource_test.go
@@ -27,6 +27,10 @@ func TestAccServerVulnerabilityAssessmentVirtualMachine_basic(t *testing.T) {
 			),
 		},
 		data.ImportStep(),
+		{
+			// reset pricing to free
+			Config: SecurityCenterSubscriptionPricingResource{}.tier("Free", "VirtualMachines"),
+		},
 	})
 }
 
@@ -42,6 +46,10 @@ func TestAccServerVulnerabilityAssessmentVirtualMachine_requiresImport(t *testin
 			),
 		},
 		data.RequiresImportErrorStep(r.requiresImport),
+		{
+			// reset pricing to free
+			Config: SecurityCenterSubscriptionPricingResource{}.tier("Free", "VirtualMachines"),
+		},
 	})
 }
 
@@ -121,8 +129,15 @@ resource "azurerm_linux_virtual_machine" "test" {
   network_interface_ids = [azurerm_network_interface.test.id]
 }
 
+resource "azurerm_security_center_subscription_pricing" "test" {
+  tier          = "Standard"
+  resource_type = "VirtualMachines"
+  subplan       = "P2"
+}
+
 resource "azurerm_security_center_server_vulnerability_assessment_virtual_machine" "test" {
   virtual_machine_id = azurerm_linux_virtual_machine.test.id
+  depends_on         = [azurerm_security_center_subscription_pricing.test]
 }
 `, data.RandomInteger, data.Locations.Primary)
 }