Synnefo » ./kamaki

# Copyright 2012 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 kamaki.clients import Client, ClientError

class AstakosClient(Client):

    """GRNet Astakos API client"""

    _cache = {}

    def __init__(self, base_url, token):

        super(AstakosClient, self).__init__(base_url, token)

    def authenticate(self, token=None):

        """Get authentication information and store it in this client

        As long as the AstakosClient instance is alive, the latest

        authentication information for this token will be available

        :param token: (str) custom token to authenticate

        :returns: (dict) authentication information

        """

        self.token = token or self.token

        self._cache[self.token] = self.get('/im/authenticate').json

        return self._cache[self.token]

    def list(self):

        """list cached user information"""

        r = []

        for k, v in self._cache.items():

            r.append(dict(v))

            r[-1].update(dict(auth_token=k))

        return r

    def info(self, token=None):

        """Get (cached) user information"""

        token_bu = self.token

        token = token or self.token

        try:

            r = self._cache[token]

        except KeyError:

            r = self.authenticate(token)

        self.token = token_bu

        return r

    def term(self, key, token=None):

        """Get (cached) term, from user credentials"""

        return self.info(token).get(key, None)

# 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

from unittest import TestCase

from kamaki.clients.astakos import AstakosClient

example = dict(

        name='Simple Name',

        username='User Full Name',

        auth_token_expires='1362583796000',

        auth_token_created='1359991796000',

        email=['user@example.gr'],

        id=42,

        uuid='aus3r-uu1d-f0r-73s71ng-as7ak0s')

example0 = dict(

        name='Simple Name 0',

        username='User Full Name 0',

        auth_token_expires='1362583796001',

        auth_token_created='1359991796001',

        email=['user0@example.gr'],

        id=32,

        uuid='an07h2r-us3r-uu1d-f0r-as7ak0s')

class FR(object):

    json = example

    headers = {}

    content = json

    status = None

    status_code = 200

    def release(self):

        pass

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

class Astakos(TestCase):

    cached = False

    def setUp(self):

        self.url = 'https://astakos.example.com'

        self.token = 'ast@k0sT0k3n=='

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

    def tearDown(self):

        FR.json = example

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

    def _authenticate(self, PR):

        r = self.client.authenticate()

        self.cached = True

        return r

    def test_authenticate(self):

        r = self._authenticate()

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

        self.assertEqual(self.client.http_client.path, '/im/authenticate')

        for term, val in example.items():

            self.assertTrue(term in r)

            self.assertEqual(val, r[term])

    def test_info(self):

        if not self.cached:

            self._authenticate()

            return self.test_info()

        self.assertTrue(set(

            example.keys()).issubset(self.client.info().keys()))

    def test_get(self):

        if not self.cached:

            self._authenticate()

            return self.test_get()

        for term, val in example.items():

            self.assertEqual(self.client.term(term, self.token), val)

        self.assertTrue(example['email'][0] in self.client.term('email'))

    def test_list(self):

        if not self.cached:

            self._authenticate

        FR.json = example0

        self._authenticate()

        r = self.client.list()

        self.assertTrue(len(r) == 1)

        self.assertEqual(r[0]['auth_token'], self.token)

if __name__ == '__main__':

    from sys import argv

    from kamaki.clients.test import runTestCase

    runTestCase(Astakos, 'AstakosClient', argv[1:])

# Copyright 2011-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 time import sleep

from kamaki.clients.cyclades_rest_api import CycladesClientApi

from kamaki.clients import ClientError

from sys import stdout

