Synnefo

# Copyright 2011, 2012, 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 astakos.im.tests.common import *

from django.test import TestCase

from urllib import quote

from urlparse import urlparse, parse_qs

from xml.dom import minidom

import json

ROOT = '/astakos/api/'

u = lambda url: ROOT + url

class QuotaAPITest(TestCase):

    def test_0(self):

        client = Client()

        component1 = Component.objects.create(name="comp1")

        register.add_service(component1, "service1", "type1", [])

        # custom service resources

        resource11 = {"name": "service1.resource11",

                      "desc": "resource11 desc",

                      "service_type": "type1",

                      "allow_in_projects": True}

        r, _ = register.add_resource(resource11)

        register.update_resource(r, 100)

        resource12 = {"name": "service1.resource12",

                      "desc": "resource11 desc",

                      "service_type": "type1",

                      "unit": "bytes"}

        r, _ = register.add_resource(resource12)

        register.update_resource(r, 1024)

        # create user

        user = get_local_user('test@grnet.gr')

        quotas.qh_sync_user(user)

        component2 = Component.objects.create(name="comp2")

        register.add_service(component2, "service2", "type2", [])

        # create another service

        resource21 = {"name": "service2.resource21",

                      "desc": "resource11 desc",

                      "service_type": "type2",

                      "allow_in_projects": False}

        r, _ = register.add_resource(resource21)

        register.update_resource(r, 3)

        resource_names = [r['name'] for r in

                          [resource11, resource12, resource21]]

        # get resources

        r = client.get(u('resources'))

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        for name in resource_names:

            assertIn(name, body)

        # get quota

        r = client.get(u('quotas'))

        self.assertEqual(r.status_code, 401)

        headers = {'HTTP_X_AUTH_TOKEN': user.auth_token}

        r = client.get(u('quotas/'), **headers)

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        system_quota = body['system']

        assertIn('system', body)

        for name in resource_names:

            assertIn(name, system_quota)

        r = client.get(u('service_quotas'))

        self.assertEqual(r.status_code, 401)

        s1_headers = {'HTTP_X_AUTH_TOKEN': component1.auth_token}

        r = client.get(u('service_quotas'), **s1_headers)

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        assertIn(user.uuid, body)

        r = client.get(u('commissions'), **s1_headers)

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        self.assertEqual(body, [])

        # issue some commissions

        commission_request = {

            "force": False,

            "auto_accept": False,

            "name": "my commission",

            "provisions": [

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource11['name'],

                    "quantity": 1

                },

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource12['name'],

                    "quantity": 30000

                }]}

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 413)

        commission_request = {

            "force": False,

            "auto_accept": False,

            "name": "my commission",

            "provisions": [

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource11['name'],

                    "quantity": 1

                },

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource12['name'],

                    "quantity": 100

                }]}

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 201)

        body = json.loads(r.content)

        serial = body['serial']

        self.assertEqual(serial, 1)

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 201)

        body = json.loads(r.content)

        self.assertEqual(body['serial'], 2)

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 201)

        body = json.loads(r.content)

        self.assertEqual(body['serial'], 3)

        r = client.get(u('commissions'), **s1_headers)

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        self.assertEqual(body, [1, 2, 3])

        r = client.get(u('commissions/' + str(serial)), **s1_headers)

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        self.assertEqual(body['serial'], serial)

        assertIn('issue_time', body)

        self.assertEqual(body['provisions'], commission_request['provisions'])

        self.assertEqual(body['name'], commission_request['name'])

        r = client.get(u('service_quotas?user=' + user.uuid), **s1_headers)

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        user_quota = body[user.uuid]

        system_quota = user_quota['system']

        r11 = system_quota[resource11['name']]

        self.assertEqual(r11['usage'], 3)

        self.assertEqual(r11['pending'], 3)

        # resolve pending commissions

        resolve_data = {

            "accept": [1, 3],

            "reject": [2, 3, 4],

        }

        post_data = json.dumps(resolve_data)

        r = client.post(u('commissions/action'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        self.assertEqual(body['accepted'], [1])

        self.assertEqual(body['rejected'], [2])

        failed = body['failed']

        self.assertEqual(len(failed), 2)

        r = client.get(u('commissions/' + str(serial)), **s1_headers)

        self.assertEqual(r.status_code, 404)

        # auto accept

        commission_request = {

            "auto_accept": True,

            "name": "my commission",

            "provisions": [

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource11['name'],

                    "quantity": 1

                },

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource12['name'],

                    "quantity": 100

                }]}

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 201)

        body = json.loads(r.content)

        serial = body['serial']

        self.assertEqual(serial, 4)

        r = client.get(u('commissions/' + str(serial)), **s1_headers)

        self.assertEqual(r.status_code, 404)

        # malformed

        commission_request = {

            "auto_accept": True,

            "name": "my commission",

            "provisions": [

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource11['name'],

                }

            ]}

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 400)

        commission_request = {

            "auto_accept": True,

            "name": "my commission",

            "provisions": "dummy"}

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 400)

        r = client.post(u('commissions'), commission_request,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 400)

        # no holding

        commission_request = {

            "auto_accept": True,

            "name": "my commission",

            "provisions": [

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": "non existent",

                    "quantity": 1

                },

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource12['name'],

                    "quantity": 100

                }]}

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 404)

        # release

        commission_request = {

            "provisions": [

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource11['name'],

                    "quantity": -1

                }

            ]}

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 201)

        body = json.loads(r.content)

        serial = body['serial']

        accept_data = {'accept': ""}

        post_data = json.dumps(accept_data)

        r = client.post(u('commissions/' + str(serial) + '/action'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 200)

        reject_data = {'reject': ""}

        post_data = json.dumps(accept_data)

        r = client.post(u('commissions/' + str(serial) + '/action'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 404)

        # force

        commission_request = {

            "force": True,

            "provisions": [

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource11['name'],

                    "quantity": 100

                }]}

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 201)

        commission_request = {

            "force": True,

            "provisions": [

                {

                    "holder": user.uuid,

                    "source": "system",

                    "resource": resource11['name'],

                    "quantity": -200

                }]}

        post_data = json.dumps(commission_request)

        r = client.post(u('commissions'), post_data,

                        content_type='application/json', **s1_headers)

        self.assertEqual(r.status_code, 413)

        r = client.get(u('quotas'), **headers)

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        system_quota = body['system']

        r11 = system_quota[resource11['name']]

        self.assertEqual(r11['usage'], 102)

        self.assertEqual(r11['pending'], 101)

