# -----------------------------------------------------------------------------
# Copyright (c) 2014--, The Qiita Development Team.
#
# Distributed under the terms of the BSD 3-clause License.
#
# The full license is in the file LICENSE, distributed with this software.
# -----------------------------------------------------------------------------

from unittest import main

from tornado.escape import json_decode

from qiita_db.study import StudyPerson
from qiita_pet.test.rest.test_base import RESTHandlerTestCase


class StudyPersonHandlerTests(RESTHandlerTestCase):
    def test_get_list(self):
        exp = [{'name': 'LabDude', 'affiliation': 'knight lab'},
               {'name': 'empDude', 'affiliation': 'broad'},
               {'name': 'PIDude', 'affiliation': 'Wash U'}]
        response = self.get('/api/v1/person', headers=self.headers)
        self.assertEqual(response.code, 200)
        obs = json_decode(response.body)
        self.assertCountEqual(obs, exp)

    def test_exist(self):
        exp = {'email': 'lab_dude@foo.bar', 'phone': '121-222-3333',
               'address': '123 lab street', 'id': 1}
        response = self.get('/api/v1/person?name=LabDude&'
                            'affiliation=knight%20lab', headers=self.headers)
        self.assertEqual(response.code, 200)
        obs = json_decode(response.body)
        self.assertEqual(obs, exp)

    def test_get_does_not_exist(self):
        exp = {'message': 'Person not found'}
        response = self.get('/api/v1/person?name=Boaty%20McBoatFace&'
                            'affiliation=UCSD', headers=self.headers)
        self.assertEqual(response.code, 404)
        obs = json_decode(response.body)
        self.assertEqual(obs, exp)

    def test_get_does_not_exist_affiliation(self):
        exp = {'message': 'Person not found'}
        response = self.get('/api/v1/person?name=LabDude%20&affiliation=UCSD',
                            headers=self.headers)
        self.assertEqual(response.code, 404)
        obs = json_decode(response.body)
        self.assertEqual(obs, exp)

    def test_get_invalid_query_string(self):
        response = self.get('/api/v1/person?name=LabDude',
                            headers=self.headers)
        self.assertEqual(response.code, 400)

    def test_get_invalid_query_string_2(self):
        response = self.get('/api/v1/person?affiliation=knight%20lab',
                            headers=self.headers)
        self.assertEqual(response.code, 400)

    def test_get_valid_extra_arguments(self):
        exp = {'email': 'lab_dude@foo.bar', 'phone': '121-222-3333',
               'address': '123 lab street', 'id': 1}
        response = self.get('/api/v1/person?name=LabDude&'
                            'affiliation=knight%20lab&foo=bar',
                            headers=self.headers)
        self.assertEqual(response.code, 200)
        obs = json_decode(response.body)
        self.assertEqual(obs, exp)

    def test_post_new_person(self):
        body = {'name': 'Boaty McBoatFace', 'affiliation': 'UCSD',
                'email': 'boat@ucsd.edu', 'phone': '720-876-5309'}

        response = self.post('/api/v1/person', data=body, headers=self.headers)
        self.assertEqual(response.code, 201)
        obs = json_decode(response.body)
        exp = StudyPerson.from_name_and_affiliation(body['name'],
                                                    body['affiliation']).id
        self.assertEqual(exp, obs['id'])

    def test_post_existing(self):
        body = {'name': 'LabDude', 'affiliation': 'knight lab',
                'email': 'lab_dude@foo.bar', 'phone': '121-222-3333'}

        response = self.post('/api/v1/person', data=body, headers=self.headers)
        self.assertEqual(response.code, 409)
        obs = json_decode(response.body)
        exp = {'message': 'Person already exists'}
        self.assertEqual(exp, obs)

    def test_post_incomplete_details(self):
        body = {'affiliation': 'knight lab',
                'email': 'lab_dude@foo.bar', 'phone': '121-222-3333'}

        response = self.post('/api/v1/person', data=body, headers=self.headers)
        self.assertEqual(response.code, 400)


if __name__ == '__main__':
    main()