class CycladesClient(CycladesClientApi):

    """GRNet Cyclades API client"""

    def start_server(self, server_id):

        """Submit a startup request

        :param server_id: integer (str or int)

        """

        req = {'start': {}}

        r = self.servers_post(server_id, 'action', json_data=req, success=202)

        r.release()

    def shutdown_server(self, server_id):

        """Submit a shutdown request

        :param server_id: integer (str or int)

        """

        req = {'shutdown': {}}

        r = self.servers_post(server_id, 'action', json_data=req, success=202)

        r.release()

    def get_server_console(self, server_id):

        """

        :param server_id: integer (str or int)

        :returns: (dict) info to set a VNC connection to VM

        """

        req = {'console': {'type': 'vnc'}}

        r = self.servers_post(server_id, 'action', json_data=req, success=200)

        return r.json['console']

    def get_firewall_profile(self, server_id):

        """

        :param server_id: integer (str or int)

        :returns: (str) ENABLED | DISABLED | PROTECTED

        :raises ClientError: 520 No Firewall Profile

        """

        r = self.get_server_details(server_id)

        try:

            return r['attachments']['values'][0]['firewallProfile']

        except KeyError:

            raise ClientError(

                'No Firewall Profile',

                details='Server %s is missing a firewall profile' % server_id)

    def set_firewall_profile(self, server_id, profile):

        """Set the firewall profile for the public interface of a server

        :param server_id: integer (str or int)

        :param profile: (str) ENABLED | DISABLED | PROTECTED

        """

        req = {'firewallProfile': {'profile': profile}}

        r = self.servers_post(server_id, 'action', json_data=req, success=202)

        r.release()

    def list_servers(self, detail=False, changes_since=None):

        """

        :param detail: (bool) append full server details to each item if true

        :param changes_since: (date)

        :returns: list of server ids and names

        """

        detail = 'detail' if detail else ''

        r = self.servers_get(command=detail, changes_since=changes_since)

        return r.json['servers']['values']

    def list_server_nics(self, server_id):

        """

        :param server_id: integer (str or int)

        :returns: (dict) network interface connections

        """

        r = self.servers_get(server_id, 'ips')

        return r.json['addresses']['values']

    def get_server_stats(self, server_id):

        """

        :param server_id: integer (str or int)

        :returns: (dict) auto-generated graphs of statistics (urls)

        """

        r = self.servers_get(server_id, 'stats')

        return r.json['stats']

    def list_networks(self, detail=False):

        """

        :param detail: (bool)

        :returns: (list) id,name if not detail else full info per network

        """

        detail = 'detail' if detail else ''

        r = self.networks_get(command=detail)

        return r.json['networks']['values']

    def list_network_nics(self, network_id):

        """

        :param network_id: integer (str or int)

        :returns: (list)

        """

        r = self.networks_get(network_id=network_id)

        return r.json['network']['attachments']['values']

    def create_network(

            self, name,

            cidr=None, gateway=None, type=None, dhcp=False):

        """

        :param name: (str)

        :param cidr: (str)

        :param geteway: (str)

        :param type: (str)

        :param dhcp: (bool)

        :returns: (dict) network detailed info

        """

        net = dict(name=name)

        if cidr:

            net['cidr'] = cidr

        if gateway:

            net['gateway'] = gateway

        if type:

            net['type'] = type

        net['dhcp'] = True if dhcp else False

        req = dict(network=net)

        r = self.networks_post(json_data=req, success=202)

        return r.json['network']

    def get_network_details(self, network_id):

        """

        :param network_id: integer (str or int)

        :returns: (dict)

        """

        r = self.networks_get(network_id=network_id)

        return r.json['network']

    def update_network_name(self, network_id, new_name):

        """

        :param network_id: integer (str or int)

        :param new_name: (str)

        """

        req = {'network': {'name': new_name}}

        r = self.networks_put(network_id=network_id, json_data=req)

        r.release()

    def delete_network(self, network_id):

        """

        :param network_id: integer (str or int)

        :raises ClientError: 421 Network in use

        """

        try:

            r = self.networks_delete(network_id)

        except ClientError as err:

            if err.status == 421:

                err.details = [

                    'Network may be still connected to at least one server']

            raise err

        r.release()

    def connect_server(self, server_id, network_id):

        """ Connect a server to a network

        :param server_id: integer (str or int)

        :param network_id: integer (str or int)

        """

        req = {'add': {'serverRef': server_id}}

        r = self.networks_post(network_id, 'action', json_data=req)

        r.release()

    def disconnect_server(self, server_id, nic_id):

        """

        :param server_id: integer (str or int)

        :param nic_id: (str)

        """

        vm_nets = self.list_server_nics(server_id)

        num_of_disconnections = 0

        for (nic_id, network_id) in [(

                net['id'],

                net['network_id']) for net in vm_nets if nic_id == net['id']]:

            req = {'remove': {'attachment': '%s' % nic_id}}

            r = self.networks_post(network_id, 'action', json_data=req)

            r.release()

            num_of_disconnections += 1

        return num_of_disconnections

    def disconnect_network_nics(self, netid):

        """

        :param netid: integer (str or int)

        """

        for nic in self.list_network_nics(netid):

            req = dict(remove=dict(attachment=nic))

            r = self.networks_post(netid, 'action', json_data=req)

            r.release()

    def wait_server(

            self,

            server_id,

            current_status='BUILD',

            delay=0.5,

            max_wait=128,

            wait_cb=None):

        """Wait for server while its status is current_status

        :param server_id: integer (str or int)

        :param current_status: (str) BUILD|ACTIVE|STOPPED|DELETED|REBOOT

        :param delay: time interval between retries

        :param wait_cb: if set a progressbar is used to show progress

        :returns: (str) the new mode if succesfull, (bool) False if timed out

        """

        r = self.get_server_details(server_id)

        if r['status'] != current_status:

            return r['status']

        old_wait = total_wait = 0

        if current_status == 'BUILD':

            max_wait = 100

            wait_gen = wait_cb(max_wait) if wait_cb else None

        elif wait_cb:

            wait_gen = wait_cb(1 + max_wait // delay)

            wait_gen.next()

        while r['status'] == current_status and total_wait <= max_wait:

            if current_status == 'BUILD':

                total_wait = int(r['progress'])

                if wait_cb:

                    for i in range(int(old_wait), int(total_wait)):

                        wait_gen.next()

                    old_wait = total_wait

                else:

                    stdout.write('.')

                    stdout.flush()

            else:

                if wait_cb:

                    wait_gen.next()

                else:

                    stdout.write('.')

                    stdout.flush()

                total_wait += delay

            sleep(delay)

            r = self.get_server_details(server_id)

        if r['status'] != current_status:

            if wait_cb:

                try:

                    while True:

                        wait_gen.next()

                except:

                    pass

            return r['status']

        return False

# 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, Mock, call

from unittest import TestCase

from json import loads

from kamaki.clients import Client, 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_send = dict(server=dict(

    flavorRef=fid,

    name=vm_name,

    imageRef=img_ref,

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

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"))))

img_recv = dict(image=dict(

    status="ACTIVE",

    updated="2013-02-26T11:10:14+00:00",

    name="Debian Base",

    created="2013-02-26T11:03:29+00:00",

    progress=100,

    id=img_ref,

    metadata=dict(values=dict(

        partition_table="msdos",

        kernel="2.6.32",

        osfamily="linux",

        users="root",

        gui="No GUI",

        sortorder="1",

        os="debian",

        root_partition="1",

        description="Debian 6.0.7 (Squeeze) Base System"))))

vm_list = dict(servers=dict(values=[

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

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

flavor_list = dict(flavors=dict(values=[

        dict(id=41, name="C1R1024D20"),

        dict(id=42, name="C1R1024D40"),

        dict(id=43, name="C1R1028D20")]))

img_list = dict(images=dict(values=[

    dict(name="maelstrom", id="0fb03e45-7d5a-4515-bd4e-e6bbf6457f06"),

    dict(name="edx_saas", id="1357163d-5fd8-488e-a117-48734c526206"),

    dict(name="Debian_Wheezy_Base", id="1f8454f0-8e3e-4b6c-ab8e-5236b728dffe"),

    dict(name="CentOS", id="21894b48-c805-4568-ac8b-7d4bb8eb533d"),

    dict(name="Ubuntu Desktop", id="37bc522c-c479-4085-bfb9-464f9b9e2e31"),

    dict(name="Ubuntu 12.10", id="3a24fef9-1a8c-47d1-8f11-e07bd5e544fd"),

    dict(name="Debian Base", id="40ace203-6254-4e17-a5cb-518d55418a7d"),

    dict(name="ubuntu_bundled", id="5336e265-5c7c-4127-95cb-2bf832a79903")]))

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')]))

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

    def test_create_server(self):

        self.client.get_image_details = Mock(return_value=img_recv['image'])

        with patch.object(Client, 'request', side_effect=ClientError(

                'REQUEST ENTITY TOO LARGE',

                status=403)):

            self.assertRaises(

                ClientError,

                self.client.create_server,

                vm_name, fid, img_ref)

        with patch.object(

                self.C,

                'perform_request',

                return_value=FR()) as perform_req:

            self.assertRaises(

                ClientError,

                self.client.create_server,

                vm_name, fid, img_ref)

            FR.status_code = 202

            r = self.client.create_server(vm_name, fid, img_ref)

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

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

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

            self.assert_dicts_are_equal(loads(data), vm_send)

            self.assert_dicts_are_equal(r, vm_recv['server'])

            prsn = 'Personality string (does not work with real servers)'

            self.client.create_server(vm_name, fid, img_ref, prsn)

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

            data = loads(data)

            self.assertTrue('personality' in data['server'])

            self.assertEqual(prsn, data['server']['personality'])

    def test_list_servers(self):

        FR.json = vm_list

        with patch.object(

                self.C,

                'perform_request',

                return_value=FR()) as perform_req:

            r = self.client.list_servers()

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

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

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

            self.assert_dicts_are_equal(dict(values=r), vm_list['servers'])

            r = self.client.list_servers(detail=True)

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

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

        with patch.object(

                CycladesClientApi,

                'servers_get',

                return_value=FR()) as servers_get:

            self.client.list_servers(changes_since=True)

            self.assertTrue(servers_get.call_args[1]['changes_since'])

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

    def test_get_server_details(self, PR):

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

        r = self.client.get_server_details(vm_id)

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

        self.assertEqual(self.client.http_client.path, '/servers/%s' % vm_id)

        self.assert_dicts_are_equal(r, vm_recv['server'])

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

    def test_update_server_name(self, PR):

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

        new_name = vm_name + '_new'

        FR.status_code = 204

        self.client.update_server_name(vm_id, new_name)

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

        self.assertEqual(self.client.http_client.path, '/servers/%s' % vm_id)

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

        self.assert_dicts_are_equal(

            dict(server=dict(name=new_name)),

            loads(data))

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

    def test_reboot_server(self, PR):

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

        FR.status_code = 202

        self.client.reboot_server(vm_id)

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

        self.assertEqual(

            self.client.http_client.path,

            '/servers/%s/action' % vm_id)

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

        self.assert_dicts_are_equal(

            dict(reboot=dict(type='SOFT')),

            loads(data))

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

    def test_create_server_metadata(self, PR):

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

        metadata = dict(m1='v1', m2='v2', m3='v3')

        FR.json = dict(meta=vm_recv['server'])

        self.assertRaises(

            ClientError,

            self.client.create_server_metadata,

            vm_id, 'key', 'value')

        FR.status_code = 201

        for k, v in metadata.items():

            r = self.client.create_server_metadata(vm_id, k, v)

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

            self.assertEqual(

                self.client.http_client.path,

                '/servers/%s/meta/%s' % (vm_id, k))

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

            self.assertEqual(dict(meta={k: v}), loads(data))

            self.assert_dicts_are_equal(r, vm_recv['server'])

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

    def test_get_server_metadata(self, PR):

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

        metadata = dict(m1='v1', m2='v2', m3='v3')

        FR.json = dict(metadata=dict(values=metadata))

        r = self.client.get_server_metadata(vm_id)

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

        self.assertEqual(

            self.client.http_client.path,

            '/servers/%s/meta' % vm_id)

        self.assert_dicts_are_equal(r, metadata)

        for k, v in metadata.items():

            FR.json = dict(meta={k: v})

            r = self.client.get_server_metadata(vm_id, k)

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

            self.assertEqual(

                self.client.http_client.path,

                '/servers/%s/meta/%s' % (vm_id, k))

            self.assert_dicts_are_equal(r, {k: v})

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

    def test_update_server_metadata(self, servers_post):

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

        metadata = dict(m1='v1', m2='v2', m3='v3')

        FR.json = dict(metadata=metadata)

        r = self.client.update_server_metadata(vm_id, **metadata)

        self.assert_dicts_are_equal(r, metadata)

        (called_id, cmd) = servers_post.call_args[0]

        self.assertEqual(called_id, vm_id)

        self.assertEqual(cmd, 'meta')

        data = servers_post.call_args[1]['json_data']

        self.assert_dicts_are_equal(data, dict(metadata=metadata))

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

    def test_delete_server_metadata(self, servers_delete):

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

        key = 'metakey'

        self.client.delete_server_metadata(vm_id, key)

        self.assertEqual((vm_id, 'meta/' + key), servers_delete.call_args[0])

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

    def test_list_flavors(self, PR):

        FR.json = flavor_list

        r = self.client.list_flavors()

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

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

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

        self.assert_dicts_are_equal(dict(values=r), flavor_list['flavors'])

        r = self.client.list_flavors(detail=True)

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

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

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

    def test_get_flavor_details(self, PR):

        FR.json = dict(flavor=flavor_list['flavors'])

        r = self.client.get_flavor_details(fid)

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

        self.assertEqual(self.client.http_client.path, '/flavors/%s' % fid)

        self.assert_dicts_are_equal(r, flavor_list['flavors'])

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

    def test_list_images(self, PR):

        FR.json = img_list

        r = self.client.list_images()

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

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

        expected = img_list['images']['values']

        for i in range(len(r)):

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

        self.client.list_images(detail=True)

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

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

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

    def test_get_image_details(self, PR):

        FR.json = img_recv

        r = self.client.get_image_details(img_ref)

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

        self.assertEqual(self.client.http_client.path, '/images/%s' % img_ref)

        self.assert_dicts_are_equal(r, img_recv['image'])

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

    def test_get_image_metadata(self, IG):

        FR.json = dict(metadata=dict(values=img_recv['image']))

        r = self.client.get_image_metadata(img_ref)

        self.assertEqual(IG.call_args[0], ('%s' % img_ref, '/meta'))

        self.assert_dicts_are_equal(img_recv['image'], r)

        FR.json = dict(meta=img_recv['image'])

        key = 'somekey'

        self.client.get_image_metadata(img_ref, key)

        self.assertEqual(IG.call_args[0], ('%s' % img_ref, '/meta/%s' % key))

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

    def test_shutdown_server(self, PR):

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

        FR.status_code = 202

        self.client.shutdown_server(vm_id)

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

        self.assertEqual(

            self.client.http_client.path,

            '/servers/%s/action' % vm_id)

        self.assertEqual(

            PR.call_args[0],

            ('post',  '{"shutdown": {}}', {}, {}))

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

    def test_start_server(self, PR):

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

        FR.status_code = 202

        self.client.start_server(vm_id)

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

        self.assertEqual(

            self.client.http_client.path,

            '/servers/%s/action' % vm_id)

        self.assertEqual(PR.call_args[0], ('post',  '{"start": {}}', {}, {}))

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

    def test_get_server_console(self, PR):

        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(self.client.http_client.url, self.url)

        self.assertEqual(

            self.client.http_client.path,

            '/servers/%s/action' % vm_id)

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

        self.assertEqual(

            PR.call_args[0],

            ('post',  '{"console": {"type": "vnc"}}', {}, {}))

    def test_get_firewall_profile(self):

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

        v = 'Some profile'

        ret = {'attachments': {'values': [{'firewallProfile': v, 1:1}]}}

        with patch.object(

                CycladesClient,

                'get_server_details',

                return_value=ret) as GSD:

            r = self.client.get_firewall_profile(vm_id)

            self.assertEqual(r, v)

            self.assertEqual(GSD.call_args[0], (vm_id,))

            ret['attachments']['values'][0].pop('firewallProfile')

            self.assertRaises(

                ClientError,

                self.client.get_firewall_profile,

                vm_id)

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

    def test_set_firewall_profile(self, PR):

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

        v = 'Some profile'

        FR.status_code = 202

        self.client.set_firewall_profile(vm_id, v)

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

        self.assertEqual(

            self.client.http_client.path,

            '/servers/%s/action' % vm_id)

        self.assertEqual(PR.call_args[0], (

            'post',

            '{"firewallProfile": {"profile": "%s"}}' % v,

            {},

            {}))

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

    def test_get_server_stats(self, PR):

        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(self.client.http_client.url, self.url)

        self.assertEqual(

            self.client.http_client.path,

            '/servers/%s/stats' % vm_id)

        self.assert_dicts_are_equal(stats, r)

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

    def test_create_network(self, PR):

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

        FR.json = net_recv

        FR.status_code = 202

        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))

        for arg, val in test_args.items():

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

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

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

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

            self.assertEqual(

                self.client.http_client.path,

                '/networks')

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

            data = PR.call_args[0][1]

            expected = dict(network=dict(net_send['network']))

            expected['network'].update(kwargs)

            self.assert_dicts_are_equal(loads(data), expected)

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

    def test_connect_server(self, PR):

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

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

        FR.status_code = 202

        self.client.connect_server(vm_id, net_id)

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

        self.assertEqual(

            self.client.http_client.path,

            '/networks/%s/action' % net_id)

        self.assertEqual(

            PR.call_args[0],

            ('post', '{"add": {"serverRef": %s}}' % vm_id, {}, {}))

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

    def test_disconnect_server(self, NP):

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

        net_id = net_recv['network']['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.call_args[0], (vm_id,))

            self.assertEqual(NP.call_args[0], (net_id, 'action'))

            self.assertEqual(

                NP.call_args[1],

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

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

    def test_list_server_nics(self, PR):

        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(self.client.http_client.url, self.url)

        self.assertEqual(

            self.client.http_client.path,

            '/servers/%s/ips' % vm_id)

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

        for i in range(len(r)):

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

    @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'])