class TokensApiTest(TestCase):

    def setUp(self):

        backend = activation_backends.get_backend()

        self.user1 = AstakosUser.objects.create(

            email='test1', email_verified=True, moderated=True,

            is_rejected=False)

        backend.activate_user(self.user1)

        assert self.user1.is_active is True

        self.user2 = AstakosUser.objects.create(

            email='test2', email_verified=True, moderated=True,

            is_rejected=False)

        backend.activate_user(self.user2)

        assert self.user2.is_active is True

        Service(name='service1', url='http://localhost/service1',

                api_url='http://localhost/api/service1',

                type='service1').save()

        Service(name='service2', url='http://localhost/service2',

                api_url='http://localhost/api/service2',

                type='service2').save()

        Service(name='service3', url='http://localhost/service3',

                api_url='http://localhost/api/service3',

                type='service3').save()

    def test_authenticate(self):

        client = Client()

        # Check not allowed method

        url = '/astakos/api/tokens'

        r = client.get(url, post_data={})

        self.assertEqual(r.status_code, 400)

        # Malformed request

        url = '/astakos/api/tokens'

        r = client.post(url, post_data={})

        self.assertEqual(r.status_code, 400)

        # Check unsupported xml input

        url = '/astakos/api/tokens'

        post_data = """

            <?xml version="1.0" encoding="UTF-8"?>

                <auth xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"

                 xmlns="http://docs.openstack.org/identity/api/v2.0"

                 tenantName="%s">

                  <passwordCredentials username="%s" password="%s"/>

                </auth>""" % (self.user1.uuid, self.user1.uuid,

                              self.user1.auth_token)

        r = client.post(url, post_data, content_type='application/xml')

        self.assertEqual(r.status_code, 400)

        body = json.loads(r.content)

        self.assertEqual(body['badRequest']['message'],

                         "Unsupported Content-type: 'application/xml'")

        # Check malformed request: missing password

        url = '/astakos/api/tokens'

        post_data = """{"auth":{"passwordCredentials":{"username":"%s"},

                                "tenantName":"%s"}}""" % (

            self.user1.uuid, self.user1.uuid)

        r = client.post(url, post_data, content_type='application/json')

        self.assertEqual(r.status_code, 400)

        body = json.loads(r.content)

        self.assertEqual(body['badRequest']['message'],

                         'Malformed request')

        # Check malformed request: missing username

        url = '/astakos/api/tokens'

        post_data = """{"auth":{"passwordCredentials":{"password":"%s"},

                                "tenantName":"%s"}}""" % (

            self.user1.auth_token, self.user1.uuid)

        r = client.post(url, post_data, content_type='application/json')

        self.assertEqual(r.status_code, 400)

        body = json.loads(r.content)

        self.assertEqual(body['badRequest']['message'],

                         'Malformed request')

        # Check invalid pass

        url = '/astakos/api/tokens'

        post_data = """{"auth":{"passwordCredentials":{"username":"%s",

                                                       "password":"%s"},

                                "tenantName":"%s"}}""" % (

            self.user1.uuid, '', self.user1.uuid)

        r = client.post(url, post_data, content_type='application/json')

        self.assertEqual(r.status_code, 401)

        body = json.loads(r.content)

        self.assertEqual(body['unauthorized']['message'],

                         'Invalid token')

        # Check inconsistent pass

        url = '/astakos/api/tokens'

        post_data = """{"auth":{"passwordCredentials":{"username":"%s",

                                                       "password":"%s"},

                                "tenantName":"%s"}}""" % (

            self.user1.uuid, self.user2.auth_token, self.user2.uuid)

        r = client.post(url, post_data, content_type='application/json')

        self.assertEqual(r.status_code, 401)

        body = json.loads(r.content)

        self.assertEqual(body['unauthorized']['message'],

                         'Invalid credentials')

        # Check invalid json data

        url = '/astakos/api/tokens'

        r = client.post(url, "not json", content_type='application/json')

        self.assertEqual(r.status_code, 400)

        body = json.loads(r.content)

        self.assertEqual(body['badRequest']['message'], 'Invalid JSON data')

        # Check auth with token

        url = '/astakos/api/tokens'

        post_data = """{"auth":{"token": {"id":"%s"},

                        "tenantName":"%s"}}""" % (

            self.user1.auth_token, self.user1.uuid)

        r = client.post(url, post_data, content_type='application/json')

        self.assertEqual(r.status_code, 200)

        self.assertTrue(r['Content-Type'].startswith('application/json'))

        try:

            body = json.loads(r.content)

        except Exception, e:

            self.fail(e)

        # Check successful json response

        url = '/astakos/api/tokens'

        post_data = """{"auth":{"passwordCredentials":{"username":"%s",

                                                       "password":"%s"},

                                "tenantName":"%s"}}""" % (

            self.user1.uuid, self.user1.auth_token, self.user1.uuid)

        r = client.post(url, post_data, content_type='application/json')

        self.assertEqual(r.status_code, 200)

        self.assertTrue(r['Content-Type'].startswith('application/json'))

        try:

            body = json.loads(r.content)

        except Exception, e:

            self.fail(e)

        try:

            token = body['token']['id']

            user = body['user']['id']

            service_catalog = body['serviceCatalog']

        except KeyError:

            self.fail('Invalid response')

        self.assertEqual(token, self.user1.auth_token)

        self.assertEqual(user, self.user1.uuid)

        self.assertEqual(len(service_catalog), 3)

        # Check successful xml response

        url = '/astakos/api/tokens'

        headers = {'HTTP_ACCEPT': 'application/xml'}

        post_data = """{"auth":{"passwordCredentials":{"username":"%s",

                                                       "password":"%s"},

                                "tenantName":"%s"}}""" % (

            self.user1.uuid, self.user1.auth_token, self.user1.uuid)

        r = client.post(url, post_data, content_type='application/json',

                        **headers)

        self.assertEqual(r.status_code, 200)

        self.assertTrue(r['Content-Type'].startswith('application/xml'))

