api/users/tests.py at 0b8bc05f98cd558128791009f56bb74e4caeeb44 · PROCOLLAB-github/api · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
from django.test import TestCase
from rest_framework.test import APIRequestFactory, force_authenticate
from tests.constants import USER_CREATE_DATA

from projects.models import Collaborator, Project
from users.models import CustomUser
from users.serializers import UserDetailSerializer
from users.views import CurrentUser, UserLeaderProjectsList, UserList, UserDetail


class UserTestCase(TestCase):
    def setUp(self):
        self.factory = APIRequestFactory()
        self.user_list_view = UserList.as_view()
        self.user_detail_view = UserDetail.as_view()
        self.user_leader_projects_view = UserLeaderProjectsList.as_view()
        self.current_user_view = CurrentUser.as_view()

    def test_user_creation(self):
        request = self.factory.post("auth/users/", USER_CREATE_DATA)
        response = self.user_list_view(request)
        self.assertEqual(response.status_code, 201)
        self.assertEqual(response.data["email"], "only_for_test@test.test")
        self.assertEqual(response.data["is_active"], False)

    def test_user_creation_with_wrong_data(self):
        request = self.factory.post(
            "auth/users/",
            {
                "email": "qwe",
                "password": "qwe",
                "first_name": "qwe",
                "last_name": "qwe",
            },
        )
        response = self.user_list_view(request)
        self.assertEqual(response.status_code, 400)

    def test_user_creation_with_existing_email(self):
        request = self.factory.post("auth/users/", USER_CREATE_DATA)
        response = self.user_list_view(request)
        self.assertEqual(response.status_code, 201)
        response = self.user_list_view(request)
        self.assertEqual(response.status_code, 400)

    def test_user_update(self):
        request = self.factory.post("auth/users/", USER_CREATE_DATA)
        response = self.user_list_view(request)
        user_id = response.data["id"]
        user = CustomUser.objects.get(id=user_id)

        request = self.factory.get(f"auth/users/{user.pk}/")
        response = self.user_detail_view(request, pk=user.pk)
        self.assertEqual(response.status_code, 401)  # Unauthorized

        force_authenticate(request, user=user)
        response = self.user_detail_view(request, pk=user.pk)
        self.assertEqual(response.status_code, 200)

        request = self.factory.patch(f"auth/users/{user.pk}/", {"first_name": "Сергей"})
        force_authenticate(request, user=user)
        response = self.user_detail_view(request, pk=user.pk)
        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.data["first_name"], "Сергей")

    def test_user_leader_projects_list(self):
        leader = self._user_create("leader@example.com")
        collaborator = self._user_create("collaborator@example.com")

        leader_project = Project.objects.create(name="Leader project", leader=leader)
        second_leader_project = Project.objects.create(
            name="Leader project 2", leader=leader, draft=False
        )
        other_project = Project.objects.create(name="Other project", leader=collaborator)
        Collaborator.objects.create(user=leader, project=other_project, role="Member")

        request = self.factory.get("users/projects/leader/")
        force_authenticate(request, user=leader)
        response = self.user_leader_projects_view(request)

        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.data["count"], 2)
        returned_ids = {item["id"] for item in response.data["results"]}
        self.assertSetEqual(
            returned_ids, {leader_project.id, second_leader_project.id}
        )

    def test_current_user_returns_authenticated_user_profile(self):
        user = self._user_create("current@example.com")

        request = self.factory.get("auth/users/current/")
        force_authenticate(request, user=user)
        response = self.current_user_view(request)
        expected_data = UserDetailSerializer(user, context={"request": request}).data

        self.assertEqual(response.status_code, 200)
        self.assertEqual(response.data, expected_data)

    def test_removed_legacy_routes_return_404(self):
        self.assertEqual(self.client.get("/auth/users/clone-data").status_code, 404)
        self.assertEqual(self.client.get("/auth/subscription/").status_code, 404)
        self.assertEqual(self.client.post("/auth/subscription/buy/").status_code, 404)

    def _user_create(self, email):
        tmp_create_data = USER_CREATE_DATA.copy()
        tmp_create_data["email"] = email
        request = self.factory.post("auth/users/", tmp_create_data)
        response = self.user_list_view(request)
        user_id = response.data["id"]
        user = CustomUser.objects.get(id=user_id)
        user.is_active = True
        user.save()
        return user