Used addCleanup() in tests where appropriate.

This commit is contained in:
Mariusz Felisiak 2023-12-31 10:01:31 +01:00 committed by GitHub
parent 81ccf92f15
commit d88ec42bd0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
37 changed files with 121 additions and 240 deletions

View file

@ -64,12 +64,7 @@ class FuncTestMixin:
vendor_impl = "as_" + connection.vendor
__getattribute__original = Func.__getattribute__
self.func_patcher = mock.patch.object(
Func, "__getattribute__", __getattribute__
)
self.func_patcher.start()
func_patcher = mock.patch.object(Func, "__getattribute__", __getattribute__)
func_patcher.start()
self.addCleanup(func_patcher.stop)
super().setUp()
def tearDown(self):
super().tearDown()
self.func_patcher.stop()