43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
# Django imports
|
|
from django.urls import reverse
|
|
|
|
# Third party import
|
|
from rest_framework import status
|
|
|
|
# Module imports
|
|
from .base import AuthenticatedAPITest
|
|
from plane.db.models import Workspace, WorkspaceMember
|
|
|
|
|
|
class WorkSpaceCreateReadUpdateDelete(AuthenticatedAPITest):
|
|
def setUp(self):
|
|
super().setUp()
|
|
|
|
def test_create_workspace(self):
|
|
|
|
url = reverse("workspace")
|
|
|
|
# Test with empty data
|
|
response = self.client.post(url, {}, format="json")
|
|
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
|
|
|
|
# Test with valid data
|
|
response = self.client.post(
|
|
url, {"name": "Plane", "slug": "pla-ne"}, format="json"
|
|
)
|
|
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
|
|
self.assertEqual(Workspace.objects.count(), 1)
|
|
# Check if the member is created
|
|
self.assertEqual(WorkspaceMember.objects.count(), 1)
|
|
|
|
# Check other values
|
|
workspace = Workspace.objects.get(pk=response.data["id"])
|
|
workspace_member = WorkspaceMember.objects.get(workspace=workspace, member_id=self.user_id)
|
|
self.assertEqual(workspace.owner_id, self.user_id)
|
|
self.assertEqual(workspace_member.role, 20)
|
|
|
|
# Create a already existing workspace
|
|
response = self.client.post(
|
|
url, {"name": "Plane", "slug": "pla-ne"}, format="json"
|
|
)
|
|
self.assertEqual(response.status_code, status.HTTP_410_GONE)
|