diff --git a/modoboa/admin/tests/test_repair.py b/modoboa/admin/tests/test_repair.py index e93efa932..c50128556 100644 --- a/modoboa/admin/tests/test_repair.py +++ b/modoboa/admin/tests/test_repair.py @@ -21,18 +21,20 @@ class RepairTestCase(ModoTestCase): def test_management_command(self): """Check that command works fine.""" ObjectAccess.objects.all().delete() - mbox = models.Mailbox.objects.all()[0] + mbox = models.Mailbox.objects.first() + alias = models.Alias.objects.first() # assert mbox has no owner self.assertIs(get_object_owner(mbox), None) # fix it. run in quiet mode because we dont want output in tests management.call_command("modo", "repair", "--quiet") # assert its fixed self.assertIsNot(get_object_owner(mbox), None) + self.assertIsNot(get_object_owner(alias), None) def test_management_command_with_dry_run(self): """Check that command works fine.""" ObjectAccess.objects.all().delete() - mbox = models.Mailbox.objects.all()[0] + mbox = models.Mailbox.objects.first() # assert mbox has no owner self.assertIs(get_object_owner(mbox), None) # show problems. run in quiet mode because we dont want output in tests