def test_done_in_range_when_exculing_classes(self):
service = TeamStats('Team 1', ['Urgent', 'Intangible'])
start_date = datetime(2012, 1, 1)
end_date = datetime(2012, 12, 31)
with mock.patch('kardboard.services.teams.Kard') as mock_Kard:
service.done_in_range(start_date, end_date)
start_expected = datetime(2012, 1, 1, 0, 0, 0)
end_expected = datetime(2012, 12, 31, 23, 59, 59)
mock_Kard.objects.filter.assert_called_with(
team='Team 1',
done_date__gte=start_expected,
done_date__lte=end_expected,
_service_class__nin=['Urgent', 'Intangible'],
)
def setUp(self):
super(TeamStatsTest, self).setUp()
self.service = TeamStats('Team Foo')
self.team = 'Team Foo'
