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 kamaki.clients import astakos, ClientError

example = dict(

    serviceCatalog=[

        dict(name='service name 1', type='compute', endpoints=[

            dict(version_id='v1', publicUrl='http://1.1.1.1/v1'),

            dict(version_id='v2', publicUrl='http://1.1.1.1/v2')]),

        dict(name='service name 2', type='image', endpoints=[

            dict(version_id='v2', publicUrl='http://1.1.1.1/v2'),

            dict(version_id='v2.1', publicUrl='http://1.1.1.1/v2/xtra')])

        ],

    user=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(

    serviceCatalog=[

        dict(name='service name 1', type='compute', endpoints=[

            dict(version_id='v1', publicUrl='http://1.1.1.1/v1'),

            dict(version_id='v2', publicUrl='http://1.1.1.1/v2')]),

        dict(name='service name 3', type='object-storage', endpoints=[

            dict(version_id='v2', publicUrl='http://1.1.1.1/v2'),

            dict(version_id='v2.1', publicUrl='http://1.1.1.1/v2/xtra')])

        ],

    user=dict(

        name='Simple Name 0',

        username='User Full Name 0',

        auth_token_expires='1362585796000',

        auth_token_created='1359931796000',

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

        id=42,

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

    )

class FR(object):

    json = example

    headers = {}

    content = json

    status = None

    status_code = 200

astakos_pkg = 'kamaki.clients.astakos.AstakosClient'

class AstakosClient(TestCase):

    cached = False

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

    def setUp(self):

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

        self.token = 'ast@k0sT0k3n=='

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

    def tearDown(self):

        FR.json = example

    @patch('%s.get' % astakos_pkg, return_value=FR())

    def _authenticate(self, get):

        r = self.client.authenticate()

        self.assertEqual(get.mock_calls[-1], call('/tokens'))

        self.cached = True

        return r

    def test_authenticate(self):

        r = self._authenticate()

        self.assert_dicts_are_equal(r, example)

    def test_get_services(self):

        if not self.cached:

            self._authenticate()

        slist = self.client.get_services()

        self.assertEqual(slist, example['serviceCatalog'])

    def test_get_service_details(self):

        if not self.cached:

            self._authenticate()

        stype = '#FAIL'

        self.assertRaises(ClientError, self.client.get_service_details, stype)

        stype = 'compute'

        expected = [s for s in example['serviceCatalog'] if s['type'] == stype]

        self.assert_dicts_are_equal(

            self.client.get_service_details(stype), expected[0])

    def test_get_service_endpoints(self):

        if not self.cached:

            self._authenticate()

        stype, version = 'compute', 'V2'

        self.assertRaises(

            ClientError, self.client.get_service_endpoints, stype)

        expected = [s for s in example['serviceCatalog'] if s['type'] == stype]

        expected = [e for e in expected[0]['endpoints'] if (

            e['version_id'] == version.lower())]

        self.assert_dicts_are_equal(

            self.client.get_service_endpoints(stype, version), expected[0])

    def test_user_info(self):

        if not self.cached:

            self._authenticate()

        self.assertTrue(set(

            example['user'].keys()).issubset(self.client.user_info().keys()))

    def test_item(self):

        if not self.cached:

            self._authenticate()

        for term, val in example['user'].items():

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

        self.assertTrue(

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

    def test_list_users(self):

        if not self.cached:

            self._authenticate

        FR.json = example0

        self._authenticate()

        r = self.client.list_users()

        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(AstakosClient, 'AstakosClient', argv[1:])