Fix Threat Detection tests for release .SDK v5 (#1)

* FIX TESTS WITH SDK VERSION 5 * small fixes * records of threat detection tests
viparek · May 7, 2021 · d653a7d · d653a7d
1 parent e0353eb
commit d653a7d
Show file tree

Hide file tree

Showing 12 changed files with 3,899 additions and 2,889 deletions.
diff --git a/...ment/Microsoft.Azure.Management.Sql/tests/DatabaseVulnerabilityAssessmentScenarioTests.cs b/...ment/Microsoft.Azure.Management.Sql/tests/DatabaseVulnerabilityAssessmentScenarioTests.cs
@@ -43,10 +43,10 @@ public async void TestCreateUpdateGetDatabaseVulnerabilityAssessments()
                 // Turn ON database threat detection as a prerequisite to use VA
                 DatabaseSecurityAlertPolicy updatedDatabasePolicy = new DatabaseSecurityAlertPolicy
                 {
-                    State = SecurityAlertPolicyState.Enabled,
-                    EmailAccountAdmins = SecurityAlertPolicyEmailAccountAdmins.Enabled
+                    State = SecurityAlertsPolicyState.Enabled,
+                    EmailAccountAdmins = true
                 };
-                sqlClient.DatabaseThreatDetectionPolicies.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, updatedDatabasePolicy);
+                sqlClient.DatabaseSecurityAlertPolicies.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, updatedDatabasePolicy);
 
                 // Verify Policy is empty to begin with
                 DatabaseVulnerabilityAssessment policyThatWasReceived = sqlClient.DatabaseVulnerabilityAssessments.Get(resourceGroup.Name, server.Name, dbName);
@@ -98,10 +98,10 @@ public async void TestCreateUpdateGetDeleteDatabaseVulnerabilityAssessmentBaseli
                 // Turn ON database threat detection as a prerequisite to use VA
                 DatabaseSecurityAlertPolicy updatedDatabasePolicy = new DatabaseSecurityAlertPolicy
                 {
-                    State = SecurityAlertPolicyState.Enabled,
-                    EmailAccountAdmins = SecurityAlertPolicyEmailAccountAdmins.Enabled
+                    State = SecurityAlertsPolicyState.Enabled,
+                    EmailAccountAdmins = true
                 };
-                sqlClient.DatabaseThreatDetectionPolicies.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, updatedDatabasePolicy);
+                sqlClient.DatabaseSecurityAlertPolicies .CreateOrUpdate(resourceGroup.Name, server.Name, dbName, updatedDatabasePolicy);
 
                 // Set policy
                 await SetPolicy(context, sqlClient, resourceGroup, server, dbName);
@@ -140,10 +140,10 @@ public async void TestExecuteGetListDatabaseVulnerabilityAssessmentScans()
                 // Turn ON database threat detection as a prerequisite to use VA
                 DatabaseSecurityAlertPolicy updatedDatabasePolicy = new DatabaseSecurityAlertPolicy
                 {
-                    State = SecurityAlertPolicyState.Enabled,
-                    EmailAccountAdmins = SecurityAlertPolicyEmailAccountAdmins.Enabled
+                    State = SecurityAlertsPolicyState.Enabled,
+                    EmailAccountAdmins = true
                 };
-                sqlClient.DatabaseThreatDetectionPolicies.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, updatedDatabasePolicy);
+                sqlClient.DatabaseSecurityAlertPolicies.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, updatedDatabasePolicy);
 
                 // Set policy
                 await SetPolicy(context, sqlClient, resourceGroup, server, dbName);
@@ -200,10 +200,10 @@ public async void TestExportDatabaseVulnerabilityAssessmentScans()
                 // Turn ON database threat detection as a prerequisite to use VA
                 DatabaseSecurityAlertPolicy updatedDatabasePolicy = new DatabaseSecurityAlertPolicy
                 {
-                    State = SecurityAlertPolicyState.Enabled,
-                    EmailAccountAdmins = SecurityAlertPolicyEmailAccountAdmins.Enabled
+                    State = SecurityAlertsPolicyState.Enabled,
+                    EmailAccountAdmins = true
                 };
-                sqlClient.DatabaseThreatDetectionPolicies.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, updatedDatabasePolicy);
+                sqlClient.DatabaseSecurityAlertPolicies.CreateOrUpdate(resourceGroup.Name, server.Name, dbName, updatedDatabasePolicy);
 
                 // Set policy
                 await SetPolicy(context, sqlClient, resourceGroup, server, dbName);

