def test_no_user(self):
"""
Test: No user in the session.
Expected result: Error is returned.
"""
request = self.factory.get('/doesnt/matter')
request.session = {}
self.assertEqual(self.user.email, ORIGINAL_EMAIL, "The starting email is not as expected.")
json_ret = json.loads(ajax.add_email(request, NEW_EMAIL))
self.user = models.FeedUser.objects.get(id = self.user.id)
self.assertEqual(json_ret["error"], "You need to be authorized to call this.", "An error wasn't returend")
self.assertEqual(self.user.email, ORIGINAL_EMAIL, "Email was changed.")
def test_add_email(self):
"""
Test: User exists.
Expected result: No exceptions are raised
"""
request = self.factory.get('/doesnt/matter')
request.session = {"cur_user" : self.user.id}
self.assertEqual(self.user.email, ORIGINAL_EMAIL, "The starting email is not as expected.")
json_ret = json.loads(ajax.add_email(request, NEW_EMAIL))
self.user = models.FeedUser.objects.get(id = self.user.id)
self.assertEqual(json_ret["error"], "", "An error was returend")
self.assertEqual(self.user.email, NEW_EMAIL, "Email was not changed.")
def test_exception(self):
"""
Test: Exception is raied in the main try of the function.
Expected result: Error is returned.
"""
request = self.factory.get('/doesnt/matter')
request.session = {"cur_user" : self.user.id}
original_get_user_from_user_id = ajax._get_user_from_user_id
ajax._get_user_from_user_id = raise_exception
self.assertEqual(self.user.email, ORIGINAL_EMAIL, "The starting email is not as expected.")
json_ret = json.loads(ajax.add_email(request, NEW_EMAIL))
self.user = models.FeedUser.objects.get(id = self.user.id)
ajax._get_user_from_user_id = original_get_user_from_user_id
self.assertTrue(EXCEPTION_STRING in json_ret["error"], "An error wasn't returend")
self.assertEqual(self.user.email, ORIGINAL_EMAIL, "Email was changed.")