From 4ca2981a30a5b363ec21ab4f66709b367af06925 Mon Sep 17 00:00:00 2001
From: Antoine Nguyen <tonio@ngyn.org>
Date: Sun, 17 Apr 2016 14:10:23 +0200
Subject: [PATCH] Increase coverage.

see #799
---
 modoboa/limits/tests.py | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/modoboa/limits/tests.py b/modoboa/limits/tests.py
index 0b6616a21..5abaa18ac 100644
--- a/modoboa/limits/tests.py
+++ b/modoboa/limits/tests.py
@@ -417,3 +417,37 @@ class ResellerTestCase(ResourceTestCase):
         self.assertEqual(resp, 'Not enough resources')
         self._check_limit('mailboxes', 1, 2)
         self._check_limit('mailbox_aliases', 0, 2)
+
+
+class DomainLimitsTestCase(ModoTestCase):
+    """Per-domain limits tests."""
+
+    @classmethod
+    def setUpTestData(cls):
+        """Create test data."""
+        super(DomainLimitsTestCase, cls).setUpTestData()
+        populate_database()
+        parameters.save_admin("ENABLE_DOMAIN_LIMITS", "yes")
+
+    def test_set_limits(self):
+        """Try to set limits for a given domain."""
+        domain = Domain.objects.get(name="test.com")
+        values = {
+            "name": domain.name, "quota": domain.quota,
+            "enabled": domain.enabled, "type": "domain",
+            "mailboxes_limit": 2, "mailbox_aliases_limit": 2,
+            "domain_aliases_limit": 2
+        }
+        self.ajax_post(
+            reverse("admin:domain_change", args=[domain.id]),
+            values
+        )
+        domain.refresh_from_db()
+        self.assertEqual(
+            domain.domainobjectlimit_set.get(name="mailboxes").max_value, 2)
+        self.assertEqual(
+            domain.domainobjectlimit_set.get(
+                name="mailbox_aliases").max_value, 2)
+        self.assertEqual(
+            domain.domainobjectlimit_set.get(
+                name="domain_aliases").max_value, 2)