Synnefo » ./kamaki

# Copyright 2013 GRNET S.A. All rights reserved.

#

# Redistribution and use in source and binary forms, with or

# without modification, are permitted provided that the following

# conditions are met:

#

#   1. Redistributions of source code must retain the above

#      copyright notice, this list of conditions and the following

#      disclaimer.

#

#   2. Redistributions in binary form must reproduce the above

#      copyright notice, this list of conditions and the following

#      disclaimer in the documentation and/or other materials

#      provided with the distribution.

#

# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS

# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR

# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,

# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT

# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF

# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED

# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT

# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN

# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

# POSSIBILITY OF SUCH DAMAGE.

#

# The views and conclusions contained in the software and

# documentation are those of the authors and should not be

# interpreted as representing official policies, either expressed

# or implied, of GRNET S.A.

from mock import patch, call

from unittest import TestCase

from json import loads

from kamaki.clients import ClientError

from kamaki.clients.cyclades import CycladesClient

from kamaki.clients.cyclades_rest_api import CycladesClientApi

img_ref = "1m4g3-r3f3r3nc3"

vm_name = "my new VM"

fid = 42

vm_recv = dict(server=dict(

    status="BUILD",

    updated="2013-03-01T10:04:00.637152+00:00",

    hostId="",

    name=vm_name,

    imageRef=img_ref,

    created="2013-03-01T10:04:00.087324+00:00",

    flavorRef=fid,

    adminPass="n0n3sh@11p@55",

    suspended=False,

    progress=0,

    id=31173,

    metadata=dict(values=dict(os="debian", users="root"))))

vm_list = dict(servers=dict(values=[

    dict(name='n1', id=1),

    dict(name='n2', id=2)]))

net_send = dict(network=dict(dhcp=False, name='someNet'))

net_recv = dict(network=dict(

    status="PENDING",

    updated="2013-03-05T15:04:51.758780+00:00",

    name="someNet",

    created="2013-03-05T15:04:51.758728+00:00",

    cidr6=None,

    id="2130",

    gateway6=None,

    public=False,

    dhcp=False,

    cidr="192.168.1.0/24",

    type="MAC_FILTERED",

    gateway=None,

    attachments=dict(values=[dict(name='att1'), dict(name='att2')])))

net_list = dict(networks=dict(values=[

    dict(id=1, name='n1'),

    dict(id=2, name='n2'),

    dict(id=3, name='n3')]))

firewalls = dict(attachments=dict(values=[

    dict(firewallProfile='50m3_pr0f1L3', otherStuff='57uff')]))

class FR(object):

    """FR stands for Fake Response"""

    json = vm_recv

    headers = {}

    content = json

    status = None

    status_code = 200

    def release(self):

        pass

khttp = 'kamaki.clients.connection.kamakicon.KamakiHTTPConnection'

cyclades_pkg = 'kamaki.clients.cyclades.CycladesClient'

