class MiddlewareTests(TransactionTestCase):
"""Test suite for middleware."""
def setUp(self):
self.factory = RequestFactory()
self.user = get_user_model().objects.create_user('zoidberg')
self.score = UserScore(user=self.user, score=10).save()
self.middleware = NPSMiddleware()
def test_process_request(self):
request = self.factory.get('/')
request.user = self.user
with mock.patch('net_promoter_score.settings.NPS_DISPLAY_FUNCTION', lambda r: True):
resp = self.middleware.process_request(request)
self.assertIsNone(resp)
self.assertTrue(request.show_nps)
# verify that unauthenticated users always return False
request.user = AnonymousUser()
resp = self.middleware.process_request(request)
self.assertIsNone(resp)
self.assertFalse(request.show_nps)
def setUp(self):
self.factory = RequestFactory()
self.user = get_user_model().objects.create_user('zoidberg')
self.score = UserScore(user=self.user, score=10).save()
self.middleware = NPSMiddleware()