[fix]: Adding pytest-mock to test dependencies

gmsn-ita · Jul 20, 2021 · c7abda9 · c7abda9
1 parent f248e37
commit c7abda9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -24,6 +24,7 @@ skipsdist=True
 deps=
     pytest>=6.2.1
     numpy>=1.19.4
+    pytest-mock
 usedevelop=True
 commands=
     pytest

diff --git a/tests/unit/handlers/test_base_handler.py b/tests/unit/handlers/test_base_handler.py
@@ -40,10 +40,9 @@ def test_handle_when_next_handler_is_defined(mocker: MockerFixture):
     mocker.patch.object(next_handler,
                         'handle',
                         return_value='Test',
-                        autospec=True)
-    spy = mocker.spy(next_handler, 'handle')
+                        autospec=True,
+                        wraps=next_handler.handle)
 
     base_handler.set_next(next_handler)
 
     assert base_handler.handle({}) == 'Test'
-    assert spy.assert_called_once_with({})