From 329ea62aac85c68bb358e8f4098f7e7544d7fbee Mon Sep 17 00:00:00 2001 From: Spitap Date: Fri, 24 Feb 2023 16:09:34 +0100 Subject: [PATCH] Updated test --- modoboa/admin/tests/test_domain.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/modoboa/admin/tests/test_domain.py b/modoboa/admin/tests/test_domain.py index 97c1e97a0..81db43495 100644 --- a/modoboa/admin/tests/test_domain.py +++ b/modoboa/admin/tests/test_domain.py @@ -450,18 +450,24 @@ class DKIMTestCase(ModoTestCase): self.set_global_parameter("dkim_keys_storage_dir", "/I_DONT_EXIST") call_command("modo", "manage_dkim_keys") domain = Domain.objects.filter(name=values["name"]) - alarm = Alarm.objects.get(domain__in=domain, - internal_name=constants.DKIM_WRITE_ERROR) # Closing the alarm and retry generating dkim keys - alarm.close() + Alarm.objects.get(domain__in=domain, + internal_name=constants.DKIM_WRITE_ERROR + ).close() call_command("modo", "manage_dkim_keys") - self.assertEqual(alarm.status, constants.ALARM_OPENED) + self.assertEqual( + Alarm.objects.get(domain__in=domain, + internal_name=constants.DKIM_WRITE_ERROR).status, + constants.ALARM_OPENED) # Generate normally self.set_global_parameter("dkim_keys_storage_dir", self.workdir) call_command("modo", "manage_dkim_keys") - self.assertEqual(alarm.status, constants.ALARM_CLOSED) + self.assertEqual( + Alarm.objects.get(domain__in=domain, + internal_name=constants.DKIM_WRITE_ERROR).status, + constants.ALARM_CLOSED) key_path = os.path.join(self.workdir, "{}.pem".format(values["name"])) self.assertTrue(os.path.exists(key_path)) domain = Domain.objects.get(name=values["name"])