diff --git a/modoboa/core/tests/stupid_extension_1/modo_extension.py b/modoboa/core/tests/stupid_extension_1/modo_extension.py index 6add2bf75..4cc7f4121 100644 --- a/modoboa/core/tests/stupid_extension_1/modo_extension.py +++ b/modoboa/core/tests/stupid_extension_1/modo_extension.py @@ -15,6 +15,6 @@ class StupidExtension1(extensions.ModoExtension): pass def load_initial_data(self): - pass + raise RuntimeError extensions.exts_pool.register_extension(StupidExtension1) diff --git a/modoboa/core/tests/stupid_extension_2/modo_extension.py b/modoboa/core/tests/stupid_extension_2/modo_extension.py index e9485c922..47824e924 100644 --- a/modoboa/core/tests/stupid_extension_2/modo_extension.py +++ b/modoboa/core/tests/stupid_extension_2/modo_extension.py @@ -11,10 +11,4 @@ class StupidExtension2(extensions.ModoExtension): version = "1.0.0" description = "A stupid extension" - def load(self): - pass - - def load_initial_data(self): - pass - extensions.exts_pool.register_extension(StupidExtension2, show=False) diff --git a/modoboa/core/tests/test_extensions.py b/modoboa/core/tests/test_extensions.py index b48889215..94fec8479 100644 --- a/modoboa/core/tests/test_extensions.py +++ b/modoboa/core/tests/test_extensions.py @@ -3,6 +3,9 @@ import os import sys +from testfixtures import compare + +from django.core import management from django.test import TestCase, override_settings from .. import extensions @@ -55,9 +58,24 @@ class ExtensionTestCase(TestCase): instance = self.pool.get_extension("stupid_extension_1") self.assertEqual(instance.__class__.__name__, "StupidExtension1") + def test_get_extension_infos(self): + """Check getter method.""" + self.assertIsNone(self.pool.get_extension_infos("toto")) + infos = self.pool.get_extension_infos("stupid_extension_1") + compare(infos, { + "name": "stupid_extension_1", "label": "Stupid extension", + "version": "1.0.0", "description": "A stupid extension", + "url": None, "always_active": False + }) + def test_list_all(self): """Check list_all method.""" result = self.pool.list_all() self.assertEqual(len(result), 1) ext = result[0] self.assertEqual(ext["id"], "stupid_extension_1") + + def test_load_initial_data(self): + """Check if method is called.""" + with self.assertRaises(RuntimeError): + management.call_command("load_initial_data")