Ensure that type validation is executed for StrictMock instances as well

facebook · Feb 20, 2021 · 3e19346 · 3e19346
1 parent 7ab0b50
commit 3e19346
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/testslide/mock_callable.py b/testslide/mock_callable.py
@@ -729,10 +729,9 @@ def _patch(
         else:
             original_callable = getattr(self._target, self._method)
 
-        if not isinstance(self._target, StrictMock):
-            new_value = _wrap_signature_and_type_validation(
-                new_value, self._target, self._method, self.type_validation
-            )
+        new_value = _wrap_signature_and_type_validation(
+            new_value, self._target, self._method, self.type_validation
+        )
 
         restore = self._method in self._target.__dict__
         restore_value = self._target.__dict__.get(self._method, None)