class DataImporterTaskTest(TestCase): def setUp(self): owner, _ = User.objects.get_or_create(username='******', email='*****@*****.**') source = os.path.join(LOCAL_DIR, 'data/person_test.csv') self.importer = TestMeta self.task = DataImpoterTask() self.task.run(importer=self.importer, source=source, owner=owner) def test_task_run(self): created_person = Person.objects.filter(first_name='Eldo', last_name='Rock', age='28').exists() self.assertTrue(created_person) def test_task_create_all(self): self.assertEqual(Person.objects.all().count(), 3) def test_send_email(self): outbox = mail.outbox[0] self.assertEqual(outbox.body, 'Your file was imported with sucess') self.assertEqual(outbox.to, ['*****@*****.**'])