def setUp(self):
self.middleware = StatsSinceLastVisitMiddleware()
self.factory = RequestFactory()
self.request = self.factory.get('/')
messages_patch = patch('affiliates.links.middleware.messages')
self.addCleanup(messages_patch.stop)
self.messages = messages_patch.start()
def setUp(self):
self.middleware = StatsSinceLastVisitMiddleware()
self.factory = RequestFactory()
self.request = self.factory.get("/")
messages_patch = patch("affiliates.links.middleware.messages")
self.addCleanup(messages_patch.stop)
self.messages = messages_patch.start()
class StatsSinceLastVisitMiddlewareTests(TestCase):
def setUp(self):
self.middleware = StatsSinceLastVisitMiddleware()
self.factory = RequestFactory()
self.request = self.factory.get("/")
messages_patch = patch("affiliates.links.middleware.messages")
self.addCleanup(messages_patch.stop)
self.messages = messages_patch.start()
def test_process_request_unauthenticated(self):
"""
If the current user isn't authed, return None and do not log a
message.
"""
self.request.user = Mock()
self.request.user.is_authenticated.return_value = False
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
def test_process_request_facebook_user(self):
"""
If the current user doesn't have a profile (either as a
FacebookUser or no profile in the database) return None and do
not log a message.
"""
self.request.user = FacebookUserFactory.create()
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
self.request.user = UserFactory.create(userprofile=None)
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
def test_process_request_no_last_visit(self):
"""
If we haven't logged a last visit for the user, set their
last_visit and return None and do not log a message.
"""
self.request.user = UserFactory.create(userprofile__last_visit=None)
with patch("affiliates.links.middleware.timezone.now") as mock_now:
mock_now.return_value = aware_datetime(2014, 1, 1)
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
profile = UserProfile.objects.get(user=self.request.user)
eq_(profile.last_visit, aware_date(2014, 1, 1))
def test_process_request_less_than_one_day(self):
"""
If it has been less than one day since the user's last visit,
return None and do not log a message.
"""
self.request.user = UserFactory.create(userprofile__last_visit=aware_date(2014, 1, 1))
with patch("affiliates.links.middleware.timezone.now") as mock_now:
mock_now.return_value = aware_datetime(2014, 1, 1)
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
def test_process_request_no_clicks_or_downloads(self):
"""
If there were no clicks or downloads since the user's last
visit, update their last_visit date and do not log a message.
"""
self.request.user = UserFactory.create(userprofile__last_visit=aware_date(2014, 1, 1))
with patch("affiliates.links.middleware.timezone.now") as mock_now:
mock_now.return_value = aware_datetime(2014, 1, 3)
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
profile = UserProfile.objects.get(user=self.request.user)
eq_(profile.last_visit, aware_date(2014, 1, 3))
def test_process_request_clicks_and_downloads(self):
"""
If there were any clicks or downloads since the user's last
visit, update their last_visit date and log a message.
"""
self.request.user = UserFactory.create(userprofile__last_visit=aware_date(2014, 1, 1))
# Date of last visit not included.
DataPointFactory.create(
link__user=self.request.user, date=aware_date(2014, 1, 1), link_clicks=3, firefox_downloads=9
)
# Current date and dates between are included.
DataPointFactory.create(
link__user=self.request.user, date=aware_date(2014, 1, 2), link_clicks=4, firefox_downloads=7
)
DataPointFactory.create(
link__user=self.request.user, date=aware_date(2014, 1, 3), link_clicks=1, firefox_downloads=2
)
with patch("affiliates.links.middleware.timezone.now") as mock_now:
mock_now.return_value = aware_datetime(2014, 1, 3)
eq_(self.middleware.process_request(self.request), None)
ok_(self.messages.info.called)
message = self.messages.info.call_args[0][1]
ok_("5" in message and "9" in message)
profile = UserProfile.objects.get(user=self.request.user)
eq_(profile.last_visit, aware_date(2014, 1, 3))
class StatsSinceLastVisitMiddlewareTests(TestCase):
def setUp(self):
self.middleware = StatsSinceLastVisitMiddleware()
self.factory = RequestFactory()
self.request = self.factory.get('/')
messages_patch = patch('affiliates.links.middleware.messages')
self.addCleanup(messages_patch.stop)
self.messages = messages_patch.start()
def test_process_request_unauthenticated(self):
"""
If the current user isn't authed, return None and do not log a
message.
"""
self.request.user = Mock()
self.request.user.is_authenticated.return_value = False
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
def test_process_request_facebook_user(self):
"""
If the current user doesn't have a profile (either as a
FacebookUser or no profile in the database) return None and do
not log a message.
"""
self.request.user = FacebookUserFactory.create()
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
self.request.user = UserFactory.create(userprofile=None)
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
def test_process_request_no_last_visit(self):
"""
If we haven't logged a last visit for the user, set their
last_visit and return None and do not log a message.
"""
self.request.user = UserFactory.create(userprofile__last_visit=None)
with patch('affiliates.links.middleware.timezone.now') as mock_now:
mock_now.return_value = aware_datetime(2014, 1, 1)
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
profile = UserProfile.objects.get(user=self.request.user)
eq_(profile.last_visit, aware_date(2014, 1, 1))
def test_process_request_less_than_one_day(self):
"""
If it has been less than one day since the user's last visit,
return None and do not log a message.
"""
self.request.user = UserFactory.create(
userprofile__last_visit=aware_date(2014, 1, 1))
with patch('affiliates.links.middleware.timezone.now') as mock_now:
mock_now.return_value = aware_datetime(2014, 1, 1)
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
def test_process_request_no_clicks_or_downloads(self):
"""
If there were no clicks or downloads since the user's last
visit, update their last_visit date and do not log a message.
"""
self.request.user = UserFactory.create(
userprofile__last_visit=aware_date(2014, 1, 1))
with patch('affiliates.links.middleware.timezone.now') as mock_now:
mock_now.return_value = aware_datetime(2014, 1, 3)
eq_(self.middleware.process_request(self.request), None)
ok_(not self.messages.info.called)
profile = UserProfile.objects.get(user=self.request.user)
eq_(profile.last_visit, aware_date(2014, 1, 3))
def test_process_request_clicks_and_downloads(self):
"""
If there were any clicks or downloads since the user's last
visit, update their last_visit date and log a message.
"""
self.request.user = UserFactory.create(
userprofile__last_visit=aware_date(2014, 1, 1))
# Date of last visit not included.
DataPointFactory.create(link__user=self.request.user,
date=aware_date(2014, 1, 1),
link_clicks=3,
firefox_downloads=9)
# Current date and dates between are included.
DataPointFactory.create(link__user=self.request.user,
date=aware_date(2014, 1, 2),
link_clicks=4,
firefox_downloads=7)
DataPointFactory.create(link__user=self.request.user,
date=aware_date(2014, 1, 3),
link_clicks=1,
firefox_downloads=2)
with patch('affiliates.links.middleware.timezone.now') as mock_now:
mock_now.return_value = aware_datetime(2014, 1, 3)
eq_(self.middleware.process_request(self.request), None)
ok_(self.messages.info.called)
message = self.messages.info.call_args[0][1]
ok_('5' in message and '9' in message)
profile = UserProfile.objects.get(user=self.request.user)
eq_(profile.last_visit, aware_date(2014, 1, 3))