9 |
9 |
10 class WorkspaceTests(APITestCase): |
10 class WorkspaceTests(APITestCase): |
11 |
11 |
12 def setUp(self): |
12 def setUp(self): |
13 User.objects.create_superuser("blop", "blop@blop.com", "blop") |
13 User.objects.create_superuser("blop", "blop@blop.com", "blop") |
14 self.user = User.objects.get(username="blop") |
14 self.user = User.objects.get_by_natural_key("blop") |
15 _ = self.client.login(username="blop", password="blop") |
15 _ = self.client.login(username="blop", password="blop") |
16 User.objects.create_user("blip", "blip@blip.com", "blip") |
16 User.objects.create_user("blip", "blip@blip.com", "blip") |
17 self.unauthorized_user = User.objects.get(username="blip") |
17 self.unauthorized_user = User.objects.get_by_natural_key("blip") |
18 |
18 |
19 def test_post_workspace(self): |
19 def test_post_workspace(self): |
20 |
20 |
21 ################################################### |
21 ################################################### |
22 # POSTing test Workspace |
22 # POSTing test Workspace |
29 post_response_dict = json.loads(post_response.content.decode()) |
29 post_response_dict = json.loads(post_response.content.decode()) |
30 self.assertEqual(post_response.status_code, status.HTTP_201_CREATED) |
30 self.assertEqual(post_response.status_code, status.HTTP_201_CREATED) |
31 |
31 |
32 # Checking data |
32 # Checking data |
33 # in the response |
33 # in the response |
34 self.assertEqual(post_response_dict.get("workspace_created_by"), self.user.username) |
34 self.assertEqual(post_response_dict.get("workspace_created_by"), getattr(self.user, User.USERNAME_FIELD)) |
35 self.assertEqual(post_response_dict.get("title"), test_title) |
35 self.assertEqual(post_response_dict.get("title"), test_title) |
36 # in the database |
36 # in the database |
37 created_workspace_guid = post_response_dict.get("id") |
37 created_workspace_guid = post_response_dict.get("id") |
38 created_workspace = Workspace.objects.get(workspace_guid=created_workspace_guid) |
38 created_workspace = Workspace.objects.get(workspace_guid=created_workspace_guid) |
39 self.assertEqual(created_workspace.creator.username, self.user.username) |
39 self.assertEqual(getattr(created_workspace.creator, User.USERNAME_FIELD), getattr(self.user, User.USERNAME_FIELD)) |
40 self.assertEqual(created_workspace.title, test_title) |
40 self.assertEqual(created_workspace.title, test_title) |
41 |
41 |
42 # Checking permission |
42 # Checking permission |
43 self.assertTrue(self.user.has_perm("view_workspace", created_workspace)) |
43 self.assertTrue(self.user.has_perm("view_workspace", created_workspace)) |
44 |
44 |
115 |
115 |
116 # Checking GET data |
116 # Checking GET data |
117 get_response_dict = json.loads(get_response.content.decode()) |
117 get_response_dict = json.loads(get_response.content.decode()) |
118 self.assertEqual(test_workspace_guid, get_response_dict.get("id", "")) |
118 self.assertEqual(test_workspace_guid, get_response_dict.get("id", "")) |
119 self.assertEqual(test_title, get_response_dict.get("title", "")) |
119 self.assertEqual(test_title, get_response_dict.get("title", "")) |
120 self.assertEqual(self.user.username, get_response_dict.get("workspace_created_by", "")) |
120 self.assertEqual(getattr(self.user, User.USERNAME_FIELD), get_response_dict.get("workspace_created_by", "")) |
121 self.assertEqual(2, get_response_dict.get("renkan_count")) |
121 self.assertEqual(2, get_response_dict.get("renkan_count")) |
122 |
122 |
123 ################################################### |
123 ################################################### |
124 # GETting wrong workspace_guid |
124 # GETting wrong workspace_guid |
125 ################################################### |
125 ################################################### |
161 put_response_dict = json.loads(put_response.content.decode()) |
161 put_response_dict = json.loads(put_response.content.decode()) |
162 self.assertEqual(put_response.status_code, status.HTTP_200_OK) |
162 self.assertEqual(put_response.status_code, status.HTTP_200_OK) |
163 |
163 |
164 # Checking data |
164 # Checking data |
165 # in the response |
165 # in the response |
166 self.assertEqual(put_response_dict.get("workspace_created_by"), self.user.username) |
166 self.assertEqual(put_response_dict.get("workspace_created_by"), getattr(self.user, User.USERNAME_FIELD)) |
167 self.assertEqual(put_response_dict.get("title"), edited_title) |
167 self.assertEqual(put_response_dict.get("title"), edited_title) |
168 self.assertEqual(put_response_dict.get("renkan_count"), 0) |
168 self.assertEqual(put_response_dict.get("renkan_count"), 0) |
169 # in the database |
169 # in the database |
170 edited_workspace_guid = put_response_dict.get("id") |
170 edited_workspace_guid = put_response_dict.get("id") |
171 edited_workspace = Workspace.objects.get(workspace_guid=edited_workspace_guid) |
171 edited_workspace = Workspace.objects.get(workspace_guid=edited_workspace_guid) |
172 self.assertEqual(edited_workspace.creator.username, self.user.username) |
172 self.assertEqual(getattr(edited_workspace.creator, User.USERNAME_FIELD), getattr(self.user, User.USERNAME_FIELD)) |
173 self.assertEqual(edited_workspace.title, edited_title) |
173 self.assertEqual(edited_workspace.title, edited_title) |
174 self.assertEqual(edited_workspace.renkan_count, 0) |
174 self.assertEqual(edited_workspace.renkan_count, 0) |
175 |
175 |
176 ################################################### |
176 ################################################### |
177 # try PUTting with unauthorized user |
177 # try PUTting with unauthorized user |