def test_creation_milestone_close_to_next_milestone(self):
"""
If the user is within 1 link of the next milestone, return
yesterday's date with the close_to_milestone message.
"""
LinkFactory.create_batch(2, banner_type='test', user=self.user)
self.display.surrounding_milestones.return_value = (None, 3)
with patch('affiliates.base.milestones.date_yesterday') as date_yesterday:
milestone = self.display.creation_milestone('test', self.messages)
eq_(milestone, (date_yesterday.return_value, 'close_to_milestone_3'))
self.display.surrounding_milestones.assert_called_with(
2, self.display.creation_milestones)
def test_creation_milestone_close_to_next_milestone(self):
"""
If the user is within 1 link of the next milestone, return
yesterday's date with the close_to_milestone message.
"""
variation = TextBannerVariationFactory.create()
LinkFactory.create_batch(2, banner_variation=variation, user=self.user)
self.display.surrounding_milestones.return_value = (None, 3)
with patch(
'affiliates.base.milestones.date_yesterday') as date_yesterday:
milestone = self.display.creation_milestone(
TextBanner, self.messages)
eq_(milestone,
(date_yesterday.return_value, 'close_to_milestone_3'))
self.display.surrounding_milestones.assert_called_with(
2, self.display.creation_milestones)
def test_basic(self):
category = CategoryFactory.create()
banner1, banner2 = TextBannerFactory.create_batch(2, category=category)
link1, link2 = LinkFactory.create_batch(2, banner_variation__banner=banner1)
link3, link4 = LinkFactory.create_batch(2, banner_variation__banner=banner2)
# Set datapoint totals.
self._datapoint(link1, 4)
self._datapoint(link1, 4)
self._datapoint(link2, 3)
self._datapoint(link2, 3)
self._datapoint(link3, 2)
self._datapoint(link3, 2)
self._datapoint(link4, 1)
self._datapoint(link4, 1)
# Assert that data hasn't been denormalized yet.
eq_(link1.link_clicks, 0)
eq_(link2.link_clicks, 0)
eq_(link3.link_clicks, 0)
eq_(link4.link_clicks, 0)
eq_(banner1.link_clicks, 0)
eq_(banner2.link_clicks, 0)
eq_(category.link_clicks, 0)
self.command.handle()
# Re-fetch data now that is has updated
category = Category.objects.get(pk=category.pk)
banner1, banner2 = TextBanner.objects.order_by('id')
link1, link2, link3, link4 = Link.objects.order_by('id')
# Assert that counts are now denormalized.
eq_(link1.link_clicks, 8)
eq_(link2.link_clicks, 6)
eq_(link3.link_clicks, 4)
eq_(link4.link_clicks, 2)
eq_(banner1.link_clicks, 14)
eq_(banner2.link_clicks, 6)
eq_(category.link_clicks, 20)
def test_basic(self):
category = CategoryFactory.create()
banner1, banner2 = TextBannerFactory.create_batch(2, category=category)
link1, link2 = LinkFactory.create_batch(2, banner_variation__banner=banner1)
link3, link4 = LinkFactory.create_batch(2, banner_variation__banner=banner2)
# Set datapoint totals.
self._datapoint(link1, 4)
self._datapoint(link1, 4)
self._datapoint(link2, 3)
self._datapoint(link2, 3)
self._datapoint(link3, 2)
self._datapoint(link3, 2)
self._datapoint(link4, 1)
self._datapoint(link4, 1)
# Assert that data hasn't been denormalized yet.
eq_(link1.link_clicks, 0)
eq_(link2.link_clicks, 0)
eq_(link3.link_clicks, 0)
eq_(link4.link_clicks, 0)
eq_(banner1.link_clicks, 0)
eq_(banner2.link_clicks, 0)
eq_(category.link_clicks, 0)
self.command.handle()
# Re-fetch data now that is has updated
category = Category.objects.get(pk=category.pk)
banner1, banner2 = TextBanner.objects.order_by('id')
link1, link2, link3, link4 = Link.objects.order_by('id')
# Assert that counts are now denormalized.
eq_(link1.link_clicks, 8)
eq_(link2.link_clicks, 6)
eq_(link3.link_clicks, 4)
eq_(link4.link_clicks, 2)
eq_(banner1.link_clicks, 14)
eq_(banner2.link_clicks, 6)
eq_(category.link_clicks, 20)
def test_not_authenticated(self):
"""
If the current user isn't authenticated, render the home
page with the correct stats.
"""
request = self.factory.get('/')
request.user = Mock()
request.user.is_authenticated.return_value = False
# User count = 27 + 3 autogenerated via related factory
# Link count = 3 autogenerated via related factory
UserFactory.create_batch(27)
LinkFactory.create_batch(3)
with patch('affiliates.base.views.render') as render:
with patch.object(Link.objects, 'total_link_clicks', return_value=64):
eq_(views.home(request), render.return_value)
render.assert_called_with(request, 'base/home.html', {
'affiliate_count': 30,
'link_count': 3,
'click_count': 64
})
def test_not_authenticated(self):
"""
If the current user isn't authenticated, render the home
page with the correct stats.
"""
request = self.factory.get('/')
request.user = Mock()
request.user.is_authenticated.return_value = False
# User count = 27 + 3 autogenerated via related factory
# Link count = 3 autogenerated via related factory
UserFactory.create_batch(27)
LinkFactory.create_batch(3)
with patch('affiliates.base.views.render') as render:
with patch.object(Link.objects, 'total_link_clicks', return_value=64):
eq_(views.home(request), render.return_value)
render.assert_called_with(request, 'base/home.html', {
'affiliate_count': 30,
'link_count': 3,
'click_count': 64
})
def test_success(self):
link1, link2 = LinkFactory.create_batch(2)
self.service.get_clicks_for_date.return_value = {
unicode(link1.pk): '4',
unicode(link2.pk): '7'
}
yesterday = aware_datetime(2014, 1, 1).date()
with patch.object(collect_ga_data, 'date_yesterday') as date_yesterday:
date_yesterday.return_value = yesterday
self.command.execute()
self.service.get_clicks_for_date.assert_called_with(yesterday)
ok_(link1.datapoint_set.filter(date=yesterday, link_clicks=4).exists())
ok_(link2.datapoint_set.filter(date=yesterday, link_clicks=7).exists())
def test_integrity_error_bulk_create(self):
"""
If an IntegrityError is raised during bulk_create, raise a
CommandError.
"""
link1, link2 = LinkFactory.create_batch(2)
self.service.get_clicks_for_date.return_value = {
unicode(link1.pk): '4',
unicode(link2.pk): '7'
}
with patch.object(collect_ga_data, 'DataPoint') as MockDataPoint:
MockDataPoint.objects.bulk_create.side_effect = IntegrityError
with self.assertRaises(CommandError):
self.command.execute()
def test_default_yesterday(self):
"""When no date is given, fetch data for the two days ago."""
link1, link2 = LinkFactory.create_batch(2)
self.service.get_clicks_for_date.return_value = {
unicode(link1.pk): '4',
unicode(link2.pk): '7'
}
two_days_ago = aware_date(2014, 1, 1)
with patch.object(collect_ga_data, 'timezone') as mock_timezone:
mock_timezone.now.return_value = aware_date(2014, 1, 3)
self.command.execute()
self.service.get_clicks_for_date.assert_called_with(two_days_ago)
eq_(link1.datapoint_set.get(date=two_days_ago).link_clicks, 4)
eq_(link2.datapoint_set.get(date=two_days_ago).link_clicks, 7)
def test_default_yesterday(self):
"""When no date is given, fetch data for the two days ago."""
link1, link2 = LinkFactory.create_batch(2)
self.service.get_clicks_for_date.return_value = {
unicode(link1.pk): '4',
unicode(link2.pk): '7'
}
two_days_ago = aware_date(2014, 1, 1)
with patch.object(collect_ga_data, 'timezone') as mock_timezone:
mock_timezone.now.return_value = aware_date(2014, 1, 3)
self.command.execute()
self.service.get_clicks_for_date.assert_called_with(two_days_ago)
eq_(link1.datapoint_set.get(date=two_days_ago).link_clicks, 4)
eq_(link2.datapoint_set.get(date=two_days_ago).link_clicks, 7)
def test_creation_milestone_no_previous_links_created(self):
"""
If there is no previous milestone, but the user has created at
least one link (which is normally impossible, as the default
milestones start at 1), show when their last link was created.
"""
links = LinkFactory.create_batch(2, banner_type='test', user=self.user)
links[0].created = aware_datetime(2014, 1, 1, 5)
links[1].created = aware_datetime(2014, 1, 2, 8)
for link in links:
link.save()
self.display.surrounding_milestones.return_value = (None, 5)
milestone = self.display.creation_milestone('test', self.messages)
eq_(milestone, (aware_date(2014, 1, 2), 'link_created'))
def test_creation_milestone_no_next_milestone(self):
"""
If there is no next milestone, show the date of their last
milestone.
"""
links = LinkFactory.create_batch(4, banner_type='test', user=self.user)
links[0].created = aware_datetime(2014, 1, 1, 5)
links[1].created = aware_datetime(2014, 1, 1, 8)
links[2].created = aware_datetime(2014, 1, 2, 5) # Winner!
links[3].created = aware_datetime(2014, 1, 3, 5)
for link in links:
link.save()
self.display.surrounding_milestones.return_value = (3, None)
milestone = self.display.creation_milestone('test', self.messages)
eq_(milestone, (aware_date(2014, 1, 2), 'achieved_milestone_3'))
def test_get_queryset(self):
"""
Available links should be limited to those owned by the current
user.
"""
request = self.factory.get('/')
request.user = UserFactory.create()
link1, link2 = LinkFactory.create_batch(2, user=request.user)
unowned_link = LinkFactory.create()
view = views.LinkDetailView()
view.request = request
qs = view.get_queryset()
ok_(link1 in qs)
ok_(link2 in qs)
ok_(unowned_link not in qs)
def test_get_queryset(self):
"""
Available links should be limited to those owned by the current
user.
"""
request = self.factory.get('/')
request.user = UserFactory.create()
link1, link2 = LinkFactory.create_batch(2, user=request.user)
unowned_link = LinkFactory.create()
view = views.LinkDetailView()
view.request = request
qs = view.get_queryset()
ok_(link1 in qs)
ok_(link2 in qs)
ok_(unowned_link not in qs)
def test_creation_milestone_far_from_next_milestone(self):
"""
If the user isn't close to the next milestone, show the date of
their last milestone.
"""
variation = TextBannerVariationFactory.create()
links = LinkFactory.create_batch(4, banner_variation=variation, user=self.user)
links[0].created = aware_datetime(2014, 1, 1, 5)
links[1].created = aware_datetime(2014, 1, 1, 8)
links[2].created = aware_datetime(2014, 1, 2, 5) # Winner!
links[3].created = aware_datetime(2014, 1, 3, 5)
for link in links:
link.save()
self.display.surrounding_milestones.return_value = (3, 10)
milestone = self.display.creation_milestone(TextBanner, self.messages)
eq_(milestone, (aware_date(2014, 1, 2), 'achieved_milestone_3'))
def test_date_argument_today(self):
"""
If the date argument is the word 'today', set the query date to
the current date.
"""
link1, link2 = LinkFactory.create_batch(2)
self.service.get_clicks_for_date.return_value = {
unicode(link1.pk): '4',
unicode(link2.pk): '7'
}
today = aware_datetime(2014, 1, 2).date()
with patch.object(collect_ga_data, 'timezone') as mock_timezone:
mock_timezone.now.return_value = aware_datetime(2014, 1, 2)
self.command.execute('today')
self.service.get_clicks_for_date.assert_called_with(today)
eq_(link1.datapoint_set.get(date=today).link_clicks, 4)
eq_(link2.datapoint_set.get(date=today).link_clicks, 7)
def test_creation_milestone_no_previous_links_created(self):
"""
If there is no previous milestone, but the user has created at
least one link (which is normally impossible, as the default
milestones start at 1), show when their last link was created.
"""
variation = TextBannerVariationFactory.create()
links = LinkFactory.create_batch(2,
banner_variation=variation,
user=self.user)
links[0].created = aware_datetime(2014, 1, 1, 5)
links[1].created = aware_datetime(2014, 1, 2, 8)
for link in links:
link.save()
self.display.surrounding_milestones.return_value = (None, 5)
milestone = self.display.creation_milestone(TextBanner, self.messages)
eq_(milestone, (aware_date(2014, 1, 2), 'link_created'))
def test_date_argument_today(self):
"""
If the date argument is the word 'today', set the query date to
the current date.
"""
link1, link2 = LinkFactory.create_batch(2)
self.service.get_clicks_for_date.return_value = {
unicode(link1.pk): '4',
unicode(link2.pk): '7'
}
today = aware_datetime(2014, 1, 2).date()
with patch.object(collect_ga_data, 'timezone') as mock_timezone:
mock_timezone.now.return_value = aware_datetime(2014, 1, 2)
self.command.execute('today')
self.service.get_clicks_for_date.assert_called_with(today)
eq_(link1.datapoint_set.get(date=today).link_clicks, 4)
eq_(link2.datapoint_set.get(date=today).link_clicks, 7)
def test_creation_milestone_no_next_milestone(self):
"""
If there is no next milestone, show the date of their last
milestone.
"""
variation = TextBannerVariationFactory.create()
links = LinkFactory.create_batch(4,
banner_variation=variation,
user=self.user)
links[0].created = aware_datetime(2014, 1, 1, 5)
links[1].created = aware_datetime(2014, 1, 1, 8)
links[2].created = aware_datetime(2014, 1, 2, 5) # Winner!
links[3].created = aware_datetime(2014, 1, 3, 5)
for link in links:
link.save()
self.display.surrounding_milestones.return_value = (3, None)
milestone = self.display.creation_milestone(TextBanner, self.messages)
eq_(milestone, (aware_date(2014, 1, 2), 'achieved_milestone_3'))
def test_date_argument(self):
"""
If a date argument is given, parse it as DD-MM-YYYY and use it
as the query date.
"""
link1, link2 = LinkFactory.create_batch(2)
self.service.get_clicks_for_date.return_value = {
unicode(link1.pk): '4',
unicode(link2.pk): '7'
}
query_date = aware_datetime(2014, 1, 1).date()
# Create pre-existing data to check that it is replaced.
DataPointFactory.create(link=link1, date=query_date, link_clicks=18)
DataPointFactory.create(link=link2, date=query_date, link_clicks=14)
self.command.execute('01-01-2014')
# There must only be one datapoint for the query date, and the
# link_clicks must match the new data.
self.service.get_clicks_for_date.assert_called_with(query_date)
eq_(link1.datapoint_set.get(date=query_date).link_clicks, 4)
eq_(link2.datapoint_set.get(date=query_date).link_clicks, 7)
def test_date_argument(self):
"""
If a date argument is given, parse it as DD-MM-YYYY and use it
as the query date.
"""
link1, link2 = LinkFactory.create_batch(2)
self.service.get_clicks_for_date.return_value = {
unicode(link1.pk): '4',
unicode(link2.pk): '7'
}
query_date = aware_datetime(2014, 1, 1).date()
# Create pre-existing data to check that it is replaced.
DataPointFactory.create(link=link1, date=query_date, link_clicks=18)
DataPointFactory.create(link=link2, date=query_date, link_clicks=14)
self.command.execute('01-01-2014')
# There must only be one datapoint for the query date, and the
# link_clicks must match the new data.
self.service.get_clicks_for_date.assert_called_with(query_date)
eq_(link1.datapoint_set.get(date=query_date).link_clicks, 4)
eq_(link2.datapoint_set.get(date=query_date).link_clicks, 7)
def test_links(self):
banner = TextBannerFactory.create()
link1, link2 = LinkFactory.create_batch(2, banner_variation__banner=banner)
eq_(set(banner.links), set([link1, link2]))
def test_links(self):
banner = TextBannerFactory.create()
link1, link2 = LinkFactory.create_batch(2, banner_variation__banner=banner)
eq_(set(banner.links), set([link1, link2]))
