+13

upvcarshare/notifications/tests/factories.py

reviewed

··· 1 1 + # -*- coding: utf-8 -*- 2 2 + from __future__ import unicode_literals, print_function, absolute_import 3 3 + 4 4 + import factory 5 5 + 6 6 + from users.tests.factories import UserFactory 7 7 + 8 8 + 9 9 + class NotificationFactory(factory.django.DjangoModelFactory): 10 10 + user = factory.SubFactory(UserFactory) 11 11 + 12 12 + class Meta: 13 13 + model = "notifications.Notification"

+47

upvcarshare/notifications/tests/test_api.py

reviewed

··· 1 1 # -*- coding: utf-8 -*- 2 2 from __future__ import unicode_literals, print_function, absolute_import 3 3 + 4 4 + import json 5 5 + import random 6 6 + 7 7 + from rest_framework import status 8 8 + from rest_framework.test import APITestCase 9 9 + 10 10 + from journeys import GOING 11 11 + from journeys.tests.factories import CampusFactory, JourneyFactory 12 12 + from journeys.tests.factories import ResidenceFactory 13 13 + from notifications import JOIN 14 14 + from notifications import LEAVE 15 15 + from notifications.tests.factories import NotificationFactory 16 16 + from users.tests.factories import UserFactory 17 17 + 18 18 + 19 19 + class NotificationAPITests(APITestCase): 20 20 + 21 21 + def setUp(self): 22 22 + self.user = UserFactory() 23 23 + 24 24 + @staticmethod 25 25 + def _make_journey(user=None, kind=GOING): 26 26 + user = UserFactory() if user is None else user 27 27 + origin = ResidenceFactory(user=user) 28 28 + destination = CampusFactory() 29 29 + return JourneyFactory(user=user, residence=origin, campus=destination, kind=kind) 30 30 + 31 31 + def test_get_notifications(self): 32 32 + [NotificationFactory( 33 33 + user=self.user, 34 34 + verb=random.choice([JOIN, LEAVE]), 35 35 + actor=UserFactory(), 36 36 + target=self._make_journey(self.user) 37 37 + ) for _ in range(5)] 38 38 + [NotificationFactory( 39 39 + user=UserFactory(), 40 40 + verb=random.choice([JOIN, LEAVE]), 41 41 + actor=UserFactory(), 42 42 + target=self._make_journey(self.user) 43 43 + ) for _ in range(5)] 44 44 + url = "/api/v1/notifications/" 45 45 + self.client.force_authenticate(user=self.user) 46 46 + response = self.client.get(url) 47 47 + self.assertEqual(response.status_code, status.HTTP_200_OK) 48 48 + response_data = json.loads(response.content.decode('utf-8')) 49 49 + self.assertEquals(5, len(response_data['results']))

+36

upvcarshare/notifications/tests/test_models.py

reviewed

··· 1 1 # -*- coding: utf-8 -*- 2 2 from __future__ import unicode_literals, print_function, absolute_import 3 3 + 4 4 + from test_plus import TestCase 5 5 + 6 6 + from journeys import GOING 7 7 + from journeys.tests.factories import CampusFactory 8 8 + from journeys.tests.factories import ResidenceFactory, JourneyFactory 9 9 + from notifications import JOIN, LEAVE 10 10 + from notifications.models import Notification 11 11 + from users.tests.factories import UserFactory 12 12 + 13 13 + 14 14 + class NotificationTests(TestCase): 15 15 + 16 16 + def setUp(self): 17 17 + self.user = UserFactory() 18 18 + 19 19 + @staticmethod 20 20 + def _make_journey(user=None, kind=GOING): 21 21 + user = UserFactory() if user is None else user 22 22 + origin = ResidenceFactory(user=user) 23 23 + destination = CampusFactory() 24 24 + return JourneyFactory(user=user, residence=origin, campus=destination, kind=kind) 25 25 + 26 26 + def test_join_generation(self): 27 27 + user = UserFactory() 28 28 + journey = self._make_journey(self.user) 29 29 + journey.join_passenger(user) 30 30 + self.assertEquals(1, Notification.objects.filter(user=self.user, verb=JOIN).count()) 31 31 + 32 32 + def test_leave_generation(self): 33 33 + user = UserFactory() 34 34 + journey = self._make_journey(self.user) 35 35 + journey.join_passenger(user) 36 36 + journey.leave_passenger(user) 37 37 + self.assertEquals(1, Notification.objects.filter(user=self.user, verb=JOIN).count()) 38 38 + self.assertEquals(1, Notification.objects.filter(user=self.user, verb=LEAVE).count())