diff --git a/backend/secfit/comments/tests.py b/backend/secfit/comments/tests.py index 35ba93a7eece92bd8732a520160b21a9e144efe9..367af6105c6903d2277578fdf964a57899709ff6 100644 --- a/backend/secfit/comments/tests.py +++ b/backend/secfit/comments/tests.py @@ -12,18 +12,27 @@ class CommentsTestCase(TestCase): self.user_1 = User.objects.get(id="1") self.client = APIClient() self.client.force_authenticate(user=self.user_1) + User.objects.create(id="2",username="Bill2",password="secret", email="hei1") + self.user_2 = User.objects.get(id="2") + self.client2 = APIClient() + self.client2.force_authenticate(user=self.user_2) self.commentURL = "http://testserver/api/comments/" self.workout1URL = "http://testserver/api/workouts/1/" - def testPUComment(self): + def testPostComment(self): Workout.objects.create(id="1",name="workout",date=timezone.now(),owner=self.user_1, visibility="PU") post = self.client.post(self.commentURL,({"workout":self.workout1URL,"content":"asd"}),format='json') self.assertEquals(post.status_code,201) def testGetComments(self): Workout.objects.create(id="1",name="workout",date=timezone.now(),owner=self.user_1, visibility="PU") + Workout.objects.create(id="2",name="workout",date=timezone.now(),owner=self.user_2, visibility="PR") post = self.client.post(self.commentURL,({"workout":self.workout1URL,"content":"asd"}),format='json') self.assertEquals(post.status_code,201) - get = self.client.get("http://testserver/api/comments/") - datadict = dict(get.data) - self.assertEquals(len(datadict["results"]), 1) \ No newline at end of file + self.client2.post(self.commentURL,({"workout":"http://testserver/api/workouts/2/","content":"assdsdd"}),format='json') + user1get = self.client.get("http://testserver/api/comments/") + datadict = dict(user1get.data) + self.assertEquals(len(datadict["results"]), 1) + user2get = self.client2.get("http://testserver/api/comments/") + datadict2 = dict(user2get.data) + self.assertEquals(len(datadict2["results"]), 2) \ No newline at end of file