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(username="blop")
_ = self.client.login(username="blop", password="blop")
User.objects.create_user("blip", "blip@blip.com", "blip")
self.unauthorized_user = User.objects.get(username="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"), self.user.username)
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(created_workspace.creator.username, self.user.username)
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(self.user.username, 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"), self.user.username)
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(edited_workspace.creator.username, self.user.username)
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)