71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
from rest_framework.test import APITestCase
|
|
|
|
from apps.clients.models import Client
|
|
from apps.projects.models import Project
|
|
from apps.users.models import User
|
|
from apps.workspaces.models import Workspace, WorkspaceMembership
|
|
|
|
|
|
class ProjectViewTests(APITestCase):
|
|
@classmethod
|
|
def setUpTestData(cls):
|
|
cls.owner = User.objects.create_user(
|
|
mobile="09121110001",
|
|
password="secret123",
|
|
first_name="Owner",
|
|
)
|
|
cls.workspace = Workspace.objects.create(name="Projects", owner=cls.owner)
|
|
cls.member = User.objects.create_user(
|
|
mobile="09121110002",
|
|
password="secret123",
|
|
first_name="Member",
|
|
)
|
|
WorkspaceMembership.objects.create(
|
|
workspace=cls.workspace,
|
|
user=cls.member,
|
|
role=WorkspaceMembership.Role.MEMBER,
|
|
is_active=True,
|
|
)
|
|
cls.first_client = Client.objects.create(workspace=cls.workspace, name="Acme")
|
|
cls.second_client = Client.objects.create(workspace=cls.workspace, name="Globex")
|
|
cls.third_client = Client.objects.create(workspace=cls.workspace, name="Initech")
|
|
Project.objects.create(
|
|
workspace=cls.workspace,
|
|
client=cls.first_client,
|
|
name="Alpha",
|
|
)
|
|
Project.objects.create(
|
|
workspace=cls.workspace,
|
|
client=cls.second_client,
|
|
name="Beta",
|
|
)
|
|
Project.objects.create(
|
|
workspace=cls.workspace,
|
|
client=cls.third_client,
|
|
name="Gamma",
|
|
)
|
|
|
|
def test_project_list_supports_multi_client_filter(self):
|
|
self.client.force_authenticate(user=self.member)
|
|
|
|
response = self.client.get(
|
|
"/api/projects/",
|
|
[
|
|
("workspace", str(self.workspace.id)),
|
|
("clients", str(self.first_client.id)),
|
|
("clients", str(self.second_client.id)),
|
|
],
|
|
)
|
|
|
|
self.assertEqual(response.status_code, 200)
|
|
items = (
|
|
response.data
|
|
if isinstance(response.data, list)
|
|
else response.data.get("results") or response.data.get("items", [])
|
|
)
|
|
result_ids = {str(item["client"]["id"]) for item in items}
|
|
self.assertEqual(
|
|
result_ids,
|
|
{str(self.first_client.id), str(self.second_client.id)},
|
|
)
|