We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ed097dd commit 36c0e5eCopy full SHA for 36c0e5e
src/testing.py
@@ -344,8 +344,11 @@ def _setup_registry(self):
344
super(IntegrityCase, self)._setup_registry()
345
cr = self.registry.cursor()
346
self.addCleanup(cr.close)
347
- self.registry.enter_test_mode(cr)
348
- self.addCleanup(self.registry.leave_test_mode)
+ if hasattr(self, "registry_enter_test_mode"):
+ self.registry_enter_test_mode(cr=cr)
349
+ else:
350
+ self.registry.enter_test_mode(cr)
351
+ self.addCleanup(self.registry.leave_test_mode)
352
353
def setUp(self):
354
super(IntegrityCase, self).setUp()
0 commit comments