diff --git a/tests/tasks/conftest.py b/tests/tasks/conftest.py index df8d801e..0a12fa3e 100644 --- a/tests/tasks/conftest.py +++ b/tests/tasks/conftest.py @@ -21,7 +21,7 @@ def unreachable_task(): device = PollDevice(name="nonexist", address="127.0.0.1", community="invalid", port=666) state = ZinoState() task = ReachableTask(device, state) - with patch("zino.tasks.task.SNMP.get") as get_mock: + with patch("zino.snmp.SNMP.get") as get_mock: get_mock.side_effect = TimeoutError yield task task._deschedule_extra_job() diff --git a/tests/tasks/test_task.py b/tests/tasks/test_task.py index d48c68af..9e43b49e 100644 --- a/tests/tasks/test_task.py +++ b/tests/tasks/test_task.py @@ -25,7 +25,7 @@ async def test_get_sysuptime_raises_timeout_error(self): device = PollDevice(name="nonexist", address="127.0.0.1", community="invalid", port=666) state = ZinoState() task = ReachableTask(device, state) - with patch("zino.tasks.task.SNMP.get") as get_mock: + with patch("zino.snmp.SNMP.get") as get_mock: get_mock.side_effect = TimeoutError with pytest.raises(TimeoutError): await task._get_uptime() diff --git a/tests/tasks/test_vendor.py b/tests/tasks/test_vendor.py index 39deb529..b81d2a07 100644 --- a/tests/tasks/test_vendor.py +++ b/tests/tasks/test_vendor.py @@ -22,7 +22,7 @@ async def test_run_should_raise_exception_when_there_is_no_response(self): device = PollDevice(name="localhost", address="127.0.0.1", community="invalid", port=666) state = ZinoState() task = VendorTask(device, state) - with patch("zino.tasks.task.SNMP.get") as get_mock: + with patch("zino.snmp.SNMP.get") as get_mock: get_mock.side_effect = TimeoutError with pytest.raises(TimeoutError): await task.run()