cloudbridge/test/test_azure_helpers.py

from test.helpers import ProviderTestBase
from cloudbridge.cloud.providers.azure import helpers as azure_helpers


class AzureHelpersTestCase(ProviderTestBase):
    def __init__(self, methodName, provider):
        super(AzureHelpersTestCase, self).__init__(
            methodName=methodName, provider=provider)

    def test_parse_url_valid(self):
        params = azure_helpers.parse_url('/subscriptionId/{subscriptionId}', '/subscriptionId/123-1345')
        self.assertTrue(len(params) == 1, 'Parameter count should be 1')

    def test_parse_url_invalid(self):
        with self.assertRaises(Exception):
            params = azure_helpers.parse_url('/subscriptionId/{subscriptionId}', '/123-1345')

    def test_filter_matched(self):
        ex1 = Expando()
        ex1.name = 'test'

        ex2 = Expando()
        ex2.name = 'abc'

        result = azure_helpers.filter([ex1, ex2], {'name': 'test'})
        self.assertTrue(len(result) == 1, 'Result count should be one')

    def test_filter_not_matched(self):
        ex1 = Expando()
        ex1.name = 'pqr'

        ex2 = Expando()
        ex2.name = 'abc'
        result = azure_helpers.filter([ex1, ex2], {'name': 'test123'})
        self.assertTrue(len(result) == 0, 'Result count should be zero')

    def test_filter_None(self):
        ex1 = Expando()
        ex1.name = 'pqr'

        ex2 = Expando()
        ex2.name = 'abc'
        result = azure_helpers.filter([ex1, ex2], None)
        self.assertTrue(len(result) == 2, 'Result count should be two')


class Expando(object):
    pass