cloudbridge/integration_test/test_integration_azure_security_group.py

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_name = '{0}'.format(uuid.uuid4())
        print(sg_name)

        listBeforeCreate = self.provider.security.security_groups.list()
        print("Before create - " + str(len(listBeforeCreate)))
        sg = self.provider.security.security_groups.create(name=sg_name, description="testCreateSecGroup", network_id="")
        self.assertEqual(sg_name, sg.name)

        listAfterCreate = self.provider.security.security_groups.list()
        print("After create - "+ str(len(listAfterCreate)))
        self.assertEqual( len(listAfterCreate) , len(listBeforeCreate)+1)

        get = self.provider.security.security_groups.get("/subscriptions/7904d702-e01c-4826-8519-f5a25c866a96/resourceGroups/cloudbridge-azure/providers/Microsoft.Network/networkSecurityGroups/"+ sg_name)
        self.assertEqual( get.name , sg_name)

        get_notfound = self.provider.security.security_groups.get("/subscriptions/7904d702-e01c-4826-8519-f5a25c866a96/resourceGroups/cloudbridge-azure/providers/Microsoft.Network/networkSecurityGroups/testCreateSecGroup1")
        self.assertEqual(get_notfound , None)

        cb = listAfterCreate.data[0]
        lenBeforeCreateRule = len(cb.rules)
        cb.add_rule('*', '25', '100', '*')
        lenAfterCreateRule = len(cb.rules)
        self.assertEqual(lenAfterCreateRule, lenBeforeCreateRule+1)

        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 = cb.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):
            cb.rules[1].delete()

        listBeforeDeleteFound = self.provider.security.security_groups.list()
        sg_del = self.provider.security.security_groups.delete("/subscriptions/7904d702-e01c-4826-8519-f5a25c866a96/resourceGroups/cloudbridge-azure/providers/Microsoft.Network/networkSecurityGroups/"+sg_name)
        listAfterDeleteFound = self.provider.security.security_groups.list()
        print(str(len(listBeforeDeleteFound))+"---"+str(len(listAfterDeleteFound)))
        self.assertEqual(len(listAfterDeleteFound), len(listBeforeDeleteFound)-1)

        sg_del_notfound = self.provider.security.security_groups.delete("/subscriptions/7904d702-e01c-4826-8519-f5a25c866a96/resourceGroups/cloudbridge-azure/providers/Microsoft.Network/networkSecurityGroups/sg5")
        listAfterDeleteNotFound = self.provider.security.security_groups.list()
        self.assertEqual(len(listAfterDeleteNotFound), len(listAfterDeleteFound))