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