from django.core.urlresolvers import reverse

from rest_framework.test import APITestCase

from rest_framework import status

from renkanmanager.models import Renkan, Workspace

from django.contrib.auth import get_user_model

import json

User = get_user_model()

class WorkspaceTests(APITestCase):

    def setUp(self):

        User.objects.create_superuser("blop", "blop@blop.com", "blop")

        self.user = User.objects.get_by_natural_key("blop")

        _ = self.client.login(username="blop", password="blop")

    def test_post_workspace(self):

        ###################################################

        # POSTing test Workspace

        ###################################################

        post_url = reverse("v1.0:workspace_list")

        test_title = "bleh_title"

        data = {"title": test_title}

        post_response = self.client.post(post_url, data, format="json")

        post_response_dict = json.loads(post_response.content.decode())

        self.assertEqual(post_response.status_code, status.HTTP_201_CREATED)

        # Checking data

        #    in the response

        self.assertEqual(post_response_dict.get("workspace_created_by"), getattr(self.user, User.USERNAME_FIELD))

        self.assertEqual(post_response_dict.get("title"), test_title)

        #    in the database

        created_workspace_guid = post_response_dict.get("id")

        created_workspace = Workspace.objects.get(workspace_guid=created_workspace_guid)

        self.assertEqual(getattr(created_workspace.creator, User.USERNAME_FIELD), getattr(self.user, User.USERNAME_FIELD))

        self.assertEqual(created_workspace.title, test_title)

        # Checking permission

        self.assertTrue(self.user.has_perm("view_workspace", created_workspace))

    def test_get_workspace_list(self):

        ###################################################

        # POSTing new Workspaces

        ###################################################

        post_url = reverse("v1.0:workspace_list")

        first_test_title = "bleh_title_2"

        second_test_title = "bleh_title_3"

        first_data = {"title": first_test_title}

        second_data = {"title": second_test_title}

        self.client.post(post_url, first_data, format="json")

        self.client.post(post_url, second_data, format="json")

        ###################################################

        # GETting Workspaces

        ###################################################

        get_url = post_url

        get_response = self.client.get(get_url, format=json)

        self.assertEqual(get_response.status_code, status.HTTP_200_OK)

        # Check data

        get_response_content = json.loads(get_response.content.decode()) # Should be a list

        self.assertTrue(isinstance(get_response_content, list))

        self.assertEqual(len(get_response_content), 2)

    def test_get_workspace(self):

        ###################################################

        # POSTing test Workspace

        ###################################################

        workspace_post_url = reverse("v1.0:workspace_list")

        test_title = "bleh_title"

        data = {"title": test_title}

        post_response = self.client.post(workspace_post_url, data, format="json")

        post_response_dict = json.loads(post_response.content.decode())

        test_workspace_guid = post_response_dict.get("id")

        ###################################################

        # POSTing new Renkans into test Workspace

        ###################################################

        renkan_post_url = reverse("v1.0:renkan_list_workspace", kwargs={"workspace_guid": test_workspace_guid})

        first_test_title = "bleh_title"

        second_test_title = "bleh_title_2"

        first_data = {"title": first_test_title}

        second_data = {"title": second_test_title}

        self.client.post(renkan_post_url, first_data, format="json")

        self.client.post(renkan_post_url, second_data, format="json")

        ###################################################

        # GETting test Workspace

        ###################################################

        get_url = reverse("v1.0:workspace_detail", kwargs={"workspace_guid": test_workspace_guid})

        get_response = self.client.get(get_url, format="json")

        self.assertEqual(get_response.status_code, status.HTTP_200_OK)

        # Checking GET data

        get_response_dict = json.loads(get_response.content.decode())

        self.assertEqual(test_workspace_guid, get_response_dict.get("id", ""))

        self.assertEqual(test_title, get_response_dict.get("title", ""))

        self.assertEqual(getattr(self.user, User.USERNAME_FIELD), get_response_dict.get("workspace_created_by", ""))

        self.assertEqual(2, get_response_dict.get("renkan_count"))

        ###################################################

        # GETting wrong workspace_guid

        ###################################################

        get_url = reverse("v1.0:workspace_detail", kwargs={"workspace_guid": "bleh-bad-workspace-id"})

        get_response = self.client.get(get_url, format="json")

        self.assertEqual(get_response.status_code, status.HTTP_404_NOT_FOUND)

    def test_update_workspace(self):

        ###################################################

        # POSTing test Workspace

        ###################################################

        workspace_post_url = reverse("v1.0:workspace_list")

        test_title = "bleh_title"

        data = {"title": test_title}

        post_response = self.client.post(workspace_post_url, data, format="json")

        post_response_dict = json.loads(post_response.content.decode())

        test_workspace_guid = post_response_dict.get("id")

        ###################################################

        # PUTting into test Workspace

        ###################################################

        put_url = reverse("v1.0:workspace_detail", kwargs={"workspace_guid": test_workspace_guid})

        edited_title = "bleh_title_2"

        put_data = {"title": edited_title}

        put_response = self.client.put(put_url, put_data, format="json")

        put_response_dict = json.loads(put_response.content.decode())

        self.assertEqual(put_response.status_code, status.HTTP_200_OK)

        # Checking data

        #    in the response

        self.assertEqual(put_response_dict.get("workspace_created_by"), getattr(self.user, User.USERNAME_FIELD))

        self.assertEqual(put_response_dict.get("title"), edited_title)

        self.assertEqual(put_response_dict.get("renkan_count"), 0)

        #    in the database

        edited_workspace_guid = put_response_dict.get("id")

        edited_workspace = Workspace.objects.get(workspace_guid=edited_workspace_guid)

        self.assertEqual(getattr(edited_workspace.creator, User.USERNAME_FIELD), getattr(self.user, User.USERNAME_FIELD))

        self.assertEqual(edited_workspace.title, edited_title)

        self.assertEqual(edited_workspace.renkan_count, 0)

    def test_delete_workspace(self):

        ###################################################

        # POSTing test Workspace

        ###################################################

        workspace_post_url = reverse("v1.0:workspace_list")

        test_title = "bleh_title"

        data = {"title": test_title}

        post_response = self.client.post(workspace_post_url, data, format="json")

        post_response_dict = json.loads(post_response.content.decode())

        to_delete_workspace_guid = post_response_dict.get("id", "")

        ###################################################

        # POSTing new Renkan into test Workspace

        ###################################################

        renkan_post_url = reverse("v1.0:renkan_list_workspace", kwargs={"workspace_guid": to_delete_workspace_guid})

        renkan_test_title = "bleh_title"

        renkan_test_data = {"title": renkan_test_title}

        post_response = self.client.post(renkan_post_url, renkan_test_data, format="json")

        to_delete_renkan_guid = json.loads(post_response.content.decode()).get("id", "")

        ###################################################

        # Try to DELETE test Workspace (should fail, it still contains a Renkan)

        ###################################################

        delete_url = reverse("v1.0:workspace_detail", kwargs={"workspace_guid":to_delete_workspace_guid})

        delete_response = self.client.delete(delete_url, {}, format="json")

        self.assertEqual(delete_response.status_code, status.HTTP_400_BAD_REQUEST)

        ###################################################

        # DELETE posted Renkan

        ###################################################

        delete_renkan_url = reverse("v1.0:renkan_detail", kwargs={"renkan_guid":to_delete_renkan_guid})

        delete_renkan_response = self.client.delete(delete_renkan_url, format="json")

        self.assertEqual(delete_renkan_response.status_code, status.HTTP_204_NO_CONTENT)

        ###################################################

        # DELETE test Workspace

        ###################################################

        delete_url = reverse("v1.0:workspace_detail", kwargs={"workspace_guid":to_delete_workspace_guid})

        delete_response = self.client.delete(delete_url, {}, format="json")

        self.assertEqual(delete_response.status_code, status.HTTP_204_NO_CONTENT)

        ###################################################

        # Try to DELETE workspace with wrong guid

        ###################################################

        delete_url = reverse("v1.0:workspace_detail", kwargs={"workspace_guid": "bad-workspace-guid"})

        delete_response = self.client.delete(delete_url, format=json)

        self.assertEqual(delete_response.status_code, status.HTTP_404_NOT_FOUND)