44import os
55from unittest .mock import patch
66
7- from odoo_test_helper import FakeModelLoader
8-
7+ from odoo .orm .model_classes import add_to_registry
98from odoo .tests import tagged # type: ignore[import-untyped]
109from odoo .tools .config import config # type: ignore[import-untyped]
1110
1918# Test need to be run post install otherwise the _register_hook is not called yet
2019@tagged ("post_install" , "-at_install" )
2120class TestEnv (common .ServerEnvironmentCase ):
22- def setUp (self ):
23- super ().setUp ()
21+ @classmethod
22+ def setUpClass (cls ):
23+ super ().setUpClass ()
2424 # Load fake models ->/
25- self .loader = FakeModelLoader (self .env , self .__module__ )
26- self .loader .backup_registry ()
27-
2825 from .models import ExternalService , ExternalService2
2926
30- self .loader .update_registry ((ExternalService , ExternalService2 ))
31- self .env ["external_service" ].create (
32- [{"name" : "ftp2" , "description" : "another" }]
27+ add_to_registry (cls .registry , ExternalService )
28+ add_to_registry (cls .registry , ExternalService2 )
29+ cls .registry ._setup_models__ (
30+ cls .env .cr , ["external_service" , "external_service_2" ]
31+ )
32+ cls .registry .init_models (
33+ cls .env .cr ,
34+ ["external_service" , "external_service_2" ],
35+ {"models_to_check" : True },
3336 )
37+ cls .addClassCleanup (cls .registry .__delitem__ , "external_service" )
38+ cls .addClassCleanup (cls .registry .__delitem__ , "external_service_2" )
3439
35- def tearDown (self ):
36- self .loader .restore_registry ()
37- super ().tearDown ()
40+ cls .env ["external_service" ].create ([{"name" : "ftp2" , "description" : "another" }])
3841
3942 @patch .dict (config .options , {"running_env" : "autocreate" })
4043 def test_autocreate (self ):
@@ -44,8 +47,6 @@ def test_autocreate(self):
4447 parser = server_env ._load_config ()
4548 server_env_mixin .serv_config = parser
4649 server_env_mixin_2 .serv_config = parser
47- # Needed to force _register_hook with auto creation
48- self .loader .update_registry (tuple ())
4950
5051 # auto created record
5152 record = self .env .ref (
0 commit comments