<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."""
        ObjectAccess.objects.all().delete()
        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
        ret = management.call_command("modo", "repair", "--quiet")
        # fix it. run in quiet mode because we dont want output in tests
        ret = management.call_command("modo", "repair", "--quiet")
        assert ret is None
        # assert its fixed
        self.assertIsNot(get_object_owner(mbox), None)
        self.assertIsNot(get_object_owner(alias), None)

        ObjectAccess.objects.all().delete()
        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
        ret = management.call_command("modo", "repair", "--quiet", "--dry-run")
        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")
        models.Alias.objects.create(address='@modoboa.xxx')
        ret = management.call_command("modo", "repair", "--quiet")
        assert ret is None

<MSG> Increase coverage.

<DFF> @@ -28,7 +28,7 @@ class RepairTestCase(ModoTestCase):
         # fix it. run in quiet mode because we dont want output in tests
         ret = management.call_command("modo", "repair", "--quiet")
         assert ret is None
-        # assert its fixed
+        # assert it's fixed
         self.assertIsNot(get_object_owner(mbox), None)
         self.assertIsNot(get_object_owner(alias), None)
 
@@ -49,3 +49,14 @@ class RepairTestCase(ModoTestCase):
         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, 2)
+        ret = management.call_command("modo", "repair", "--quiet")
+        assert ret is None
+        self.assertTrue(
+            models.Alias.objects.filter(
+                address="user@test.com", internal=True).exists())

Increase coverage.