Test for Delete Security Group added

cloudbridge/cloud/providers/azure/mock_azure_client.py

@@ -30,6 +30,13 @@ class MockAzureClient:
     def list_security_group(self):
         return self.security_groups
 
+    def delete_security_group(self, name):
+        for item in self.security_groups:
+            if item.name == name:
+                self.security_groups.remove(item)
+                return True
+        return False
+
     def get_resource_group(self, resource_group_name):
         return self.rg
 
@@ -48,3 +55,5 @@ class MockAzureClient:
         new_container = Container()
         new_container.name = "newContainerCreate"
         return None
+
+

test/test_azure_security_service.py

@@ -20,15 +20,15 @@ class AzureSecurityServiceTestCase(ProviderTestBase):
         for group in sgl:
             print("List( " + "Name-" + group.name + "  Id-" + group.id + " Rules - " + " )")
         self.assertTrue(
-            len(sgl) == 3,
+            len(sgl) == 2,
             "Count should be 3")
 
     @helpers.skipIfNoService(['security.security_groups'])
     def test_azure_security_group_get_found(self):
-        sgl = self.provider.security.security_groups.get("sg2")
+        sgl = self.provider.security.security_groups.get("sg3")
         print("Get ( " + "Name - " + sgl.name + "  Id - " + sgl.id + " )")
         self.assertTrue(
-            sgl.name == "sec_group2",
+            sgl.name == "sec_group3",
             "SG name should be sec_group2")
 
     @helpers.skipIfNoService(['security.security_groups'])
@@ -38,3 +38,14 @@ class AzureSecurityServiceTestCase(ProviderTestBase):
         self.assertTrue(
             sgl == None,
             "Security group does not exist. Should return None.")
+
+    @helpers.skipIfNoService(['security.security_groups'])
+    def test_azure_security_group_delete_IdExists(self):
+        sg = self.provider.security.security_groups.delete("sg2")
+        print("Delete - ")
+        self.assertEqual(sg, True)
+
+    @helpers.skipIfNoService(['security.security_groups'])
+    def test_azure_security_group_delete_IdNotExist(self):
+        sg = self.provider.security.security_groups.delete("sg5")
+        self.assertEqual(sg, False)