# -*- coding: utf-8 -*-

"""

 (c) 2015 - Copyright Red Hat Inc

 Authors:

   Pierre-Yves Chibon <pingou@pingoured.fr></pingou@pingoured.fr>

"""

from __future__ import unicode_literals, absolute_import

import unittest

import shutil

import sys

import os

import json

from mock import patch, MagicMock

sys.path.insert(0, os.path.join(os.path.dirname(

    os.path.abspath(__file__)), '..'))

import pagure.api

import pagure.flask_app

import pagure.lib.query

import tests

class PagureFlaskApitests(tests.SimplePagureTest):

    """ Tests for flask API controller of pagure """

    maxDiff = None

    def test_api_doc(self):

        """ Test the API documentation page. """

        print(dir(self.app))

        output = self.app.get('/api/0/')

        output_text = output.get_data(as_text=True)

        self.assertIn(

            '<title> API | pagure  - Pagure</title>\n', output_text)

        self.assertIn(

            '  Pagure API Reference\n        \n', output_text)

    def test_api_doc_authenticated(self):

        """ Test the API documentation page. """

        user = tests.FakeUser(username='foo')

        with tests.user_set(self.app.application, user):

            output = self.app.get('/api/0/')

            output_text = output.get_data(as_text=True)

            self.assertIn(

                '<title> API | pagure  - Pagure</title>\n', output_text)

            self.assertIn(

                '  Pagure API Reference\n        \n', output_text)

    def test_api_get_request_data(self):

        data = {'foo': 'bar'}

        # test_request_context doesn't set flask.g, but some teardown

        # functions try to use that, so let's exclude them

        self._app.teardown_request_funcs = {}

        with self._app.test_request_context(

                '/api/0/version', method="POST", data=data):

            self.assertEqual(pagure.api.get_request_data()['foo'], 'bar')

        data = json.dumps(data)

        with self._app.test_request_context('/api/0/version', data=data,

                                            content_type='application/json'):

            self.assertEqual(pagure.api.get_request_data()['foo'], 'bar')

    def test_api_version_old_url(self):

        """ Test the api_version function.  """

        output = self.app.get('/api/0/version')

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(data['version'], pagure.__api_version__)

        self.assertEqual(sorted(data.keys()), ['version'])

    def test_api_version_new_url(self):

        """ Test the api_version function at its new url.  """

        output = self.app.get('/api/0/-/version')

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(data['version'], pagure.__api_version__)

        self.assertEqual(sorted(data.keys()), ['version'])

    def test_api_project_tags(self):

        """ Test the api_project_tags function.  """

        tests.create_projects(self.session)

        output = self.app.get('/api/0/foo/tags/')

        self.assertEqual(output.status_code, 404)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(set(data.keys()), set(['output', 'error']))

        self.assertEqual(data['output'], 'notok')

        self.assertEqual(data['error'], 'Project not found')

        output = self.app.get('/api/0/test/tags/')

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(sorted(data.keys()), ['tags', 'total_tags'])

        self.assertEqual(data['tags'], [])

        self.assertEqual(data['total_tags'], 0)

        # Add an issue and tag it so that we can list them

        item = pagure.lib.model.Issue(

            id=1,

            uid='foobar',

            project_id=1,

            title='issue',

            content='a bug report',

            user_id=1,  # pingou

        )

        self.session.add(item)

        self.session.commit()

        item = pagure.lib.model.TagColored(

            tag='tag1', tag_color='DeepBlueSky', project_id=1,

        )

        self.session.add(item)

        self.session.commit()

        item = pagure.lib.model.TagIssueColored(

            issue_uid='foobar',

            tag_id=item.id

        )

        self.session.add(item)

        self.session.commit()

        output = self.app.get('/api/0/test/tags/')

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(sorted(data.keys()), ['tags', 'total_tags'])

        self.assertEqual(data['tags'], ['tag1'])

        self.assertEqual(data['total_tags'], 1)

        output = self.app.get('/api/0/test/tags/?pattern=t')

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(sorted(data.keys()), ['tags', 'total_tags'])

        self.assertEqual(data['tags'], ['tag1'])

        self.assertEqual(data['total_tags'], 1)

        output = self.app.get('/api/0/test/tags/?pattern=p')

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(sorted(data.keys()), ['tags', 'total_tags'])

        self.assertEqual(data['tags'], [])

        self.assertEqual(data['total_tags'], 0)

    def test_api_groups(self):

        """ Test the api_groups function.  """

        # Add a couple of groups so that we can list them

        item = pagure.lib.model.PagureGroup(

            group_name='group1',

            group_type='user',

            display_name='User group',

            user_id=1,  # pingou

        )

        self.session.add(item)

        item = pagure.lib.model.PagureGroup(

            group_name='rel-eng',

            group_type='user',

            display_name='Release engineering group',

            user_id=1,  # pingou

        )

        self.session.add(item)

        self.session.commit()

        output = self.app.get('/api/0/groups')

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(data['groups'], ['group1', 'rel-eng'])

        self.assertEqual(

            sorted(data.keys()),

            ['groups', 'pagination', 'total_groups'])

        self.assertEqual(data['total_groups'], 2)

        output = self.app.get('/api/0/groups?pattern=re')

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(data['groups'], ['rel-eng'])

        self.assertEqual(

            sorted(data.keys()),

            ['groups', 'pagination', 'total_groups'])

        self.assertEqual(data['total_groups'], 1)

    def test_api_whoami_unauth(self):

        """ Test the api_whoami function. """

        output = self.app.post('/api/0/-/whoami')

        self.assertEqual(output.status_code, 401)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(

            data,

            {

                u'error': u'Invalid or expired token. Please visit '

                'http://localhost.localdomain/settings#api-keys to get or '

                'renew your API token.',

                u'error_code': u'EINVALIDTOK'

            }

        )

    def test_api_whoami_invalid_auth(self):

        """ Test the api_whoami function with an invalid token. """

        tests.create_projects(self.session)

        tests.create_tokens(self.session)

        headers = {'Authorization': 'token invalid'}

        output = self.app.post('/api/0/-/whoami', headers=headers)

        self.assertEqual(output.status_code, 401)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(

            data,

            {

                u'error': u'Invalid or expired token. Please visit '

                'http://localhost.localdomain/settings#api-keys to get or '

                'renew your API token.',

                u'error_code': u'EINVALIDTOK',

                u'errors': 'Invalid token',

            }

        )

    def test_api_whoami_auth(self):

        """ Test the api_whoami function with a valid token. """

        tests.create_projects(self.session)

        tests.create_tokens(self.session)

        headers = {'Authorization': 'token aaabbbcccddd'}

        output = self.app.post('/api/0/-/whoami', headers=headers)

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(data, {u'username': u'pingou'})

    def test_api_error_codes(self):

        """ Test the api_error_codes endpoint. """

        output = self.app.get('/api/0/-/error_codes')

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(len(data), 36)

        self.assertEqual(

            sorted(data.keys()),

            sorted([

                'EDATETIME',

                'EDBERROR',

                'EGITERROR',

                'EINVALIDISSUEFIELD',

                'EINVALIDISSUEFIELD_LINK',

                'EINVALIDPERPAGEVALUE',

                'EINVALIDPRIORITY',

                'EINVALIDREQ',

                'EINVALIDTOK',

                'EISSUENOTALLOWED',

                'EMODIFYPROJECTNOTALLOWED',

                'ENEWPROJECTDISABLED',

                'ENOCODE',

                'ENOCOMMENT',

                'ENOCOMMIT',

                'ENOGROUP',

                'ENOISSUE',

                'ENOPRCLOSE',

                'ENOPROJECT',

                'ENOPROJECTS',

                'ENOPRSTATS',

                'ENOREQ',

                'ENOSIGNEDOFF',

                'ENOTASSIGNED',

                'ENOTASSIGNEE',

                'ENOTHIGHENOUGH',

                'ENOTMAINADMIN',

                'ENOUSER',

                'EPRCONFLICTS',

                'EPRNOTALLOWED',

                'EPRSCORE',

                'EPULLREQUESTSDISABLED',

                'ETIMESTAMP',

                'ETRACKERDISABLED',

                'ETRACKERREADONLY',

                'EUBLOCKED',

            ])

        )

    @patch("pagure.lib.tasks.get_result")

    def test_api_task_status(self, get_result):

        async_result = MagicMock()

        async_result.status = "running"

        async_result.ready.return_value = False

        get_result.return_value = async_result

        output = self.app.get("/api/0/task/123abc/status/")

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(data, {"ready": False, "status": "running"})

        async_result = MagicMock()

        async_result.status = "finished"

        async_result.ready.return_value = True

        async_result.successful.return_value = True

        get_result.return_value = async_result

        output = self.app.get("/api/0/task/123abc/status/")

        self.assertEqual(output.status_code, 200)

        data = json.loads(output.get_data(as_text=True))

        self.assertEqual(

            data,

            {"ready": True, "status": "finished", "successful": True}

        )

if __name__ == '__main__':

    unittest.main(verbosity=2)