def test_basic(self): # create some work items items = [ # user, date, hours, text, job (self.user, self.today, 1, "item1", Job.objects.filter(name="Job_Today")[0]), (self.user2, self.today, 2, "item2", Job.objects.filter(name="Job_Today")[0]), (self.user2, self.yesterday, 3, "item3", Job.objects.filter(name="Job_Today")[0]), (self.user3, self.yesterday, 4, "item4", Job.objects.filter(name="Job_Today")[0]), (self.user4, self.last_week, 5, "item5", Job.objects.filter(name="Job_Today")[0]), (self.user5, self.tomorrow, 6, "item6", Job.objects.filter(name="Job_Today")[0]), ] for item in items: wi = WorkItem.objects.create(user=item[0], date=item[1], hours=item[2], text=item[3], job=item[4]) wi.save() # try to send emails tasks.send_reminder_emails() self.assertEquals(len(mail.outbox), 3) # user3, user4, user5 all_recipients = list(m.to[0] for m in mail.outbox) self.assertEquals(len(all_recipients), 3) self.assertTrue("*****@*****.**" in all_recipients) self.assertTrue("*****@*****.**" in all_recipients) self.assertTrue("*****@*****.**" in all_recipients) qs = WorkLogReminder.objects.all() self.assertEquals(qs.count(), 3)
def test_empty(self): # create some work items items = [ # user, date, hours, text, job (self.user, self.today, 1, "item1", Job.objects.filter(name="Job_Today")[0]), (self.user2, self.today, 2, "item2", Job.objects.filter(name="Job_Today")[0]), (self.user3, self.today, 3, "item3", Job.objects.filter(name="Job_Today")[0]), (self.user4, self.today, 4, "item4", Job.objects.filter(name="Job_Today")[0]), (self.user5, self.today, 5, "item5", Job.objects.filter(name="Job_Today")[0]), ] for item in items: wi = WorkItem.objects.create(user=item[0], date=item[1], hours=item[2], text=item[3], job=item[4]) wi.save() # try to send emails tasks.send_reminder_emails() self.assertEquals(len(mail.outbox), 0) all_recipients = list(m.to[0] for m in mail.outbox) self.assertEquals(len(all_recipients), 0) qs = WorkLogReminder.objects.all() self.assertEquals(qs.count(), 0)