class TestHelperFunction(TestCase): def setUp(self): self.user = factory.create_user('*****@*****.**', '*****@*****.**', 'john123', 'John', 'Doe') self.location = factory.create_location('Thailand', 'Nakhonsawan') self.category = factory.create_category('Travel', 'travel') self.blog = factory.create_blog('Travel in Nakhonsawan', self.user, self.category, self.location) self.user_with_notification = factory.create_user('*****@*****.**', '*****@*****.**', 'adam123', 'Adam', 'Johnson') self.notification = Notification(subject=self.user_with_notification, action=1, blog=self.blog) self.notification.save() def tearDown(self): rm_user(self.user.id) rm_user(self.user_with_notification.id) self.notification.delete() def test_get_notifications(self): self.assertEqual(1, len(get_notifications(self.user))) self.assertEqual(0, len(get_notifications(self.user_with_notification)))
class TestNotification(TestCase): def setUp(self): self.user = factory.create_user('*****@*****.**', '*****@*****.**', 'john123', 'John', 'Doe') self.location = factory.create_location('Thailand', 'Nakhonsawan') self.category = factory.create_category('Travel', 'travel') self.blog = factory.create_blog('Travel in Nakhonsawan', self.user, self.category, self.location) self.love_user = factory.create_user('*****@*****.**', '*****@*****.**', 'adam123', 'Adam', 'Johnson') self.love_notification = Notification(subject=self.love_user, action=1, blog=self.blog) self.love_notification.save() self.download_notification = Notification(subject=self.user, action=2, blog=self.blog) self.download_notification.save() def tearDown(self): self.location.delete() self.category.delete() self.blog.delete() self.love_notification.delete() self.download_notification.delete() # UserInvitation.objects.get(email=self.love_user.email).delete() UserProfile.objects.get(user=self.love_user).delete() self.love_user.delete() # UserInvitation.objects.get(email=self.user.email).delete() UserProfile.objects.get(user=self.user).delete() self.user.delete() def test_unicode(self): self.assertEqual('Adam Johnson loved Travel in Nakhonsawan', self.love_notification.__unicode__()) self.assertEqual('John Doe downloaded Travel in Nakhonsawan', self.download_notification.__unicode__()) def test_get_action_text(self): self.assertEqual('loved', self.love_notification.get_action_text()) self.assertEqual('downloaded', self.download_notification.get_action_text())