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)