Synnefo » ./kamaki

    def list_networks(self):

        r = self.networks_get(success=200)

        return r.json['networks']

    def create_network(self, name=None, admin_state_up=None, shared=None):

        req = dict(network=dict(

            name=name or '', admin_state_up=bool(admin_state_up)))

        r = self.networks_post(json_data=req, shared=shared, success=201)

        return r.json['network']

        del self.client

class FakeObject(object):

    json = None

class NetworkClient(TestCase):

    """Set up a ComputesRest thorough test"""

    def setUp(self):

        self.url = 'http://network.example.com'

        self.token = 'n2tw0rk70k3n'

        self.client = network.NetworkClient(self.url, self.token)

    def tearDown(self):

        del self.client

    @patch(

        'kamaki.clients.network.NetworkClient.networks_get',

        return_value=FakeObject())

    def test_list_networks(self, networks_get):

        FakeObject.json = dict(networks='ret val')

        self.assertEqual(self.client.list_networks(), 'ret val')

        networks_get.assert_called_once_with(success=200)

    @patch(

        'kamaki.clients.network.NetworkClient.networks_post',

        return_value=FakeObject())

    def test_create_network(self, networks_post):

        FakeObject.json = dict(network='ret val')

        req = dict()

        for kwargs in (dict(shared=None), dict(shared=True)):

            for body_params in product(

                    (('name', None, ''), ('name', 'some name', 'some name')),

                    (

                        ('admin_state_up', None, False),

                        ('admin_state_up', True, True))):

                fullargs = dict(kwargs)

                for k, v, exp in body_params:

                    fullargs[k] = v

                    req[k] = exp

                self.assertEqual(

                    self.client.create_network(**fullargs), 'ret val')

                expargs = dict(kwargs)

                expargs = dict(json_data=dict(network=req), success=201)

                expargs.update(kwargs)

                self.assertEqual(networks_post.mock_calls[-1], call(**expargs))

    if not argv[1:] or argv[1] == 'NetworkClient':

        not_found = False

        runTestCase(NetworkClient, 'Network Client', argv[2:])

from kamaki.clients.network.test import NetworkClient, NetworkRestClient