diff --git a/...crosoft.Azure.Management.Sql/tests/ManagedDatabaseVulnerabilityAssessmentScenarioTests.cs b/...crosoft.Azure.Management.Sql/tests/ManagedDatabaseVulnerabilityAssessmentScenarioTests.cs
@@ -170,7 +170,7 @@ private async Task<DatabaseVulnerabilityAssessment> SetPolicy(SqlManagementTestC
 			// Modify the policy properties, send and receive and see it its still ok
 			ManagedServerSecurityAlertPolicy updatedManagedServerPolicy = new ManagedServerSecurityAlertPolicy
 			{
-				State = SecurityAlertPolicyState.Enabled,
+				State = SecurityAlertsPolicyState.Enabled,
 				EmailAccountAdmins = true
 			};
 

diff --git a/...ement/Microsoft.Azure.Management.Sql/tests/ManagedInstanceThreatDetectionScenarioTests.cs b/...ement/Microsoft.Azure.Management.Sql/tests/ManagedInstanceThreatDetectionScenarioTests.cs
@@ -52,7 +52,7 @@ public void TestManagedInstanceThreatDetection()
             // Modify the policy properties, send and receive and see it its still ok
             ManagedServerSecurityAlertPolicy updatedManagedServerPolicy = new ManagedServerSecurityAlertPolicy
             {
-                State = SecurityAlertPolicyState.Enabled,
+                State = SecurityAlertsPolicyState.Enabled,
                 EmailAccountAdmins = true
             };
 
@@ -69,7 +69,7 @@ public void TestManagedInstanceThreatDetection()
             // Modify the policy properties again, send and receive and see it its still ok
             updatedManagedServerPolicy = new ManagedServerSecurityAlertPolicy
             {
-                State = SecurityAlertPolicyState.Disabled,
+                State = SecurityAlertsPolicyState.Disabled,
                 EmailAccountAdmins = true,
                 EmailAddresses = new List<string>() { "testSecurityAlert@microsoft.com", "testServerPolicy@microsoft.com" },
                 DisabledAlerts = new List<string>() { "Sql_Injection" },
@@ -139,7 +139,7 @@ private ManagedServerSecurityAlertPolicy GetDefaultManagedInstanceSecurityAlertP
         {
             ManagedServerSecurityAlertPolicy properties = new ManagedServerSecurityAlertPolicy
             {
-                State = SecurityAlertPolicyState.Disabled,
+                State = SecurityAlertsPolicyState.Disabled,
                 EmailAccountAdmins = false,
                 DisabledAlerts = new List<string>() { string.Empty },
                 EmailAddresses = new List<string>() { string.Empty },

diff --git a/...crosoft.Azure.Management.Sql/tests/ManagedInstanceVulnerabilityAssessmentScenarioTests.cs b/...crosoft.Azure.Management.Sql/tests/ManagedInstanceVulnerabilityAssessmentScenarioTests.cs
@@ -58,7 +58,7 @@ private async Task<ManagedInstanceVulnerabilityAssessment> SetPolicy(SqlManageme
 			// Modify the policy properties, send and receive and see it its still ok
 			ManagedServerSecurityAlertPolicy updatedManagedServerPolicy = new ManagedServerSecurityAlertPolicy
 			{
-				State = SecurityAlertPolicyState.Enabled,
+				State = SecurityAlertsPolicyState.Enabled,
 				EmailAccountAdmins = true
 			};
 

diff --git a/...gement/Microsoft.Azure.Management.Sql/tests/ServerVulnerabilityAssessmentScenarioTests.cs b/...gement/Microsoft.Azure.Management.Sql/tests/ServerVulnerabilityAssessmentScenarioTests.cs
@@ -26,7 +26,7 @@ public async void TestCreateUpdateGetServerVulnerabilityAssessments()
                 // Turn ON Server ATP(threat detection) as a prerequisite to use VA
                 ServerSecurityAlertPolicy updatedServerPolicy = new ServerSecurityAlertPolicy
                 {
-                    State = SecurityAlertPolicyState.Enabled,
+                    State = SecurityAlertsPolicyState.Enabled,
                     EmailAccountAdmins = true
                 };