def test_milestone_date_invalid_metric(self):
"""
If an invalid metric name is given, raise an AttributeError.
"""
user = UserFactory.create()
display = MilestoneDisplay(user)
DataPointFactory.create(link_clicks=4, date=aware_date(2014, 1, 1), link__user=user)
with self.assertRaises(AttributeError):
display.milestone_date('invalid', 2, 1)
def test_milestone_date_invalid_metric(self):
"""
If an invalid metric name is given, raise an AttributeError.
"""
user = UserFactory.create()
display = MilestoneDisplay(user)
DataPointFactory.create(link_clicks=4,
date=aware_date(2014, 1, 1),
link__user=user)
with self.assertRaises(AttributeError):
display.milestone_date('invalid', 2, 1)
def test_milestone_date(self):
user = UserFactory.create()
display = MilestoneDisplay(user)
DataPointFactory.create(link_clicks=4, date=aware_date(2014, 1, 1), link__user=user)
DataPointFactory.create(link_clicks=3, date=aware_date(2014, 1, 2), link__user=user)
DataPointFactory.create(link_clicks=2, date=aware_date(2014, 1, 3), link__user=user)
eq_(display.milestone_date('link_clicks', 10, 4), aware_date(2014, 1, 2))
def test_milestone_date_not_reached(self):
"""
If the milestone hasn't been reached by the user, return None.
"""
user = UserFactory.create()
display = MilestoneDisplay(user)
DataPointFactory.create(link_clicks=4, date=aware_date(2014, 1, 1), link__user=user)
eq_(display.milestone_date('link_clicks', 8, 2), None)
def test_milestone_date_not_reached(self):
"""
If the milestone hasn't been reached by the user, return None.
"""
user = UserFactory.create()
display = MilestoneDisplay(user)
DataPointFactory.create(link_clicks=4,
date=aware_date(2014, 1, 1),
link__user=user)
eq_(display.milestone_date('link_clicks', 8, 2), None)
def test_milestone_date(self):
user = UserFactory.create()
display = MilestoneDisplay(user)
DataPointFactory.create(link_clicks=4,
date=aware_date(2014, 1, 1),
link__user=user)
DataPointFactory.create(link_clicks=3,
date=aware_date(2014, 1, 2),
link__user=user)
DataPointFactory.create(link_clicks=2,
date=aware_date(2014, 1, 3),
link__user=user)
eq_(display.milestone_date('link_clicks', 10, 4),
aware_date(2014, 1, 2))
