import unittest

from test_base import TestBase
from resources.testcase_models import TestcaseCreateRequest, \
    Testcase, Testcases, TestcaseUpdateRequest
from resources.project_models import ProjectCreateRequest, Project
from common.constants import HTTP_OK, HTTP_BAD_REQUEST, \
    HTTP_FORBIDDEN, HTTP_NOT_FOUND


__author__ = '__serena__'


class TestCaseBase(TestBase):
    def setUp(self):
        super(TestCaseBase, self).setUp()
        self.req_d = TestcaseCreateRequest('/cases/vping_1', 'vping_1', 'vping-ssh test')
        self.req_e = TestcaseCreateRequest('/cases/doctor_1', 'doctor_1', 'create doctor')
        self.update_d = TestcaseUpdateRequest('vping_1', 'vping-ssh test', 'functest')
        self.update_e = TestcaseUpdateRequest('doctor_1', 'create doctor', 'functest')
        self.get_res = Testcase
        self.list_res = Testcases
        self.update_res = Testcase
        self.basePath = '/projects/%s/cases'
        self.create_project()

    def assert_body(self, case, req=None):
        if not req:
            req = self.req_d
        self.assertEqual(case.name, req.name)
        self.assertEqual(case.description, req.description)
        self.assertEqual(case.url, req.url)
        self.assertIsNotNone(case._id)
        self.assertIsNotNone(case.creation_date)

    def assert_update_body(self, old, new, req=None):
        if not req:
            req = self.req_d
        self.assertEqual(new.name, req.name)
        self.assertEqual(new.description, req.description)
        self.assertEqual(new.url, old.url)
        self.assertIsNotNone(new._id)
        self.assertIsNotNone(new.creation_date)

    def create_project(self):
        req_p = ProjectCreateRequest('functest', 'vping-ssh test')
        self.create_help('/projects', req_p, Project)
        self.project = req_p.name

    def create_d(self):
        return super(TestCaseBase, self).create_d(self.project)

    def create_e(self):
        return super(TestCaseBase, self).create_e(self.project)

    def get(self, case=None):
        return super(TestCaseBase, self).get(self.project, case)

    def update(self, new=None, case=None):
        return super(TestCaseBase, self).update(new, self.project, case)

    def delete(self, case):
        return super(TestCaseBase, self).delete(self.project, case)


class TestCaseCreate(TestCaseBase):
    def test_noBody(self):
        (code, body) = self.create(None, 'vping')
        self.assertEqual(code, HTTP_BAD_REQUEST)

    def test_noProject(self):
        code, body = self.create(self.req_d, 'noProject')
        self.assertEqual(code, HTTP_FORBIDDEN)
        self.assertIn('Could not find project', body)

    def test_success(self):
        code, body = self.create_d()
        self.assertEqual(code, HTTP_OK)
        self.assert_create_body(body, None, self.project)

    def test_alreadyExist(self):
        self.create_d()
        code, body = self.create_d()
        self.assertEqual(code, HTTP_FORBIDDEN)
        self.assertIn('already exists', body)


class TestCaseGet(TestCaseBase):
    def test_notExist(self):
        code, body = self.get('notExist')
        self.assertEqual(code, HTTP_NOT_FOUND)

    def test_getOne(self):
        self.create_d()
        code, body = self.get(self.req_d.name)
        self.assertEqual(code, HTTP_OK)
        self.assert_body(body)

    def test_list(self):
        self.create_d()
        self.create_e()
        code, body = self.get()
        for case in body.testcases:
            if self.req_d.name == case.name:
                self.assert_body(case)
            else:
                self.assert_body(case, self.req_e)


class TestCaseUpdate(TestCaseBase):
    def test_noBody(self):
        code, _ = self.update(case='noBody')
        self.assertEqual(code, HTTP_BAD_REQUEST)

    def test_notFound(self):
        code, _ = self.update(self.update_e, 'notFound')
        self.assertEqual(code, HTTP_NOT_FOUND)

    def test_newNameExist(self):
        self.create_d()
        self.create_e()
        code, body = self.update(self.update_e, self.req_d.name)
        self.assertEqual(code, HTTP_FORBIDDEN)
        self.assertIn("already exists", body)

    def test_noUpdate(self):
        self.create_d()
        code, body = self.update(self.update_d, self.req_d.name)
        self.assertEqual(code, HTTP_FORBIDDEN)
        self.assertIn("Nothing to update", body)

    def test_success(self):
        self.create_d()
        code, body = self.get(self.req_d.name)
        _id = body._id

        code, body = self.update(self.update_e, self.req_d.name)
        self.assertEqual(code, HTTP_OK)
        self.assertEqual(_id, body._id)
        self.assert_update_body(self.req_d, body, self.update_e)

        _, new_body = self.get(self.req_e.name)
        self.assertEqual(_id, new_body._id)
        self.assert_update_body(self.req_d, new_body, self.update_e)


class TestCaseDelete(TestCaseBase):
    def test_notFound(self):
        code, body = self.delete('notFound')
        self.assertEqual(code, HTTP_NOT_FOUND)

    def test_success(self):
        self.create_d()
        code, body = self.delete(self.req_d.name)
        self.assertEqual(code, HTTP_OK)
        self.assertEqual(body, '')

        code, body = self.get(self.req_d.name)
        self.assertEqual(code, HTTP_NOT_FOUND)


if __name__ == '__main__':
    unittest.main()