Tests for SG Create and SG rules create and delete added

integration_test/__init__.py

@@ -26,11 +26,13 @@ discovery. (The test generator will automatically add ``unittest.TestCase``
 as a base class to each combination).
 """
 import cloudbridge
-from integration_test.test_integration_azure_object_store_service import AzureIntegrationObjectStoreServiceTestCase
 from integration_test.helpers import ProviderTestCaseGenerator
+from integration_test.test_integration_azure_object_store_service import AzureIntegrationObjectStoreServiceTestCase
+from integration_test.test_integration_azure_security_group import AzureIntegrationSecurityServiceTestCase
 
 PROVIDER_TESTS ={'azure': [
-     AzureIntegrationObjectStoreServiceTestCase
+     AzureIntegrationObjectStoreServiceTestCase,
+    AzureIntegrationSecurityServiceTestCase
 ]}
 
 

integration_test/test_integration_azure_security_group.py

@@ -0,0 +1,53 @@
+import json
+import unittest
+import uuid
+
+from cloudbridge.cloud.interfaces import TestMockHelperMixin
+
+from test.helpers import ProviderTestBase
+import test.helpers as helpers
+
+class AzureIntegrationSecurityServiceTestCase(ProviderTestBase):
+    def __init__(self, methodName, provider):
+        super(AzureIntegrationSecurityServiceTestCase, self).__init__(
+            methodName=methodName, provider=provider)
+
+    @helpers.skipIfNoService(['security.security_groups'])
+    def test_azure_security_group(self):
+        sg = self.provider.security.security_groups.create(name="testCreateSecGroup", description="testCreateSecGroup", network_id="")
+        self.assertEqual("testCreateSecGroup", sg.name)
+
+        list = self.provider.security.security_groups.list()
+        self.assertEqual( len(list) , 2)
+
+        get = self.provider.security.security_groups.get("/subscriptions/7904d702-e01c-4826-8519-f5a25c866a96/resourceGroups/cloudbridge-azure/providers/Microsoft.Network/networkSecurityGroups/testCreateSecGroup")
+        self.assertEqual( get.name , "testCreateSecGroup")
+
+        get_notfound = self.provider.security.security_groups.get("testCreateSecGroup1")
+        self.assertEqual(get_notfound , None)
+
+        cb = list.data[0]
+        rules = cb.rules
+        cb.add_rule('*', '25', '100', '*')
+        self.assertEqual(len(rules), 3)
+
+        get_rule = cb.get_rule('*', '*', '*', '*')
+        self.assertEqual(str(get_rule), "<CBSecurityGroupRule: IP: *; from: *; to: *; grp: None>")
+
+        get_rule_notfound = cb.get_rule('*', '25', '1', '1')
+        self.assertEqual(str(get_rule_notfound), 'None')
+
+        rule_json = rules[1].to_json()
+        self.assertEqual(rule_json[2:9], "cidr_ip")
+
+        sg_json = cb.to_json()
+        self.assertEqual(sg_json[2:4], "id")
+
+        with self.assertRaises(Exception):
+            rules[1].delete()
+
+        sg_del = self.provider.security.security_groups.delete("/subscriptions/7904d702-e01c-4826-8519-f5a25c866a96/resourceGroups/cloudbridge-azure/providers/Microsoft.Network/networkSecurityGroups/testCreateSecGroup")
+        self.assertEqual(sg_del, True)
+
+        sg_del_notfound = self.provider.security.security_groups.delete("sg5")
+        self.assertEqual(sg_del_notfound, False)