class Cyclades(TestCase):

    def assert_dicts_are_equal(self, d1, d2):

        for k, v in d1.items():

            self.assertTrue(k in d2)

            if isinstance(v, dict):

                self.assert_dicts_are_equal(v, d2[k])

            else:

                self.assertEqual(unicode(v), unicode(d2[k]))

    """Set up a Cyclades thorough test"""

    def setUp(self):

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

        self.token = 'cyc14d3s70k3n'

        self.client = CycladesClient(self.url, self.token)

        from kamaki.clients.connection.kamakicon import KamakiHTTPConnection

        self.C = KamakiHTTPConnection

    def tearDown(self):

        FR.status_code = 200

        FR.json = vm_recv

    @patch('%s.servers_get' % cyclades_pkg, return_value=FR())

    def test_list_servers(self, SG):

        FR.json = vm_list

        for detail, since in ((0, 0), (True, 0), (0, 'd473'), (True, 'd473')):

            r = self.client.list_servers(detail=detail, changes_since=since)

            self.assertEqual(SG.mock_calls[-1], call(

                command='detail' if detail else '',

                changes_since=since))

            expected = vm_list['servers']['values']

            for i, vm in enumerate(r):

                self.assert_dicts_are_equal(vm, expected[i])

            self.assertEqual(i + 1, len(expected))

    @patch('%s.servers_post' % cyclades_pkg, return_value=FR())

    def test_shutdown_server(self, SP):

        vm_id = vm_recv['server']['id']

        self.client.shutdown_server(vm_id)

        self.assertEqual(SP.mock_calls[-1], call(

            vm_id, 'action',

            json_data=dict(shutdown=dict()), success=202))

    @patch('%s.servers_post' % cyclades_pkg, return_value=FR())

    def test_start_server(self, SP):

        vm_id = vm_recv['server']['id']

        self.client.start_server(vm_id)

        self.assertEqual(SP.mock_calls[-1], call(

            vm_id, 'action',

            json_data=dict(start=dict()), success=202))

    @patch('%s.servers_post' % cyclades_pkg, return_value=FR())

    def test_get_server_console(self, SP):

        cnsl = dict(console=dict(info1='i1', info2='i2', info3='i3'))

        FR.json = cnsl

        vm_id = vm_recv['server']['id']

        r = self.client.get_server_console(vm_id)

        self.assertEqual(SP.mock_calls[-1], call(

            vm_id, 'action',

            json_data=dict(console=dict(type='vnc')), success=200))

        self.assert_dicts_are_equal(r, cnsl['console'])

    def test_get_firewall_profile(self):

        vm_id = vm_recv['server']['id']

        v = firewalls['attachments']['values'][0]['firewallProfile']

        with patch.object(

                CycladesClient, 'get_server_details',

                return_value=firewalls) as GSD:

            r = self.client.get_firewall_profile(vm_id)

            self.assertEqual(r, v)

            self.assertEqual(GSD.mock_calls[-1], call(vm_id))

        with patch.object(

                CycladesClient, 'get_server_details',

                return_value=dict()):

            self.assertRaises(

                ClientError,

                self.client.get_firewall_profile,

                vm_id)

    @patch('%s.servers_post' % cyclades_pkg, return_value=FR())

    def test_set_firewall_profile(self, SP):

        vm_id = vm_recv['server']['id']

        v = firewalls['attachments']['values'][0]['firewallProfile']

        self.client.set_firewall_profile(vm_id, v)

        self.assertEqual(SP.mock_calls[-1], call(

            vm_id, 'action',

            json_data=dict(firewallProfile=dict(profile=v)), success=202))

    @patch('%s.servers_get' % cyclades_pkg, return_value=FR())

    def test_get_server_stats(self, SG):

        vm_id = vm_recv['server']['id']

        stats = dict(stat1='v1', stat2='v2', stat3='v3', stat4='v4')

        FR.json = dict(stats=stats)

        r = self.client.get_server_stats(vm_id)

        self.assertEqual(SG.mock_calls[-1], call(vm_id, 'stats'))

        self.assert_dicts_are_equal(stats, r)

    @patch('%s.networks_post' % cyclades_pkg, return_value=FR())

    def test_create_network(self, NP):

        net_name = net_send['network']['name']

        FR.json = net_recv

        full_args = dict(

                cidr='192.168.0.0/24',

                gateway='192.168.0.1',

                type='MAC_FILTERED',

                dhcp=True)

        test_args = dict(full_args)

        test_args.update(dict(empty=None, full=None))

        net_exp = dict(dhcp=False, name=net_name)

        for arg, val in test_args.items():

            kwargs = {} if arg == 'empty' else full_args if (

                arg == 'full') else {arg: val}

            expected = dict(network=dict(net_exp))

            expected['network'].update(kwargs)

            r = self.client.create_network(net_name, **kwargs)

            self.assertEqual(

                NP.mock_calls[-1],

                call(json_data=expected, success=202))

            self.assert_dicts_are_equal(r, net_recv['network'])

    @patch('%s.networks_post' % cyclades_pkg, return_value=FR())

    def test_connect_server(self, NP):

        vm_id = vm_recv['server']['id']

        net_id = net_recv['network']['id']

        self.client.connect_server(vm_id, net_id)

        self.assertEqual(NP.mock_calls[-1], call(

            net_id, 'action',

            json_data=dict(add=dict(serverRef=vm_id))))

    @patch('%s.networks_post' % cyclades_pkg, return_value=FR())

    def test_disconnect_server(self, NP):

        net_id, vm_id = net_recv['network']['id'], vm_recv['server']['id']

        nic_id = 'nic-%s-%s' % (net_id, vm_id)

        vm_nics = [

            dict(id=nic_id, network_id=net_id),

            dict(id='another-nic-id', network_id='another-net-id'),

            dict(id=nic_id * 2, network_id=net_id * 2)]

        with patch.object(

                CycladesClient,

                'list_server_nics',

                return_value=vm_nics) as LSN:

            r = self.client.disconnect_server(vm_id, nic_id)

            self.assertEqual(r, 1)

            self.assertEqual(LSN.mock_calls[-1], call(vm_id,))

            self.assertEqual(NP.mock_calls[-1], call(

                net_id, 'action',

                json_data=dict(remove=dict(attachment=nic_id))))

    @patch('%s.servers_get' % cyclades_pkg, return_value=FR())

    def test_list_server_nics(self, SG):

        vm_id = vm_recv['server']['id']

        nics = dict(addresses=dict(values=[dict(id='nic1'), dict(id='nic2')]))

        FR.json = nics

        r = self.client.list_server_nics(vm_id)

        self.assertEqual(SG.mock_calls[-1], call(vm_id, 'ips'))

        expected = nics['addresses']['values']

        for i in range(len(r)):

            self.assert_dicts_are_equal(r[i], expected[i])

        self.assertEqual(i + 1, len(r))

    @patch('%s.perform_request' % khttp, return_value=FR())

    def test_list_networks(self, PR):

        FR.json = net_list

        r = self.client.list_networks()

        self.assertEqual(self.client.http_client.url, self.url)

        self.assertEqual(self.client.http_client.path, '/networks')

        expected = net_list['networks']['values']

        for i in range(len(r)):

            self.assert_dicts_are_equal(expected[i], r[i])

        self.client.list_networks(detail=True)

        self.assertEqual(self.client.http_client.url, self.url)

        self.assertEqual(self.client.http_client.path, '/networks/detail')

    @patch('%s.perform_request' % khttp, return_value=FR())

    def test_list_network_nics(self, PR):

        net_id = net_recv['network']['id']

        FR.json = net_recv

        r = self.client.list_network_nics(net_id)

        self.assertEqual(self.client.http_client.url, self.url)

        self.assertEqual(

            self.client.http_client.path,

            '/networks/%s' % net_id)

        expected = net_recv['network']['attachments']['values']

        for i in range(len(r)):

            self.assert_dicts_are_equal(r[i], expected[i])

    @patch('%s.networks_post' % cyclades_pkg, return_value=FR())

    def test_disconnect_network_nics(self, NP):

        net_id = net_recv['network']['id']

        nics = ['nic1', 'nic2', 'nic3']

        with patch.object(

                CycladesClient,

                'list_network_nics',

                return_value=nics) as lnn:

            self.client.disconnect_network_nics(net_id)

            lnn.assert_called_once_with(net_id)

            for i in range(len(nics)):

                expected = call(net_id, 'action', json_data=dict(

                    remove=dict(attachment=nics[i])))

                self.assertEqual(expected, NP.mock_calls[i])

    @patch('%s.perform_request' % khttp, return_value=FR())

    def test_get_network_details(self, PR):

        FR.json = net_recv

        net_id = net_recv['network']['id']

        r = self.client.get_network_details(net_id)

        self.assertEqual(self.client.http_client.url, self.url)

        self.assertEqual(

            self.client.http_client.path,

            '/networks/%s' % net_id)

        self.assert_dicts_are_equal(r, net_recv['network'])

    @patch('%s.perform_request' % khttp, return_value=FR())

    def test_update_network_name(self, PR):

        net_id = net_recv['network']['id']

        new_name = '%s_new' % net_id

        FR.status_code = 204

        self.client.update_network_name(net_id, new_name)

        self.assertEqual(self.client.http_client.url, self.url)

        self.assertEqual(self.client.http_client.path, '/networks/%s' % net_id)

        (method, data, a_headers, a_params) = PR.call_args[0]

        self.assert_dicts_are_equal(

            dict(network=dict(name=new_name)),

            loads(data))

    @patch('%s.perform_request' % khttp, return_value=FR())

    def test_delete_network(self, PR):

        net_id = net_recv['network']['id']

        FR.status_code = 204

        self.client.delete_network(net_id)

        self.assertEqual(self.client.http_client.url, self.url)

        self.assertEqual(self.client.http_client.path, '/networks/%s' % net_id)

if __name__ == '__main__':

    from sys import argv

    from kamaki.clients.test import runTestCase

    runTestCase(Cyclades, 'Cyclades (multi) Client', argv[1:])