cloudbridge/test/test_provider_security_service.py

import uuid
from test.helpers import ProviderTestBase
import test.helpers as helpers


class ProviderSecurityServiceTestCase(ProviderTestBase):

    def __init__(self, methodName, provider):
        super(ProviderSecurityServiceTestCase, self).__init__(
            methodName=methodName, provider=provider)

    def test_crud_key_pair_service(self):
        name = 'cbtestkeypairA-{0}'.format(uuid.uuid4())
        kp = self.provider.security.key_pairs.create(name=name)
        with helpers.exception_action(
            lambda:
                self.provider.security.key_pairs.delete(name=kp.name)
        ):
            kpl = self.provider.security.key_pairs.list()
            found_kp = [k for k in kpl if k.name == name]
            self.assertTrue(
                len(found_kp) == 1,
                "List key pairs did not return the expected key {0}."
                .format(name))
            self.provider.security.key_pairs.delete(name=kp.name)
            kpl = self.provider.security.key_pairs.list()
            found_kp = [k for k in kpl if k.name == name]
            self.assertTrue(
                len(found_kp) == 0,
                "Key pair {0} should have been deleted but still exists."
                .format(name))
        no_kp = self.provider.security.key_pairs.delete(name='bogus_kp')
        self.assertTrue(
            no_kp,
            "Found a key pair {0} that should not exist?".format(no_kp))

    def test_key_pair(self):
        name = 'cbtestkeypairB-{0}'.format(uuid.uuid4())
        kp = self.provider.security.key_pairs.create(name=name)
        with helpers.exception_action(
            lambda:
                self.provider.security.key_pairs.delete(name=kp.name)
        ):
            kpl = self.provider.security.key_pairs.list()
            found_kp = [k for k in kpl if k.name == name]
            self.assertTrue(
                len(found_kp) == 1,
                "List key pairs did not return the expected key {0}."
                .format(name))
            self.assertTrue(
                repr(kp) == "<CBKeyPair: {0}>".format(name),
                "KeyPair repr {0} not matching expected format.".format(kp))
            self.assertIsNotNone(
                kp.material,
                "KeyPair material is empty but it should not be.")
            kp.delete()
            kpl = self.provider.security.key_pairs.list()
            found_kp = [k for k in kpl if k.name == name]
            self.assertTrue(
                len(found_kp) == 0,
                "Key pair {0} should have been deleted but still exists."
                .format(name))

    def test_crud_security_group_service(self):
        name = 'cbtestsecuritygroupA-{0}'.format(uuid.uuid4())
        sg = self.provider.security.security_groups.create(
            name=name, description=name)
        with helpers.exception_action(
            lambda:
                self.provider.security.security_groups.delete(group_id=sg.id)
        ):
            sgl = self.provider.security.security_groups.get(
                group_names=[
                    sg.name])
            found_sg = [g for g in sgl if g.name == name]
            self.assertTrue(
                len(found_sg) == 1,
                "List security groups did not return the expected group {0}."
                .format(name))
            self.provider.security.security_groups.delete(group_id=sg.id)
            sgl = self.provider.security.security_groups.list()
            found_sg = [g for g in sgl if g.name == name]
            self.assertTrue(
                len(found_sg) == 0,
                "Security group {0} should have been deleted but still exists."
                .format(name))

    def test_security_group(self):
        name = 'cbtestsecuritygroupB-{0}'.format(uuid.uuid4())
        sg = self.provider.security.security_groups.create(
            name=name, description=name)
        with helpers.exception_action(
            lambda:
                self.provider.security.security_groups.delete(group_id=sg.id)
        ):
            sg.add_rule('tcp', 1111, 1111, '0.0.0.0/0')
            found_rules = [rule for rule in sg.rules if
                           rule.cidr_ip == '0.0.0.0/0' and
                           rule.ip_protocol == 'tcp' and
                           rule.from_port == 1111 and
                           rule.to_port == 1111]
            self.assertTrue(
                len(found_rules) == 1,
                "Expected rule not found in security group: {0}".format(name))
            self.assertTrue(
                repr(sg.rules[0]) == ("<CBSecurityGroupRule: IP: {0}; from: "
                                      "{1}; to: {2}>"
                                      .format(sg.rules[0].ip_protocol,
                                              sg.rules[0].from_port,
                                              sg.rules[0].to_port)),
                ("Security group rule repr {0} not matching expected format."
                 .format(sg.rules[0])))
            sg.delete()
            sgl = self.provider.security.security_groups.list()
            found_sg = [g for g in sgl if g.name == name]
            self.assertTrue(
                len(found_sg) == 0,
                "Security group {0} should have been deleted but still exists."
                .format(name))