def setUp(self):
self.user = UserFactory(username="******")
self.today = date.today()
self.workitem1 = WorkItemFactory.create(user=self.user, date=self.today, hours=7.75)
# create 2 new repos
self.repos = RepoFactory.create_batch(2)
# create 2 new issues assigned to the user for each repo
self.issues_for_repo_1 = IssueFactory.create_batch(2, assignee=self.user, open=True, repo=self.repos[0])
self.issues_for_repo_2 = IssueFactory.create_batch(2, assignee=self.user, open=True, repo=self.repos[1])
def setUp(self):
self.user = UserFactory(username="******")
self.user2 = UserFactory(username="******")
self.job = JobFactory(name="test1", available_all_users=True)
self.workitemJ = WorkItemFactory.create(user=self.user, job=self.job, date=date(2015,01,13), hours=8.0)
self.workitemJ2 = WorkItemFactory.create(user=self.user2, job=self.job, date=date(2015, 01, 13), hours=1.0)
self.workitemD = WorkItemFactory.create(user=self.user, job=self.job, date=date(2014,12,30), hours=7.0)
self.workitemD2 = WorkItemFactory.create(user=self.user2, job=self.job, date=date(2014, 12, 30), hours=2.0)
self.workitemT = WorkItemFactory.create(user=self.user, job=self.job, date=date.today(), hours=3.0)
self.workitemT2 = WorkItemFactory.create(user=self.user2, job=self.job, date=date.today(), hours=3.5)
def test_get_total_hours_from_workitems(self):
user = UserFactory(username="******")
WorkItemFactory.create_batch(4, user=user, hours=2)
items = WorkItem.objects.filter(user=user)
self.assertEqual(get_total_hours_from_workitems(items), 8.0)
