Fix unittests

Author: fkantelberg

tests/test_action.py

@@ -7,7 +7,6 @@
 from unittest import mock
 
 import pytest
-
 from doblib.action import ALNUM, ActionEnvironment
 
 
@@ -426,6 +425,8 @@ def test_apply_action(env):
 
     odoo = sys.modules["odoo"] = mock.MagicMock()
     sys.modules["odoo.tools"] = mock.MagicMock()
+    sys.modules["odoo.modules"] = mock.MagicMock()
+    sys.modules["odoo.modules.registry"] = mock.MagicMock()
     sys.modules["odoo.release"] = odoo.release
     odoo.release.version_info = (14, 0)
     env._init_odoo.return_value = True

tests/test_ci.py

@@ -6,7 +6,6 @@
 from unittest import mock
 
 import pytest
-
 from doblib import base
 from doblib.ci import CIEnvironment
 
@@ -26,6 +25,8 @@ def env():
 def test_test(pytest_mock, env):
     odoo = sys.modules["odoo"] = mock.MagicMock()
     tools = sys.modules["odoo.tools"] = mock.MagicMock()
+    sys.modules["odoo.modules"] = mock.MagicMock()
+    sys.modules["odoo.modules.registry"] = mock.MagicMock()
     sys.modules["odoo.release"] = odoo.release
     odoo.release.version_info = (14, 0)
 

tests/test_environment.py

@@ -8,7 +8,6 @@
 from unittest import mock
 
 import pytest
-
 from doblib import base
 from doblib.env import Environment, load_config_arguments
 
@@ -111,7 +110,9 @@ def test_env(env):
             pass
 
     odoo = sys.modules["odoo"] = mock.MagicMock()
-    reg = odoo.registry.return_value = mock.MagicMock()
+    sys.modules["odoo.modules"] = mock.MagicMock()
+    registry = sys.modules["odoo.modules.registry"] = mock.MagicMock()
+    reg = registry.Registry.return_value = mock.MagicMock()
     cr = reg.cursor.return_value = mock.MagicMock()
 
     # Test the normal commit

tests/test_migrate.py

@@ -6,7 +6,6 @@
 from unittest import mock
 
 import pytest
-
 from doblib import utils
 from doblib.migrate import MigrateEnvironment
 
@@ -24,6 +23,8 @@ def env():
 def test_migrate(repos, env):
     odoo = sys.modules["odoo"] = mock.MagicMock()
     tools = sys.modules["odoo.tools"] = mock.MagicMock()
+    sys.modules["odoo.modules"] = mock.MagicMock()
+    sys.modules["odoo.modules.registry"] = mock.MagicMock()
     tools.config.__getitem__.return_value = "odoo"
     sys.modules["odoo.release"] = odoo.release
     odoo.release.version_info = (14, 0)

tests/test_module.py

@@ -7,7 +7,6 @@
 from unittest import mock
 
 import pytest
-
 from doblib import base
 from doblib.module import ModuleEnvironment, no_flags
 
@@ -86,6 +85,8 @@ def test_get_installed_modules(env):
 def test_install_all(env):
     odoo = sys.modules["odoo"] = mock.MagicMock()
     sys.modules["odoo.tools"] = mock.MagicMock()
+    sys.modules["odoo.modules"] = mock.MagicMock()
+    sys.modules["odoo.modules.registry"] = mock.MagicMock()
 
     env.install_all("odoo", ["module"])
     odoo.modules.registry.Registry.new.assert_called_once_with(
@@ -101,6 +102,8 @@ def test_install_all(env):
 def test_update_all(env):
     odoo = sys.modules["odoo"] = mock.MagicMock()
     sys.modules["odoo.tools"] = mock.MagicMock()
+    sys.modules["odoo.modules"] = mock.MagicMock()
+    sys.modules["odoo.modules.registry"] = mock.MagicMock()
 
     env.update_specific("odoo", installed=True)
     odoo.modules.registry.Registry.new.assert_called_once_with(
@@ -112,6 +115,8 @@ def test_update_all(env):
 def test_update_listed(env):
     odoo = sys.modules["odoo"] = mock.MagicMock()
     sys.modules["odoo.tools"] = mock.MagicMock()
+    sys.modules["odoo.modules"] = mock.MagicMock()
+    sys.modules["odoo.modules.registry"] = mock.MagicMock()
     env._get_modules = mock.MagicMock()
 
     env.update_specific("odoo", listed=True)
@@ -143,6 +148,8 @@ def test_update(env):
     # Quite complex and we have to mock plenty of stuff
     odoo = sys.modules["odoo"] = mock.MagicMock()
     tools = sys.modules["odoo.tools"] = mock.MagicMock()
+    sys.modules["odoo.modules"] = mock.MagicMock()
+    sys.modules["odoo.modules.registry"] = mock.MagicMock()
     sys.modules["odoo.release"] = odoo.release
     tools.config.__getitem__.return_value = "odoo"
     odoo.release.version_info = (14, 0)