#        try:

#            body = minidom.parseString(r.content)

#        except Exception, e:

#            self.fail(e)

    def test_get_endpoints(self):

        client = Client()

        # Check in active user token

        inactive_user = AstakosUser.objects.create(email='test3')

        url = '/astakos/api/tokens/%s/endpoints' % quote(

            inactive_user.auth_token)

        r = client.get(url)

        self.assertEqual(r.status_code, 401)

        # Check invalid user token in path

        url = '/astakos/api/tokens/nouser/endpoints'

        r = client.get(url)

        self.assertEqual(r.status_code, 401)

        # Check forbidden

        url = '/astakos/api/tokens/%s/endpoints' % quote(self.user1.auth_token)

        headers = {'HTTP_X_AUTH_TOKEN': AstakosUser.objects.create(

            email='test4').auth_token}

        r = client.get(url, **headers)

        self.assertEqual(r.status_code, 401)

        # Check bad request method

        url = '/astakos/api/tokens/%s/endpoints' % quote(self.user1.auth_token)

        r = client.post(url)

        self.assertEqual(r.status_code, 400)

        # Check forbidden

        url = '/astakos/api/tokens/%s/endpoints' % quote(self.user1.auth_token)

        headers = {'HTTP_X_AUTH_TOKEN': self.user2.auth_token}

        r = client.get(url, **headers)

        self.assertEqual(r.status_code, 403)

        # Check belongsTo BadRequest

        url = '/astakos/api/tokens/%s/endpoints?belongsTo=%s' % (

            quote(self.user1.auth_token), quote(self.user2.uuid))

        headers = {'HTTP_X_AUTH_TOKEN': self.user1.auth_token}

        r = client.get(url, **headers)

        self.assertEqual(r.status_code, 400)

        # Check successful request

        url = '/astakos/api/tokens/%s/endpoints' % quote(self.user1.auth_token)

        headers = {'HTTP_X_AUTH_TOKEN': self.user1.auth_token}

        r = client.get(url, **headers)

        self.assertEqual(r.status_code, 200)

        self.assertEqual(r['Content-Type'], 'application/json; charset=UTF-8')

        try:

            body = json.loads(r.content)

        except:

            self.fail('json format expected')

        endpoints = body.get('endpoints')

        self.assertEqual(len(endpoints), 3)

         # Check xml serialization

        url = '/astakos/api/tokens/%s/endpoints?format=xml' %\

            quote(self.user1.auth_token)

        headers = {'HTTP_X_AUTH_TOKEN': self.user1.auth_token}

        r = client.get(url, **headers)

        self.assertEqual(r.status_code, 200)

        self.assertEqual(r['Content-Type'], 'application/xml; charset=UTF-8')

#        try:

#            body = minidom.parseString(r.content)

#        except Exception, e:

#            self.fail('xml format expected')

        endpoints = body.get('endpoints')

        self.assertEqual(len(endpoints), 3)

        # Check limit

        url = '/astakos/api/tokens/%s/endpoints?limit=2' %\

            quote(self.user1.auth_token)

        headers = {'HTTP_X_AUTH_TOKEN': self.user1.auth_token}

        r = client.get(url, **headers)

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        endpoints = body.get('endpoints')

        self.assertEqual(len(endpoints), 2)

        endpoint_link = body.get('endpoint_links', [])[0]

        next = endpoint_link.get('href')

        p = urlparse(next)

        params = parse_qs(p.query)

        self.assertTrue('limit' in params)

        self.assertTrue('marker' in params)

        self.assertEqual(params['marker'][0], '2')

        # Check marker

        headers = {'HTTP_X_AUTH_TOKEN': self.user1.auth_token}

        r = client.get(next, **headers)

        self.assertEqual(r.status_code, 200)

        body = json.loads(r.content)

        endpoints = body.get('endpoints')

        self.assertEqual(len(endpoints), 1)