Merge 138e22d into 33bb75b

JeffLIrion · Nov 29, 2020 · 40dc326 · 40dc326
2 parents 33bb75b + 138e22d
commit 40dc326
Show file tree

Hide file tree

Showing 4 changed files with 87 additions and 1,249 deletions.
diff --git a/tests/patchers.py b/tests/patchers.py
@@ -143,3 +143,21 @@ class CustomException(Exception):
 
 
 PATCH_CONNECT_FAIL_CUSTOM_EXCEPTION = {KEY_PYTHON: patch("{}.{}.connect".format(__name__, ADB_DEVICE_TCP_FAKE), side_effect=CustomException), KEY_SERVER: patch("{}.{}.device".format(__name__, CLIENT_FAKE_SUCCESS), side_effect=CustomException)}
+
+
+def patch_calls(obj, wraps):
+    """Patch a method call without changing its behavior.
+
+    Parameters
+    ----------
+    obj
+        The object whose method will be patched (i.e., `self`)
+    wraps
+        The method that is being patched (i.e., `self.method`)
+
+    Returns
+    -------
+    The patched method
+
+    """
+    return patch.object(type(obj), wraps.__name__.split()[-1], wraps=wraps)