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")
        User.objects.create_user("blip", "blip@blip.com", "blip")
        self.unauthorized_user = User.objects.get_by_natural_key("blip")
    
    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))
        
        ###################################################
        # try POSTing with unauthorized user 
        ###################################################
        
        _ = self.client.login(username="blip", password="blip")
        post_url = reverse("v1.0:workspace_list")
        data = {"title": "wont_work"}
        post_response = self.client.post(post_url, data, format="json")
        self.assertEqual(post_response.status_code, status.HTTP_403_FORBIDDEN)
        
     
    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)
        
        ###################################################
        # try GETting with unauthorized user 
        ###################################################
        
        _ = self.client.login(username="blip", password="blip")
        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_403_FORBIDDEN)
        
    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)
        
        ###################################################
        # try PUTting with unauthorized user 
        ###################################################
        
        _ = self.client.login(username="blip", password="blip")
        put_url = reverse("v1.0:workspace_detail", kwargs={"workspace_guid": test_workspace_guid})
        put_data = {
            "title" : "wont_work"
        }
        put_response = self.client.put(put_url, put_data, format="json")
        self.assertEqual(put_response.status_code, status.HTTP_403_FORBIDDEN)
        
    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)
        
        ###################################################
        # try to DELETE with unauthorized user 
        ###################################################
        
        _ = self.client.login(username="blip", password="blip")
        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_403_FORBIDDEN)
        
        # Restoring rightful user
        _ = self.client.login(username="blop", password="blop")
        
        ###################################################
        # 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)
        