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)
