<NME> test_repair.py
<BEF> """Repair command tests"""

from django.core import management

from modoboa.lib.permissions import ObjectAccess, get_object_owner
from modoboa.lib.tests import ModoTestCase
from .. import factories, models


class RepairTestCase(ModoTestCase):
    """TestCase for repair command."""

    @classmethod
    def setUpTestData(cls):  # NOQA:N802
        """Create some data."""
        super(RepairTestCase, cls).setUpTestData()
        factories.populate_database()

    def test_management_command(self):
        """Check that command works fine."""
    def test_management_command(self):
        """Check that command works fine."""
        ObjectAccess.objects.all().delete()
        mbox = models.Mailbox.objects.all()[0]
        # 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)

    def test_management_command_with_dry_run(self):
        """Check that command works fine."""
        ObjectAccess.objects.all().delete()
        mbox = models.Mailbox.objects.all()[0]
        # 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
        assert ret is None
        # assert its not fixed
        self.assertIs(get_object_owner(mbox), None)

    def test_management_command_with_nul_domain(self):
        """Just assume nothing raise when an alias has no domain."""
        models.Alias.objects.create(address="@modoboa.xxx")
        ret = management.call_command("modo", "repair", "--quiet")
        assert ret is None

    def test_management_command_with_no_alias(self):
        """Check that problem is fixed."""
        count, detail = models.Alias.objects.filter(
            address="user@test.com", internal=True).delete()
        self.assertEqual(count, 3)
        ret = management.call_command("modo", "repair", "--quiet")
        assert ret is None
        self.assertTrue(
            models.Alias.objects.filter(
                address="user@test.com", internal=True).exists())

<MSG> assume that aliases ar fixed

<DFF> @@ -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

assume that aliases ar fixed