33
44from logging import getLogger
55
6- from odoo_test_helper import FakeModelLoader
7-
86from odoo import api , fields , models
7+ from odoo .orm .model_classes import add_to_registry
98from odoo .tests import TransactionCase
109from odoo .tools .misc import mute_logger
1110
@@ -24,9 +23,6 @@ def setUpClass(cls):
2423 cls .env ["base" ]._register_hook ()
2524
2625 # Load test model
27- cls .loader = FakeModelLoader (cls .env , cls .__module__ )
28- cls .loader .backup_registry ()
29-
3026 class BWDTestModel (models .Model ):
3127 _name = "bwd.test.model"
3228 _description = "Base Write Diff - Test Model"
@@ -68,12 +64,12 @@ def _compute_volume(self):
6864 for rec in self :
6965 rec .volume = rec .area * rec .height
7066
71- cls .loader . update_registry ([ BWDTestModel ] )
72-
73- @ classmethod
74- def tearDownClass ( cls ):
75- cls . loader . restore_registry ( )
76- super (). tearDownClass ( )
67+ add_to_registry ( cls .registry , BWDTestModel )
68+ cls . registry . _setup_models__ ( cls . env . cr , [ "bwd.test.model" ])
69+ cls . registry . init_models (
70+ cls . env . cr , [ "bwd.test.model" ], { "models_to_check" : True }
71+ )
72+ cls . addClassCleanup ( cls . registry . __delitem__ , "bwd.test.model" )
7773
7874 def _create_records (self , count = 1 ):
7975 records = self .env ["bwd.test.model" ].create ([{} for _ in range (1 , count + 1 )])
0 commit comments