From 6951c0b6c9a4271962e3ac2698148b47b48c2ae6 Mon Sep 17 00:00:00 2001 From: Morten Brekkevold Date: Fri, 24 Jan 2025 09:43:22 +0000 Subject: [PATCH] Fix obsolete patch statements These test patches will not work now that everything uses `zino.snmp.get_snmp_session()` - the SNMP class needs to be patched in the correct location. --- tests/tasks/conftest.py | 2 +- tests/tasks/test_task.py | 2 +- tests/tasks/test_vendor.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/tasks/conftest.py b/tests/tasks/conftest.py index df8d801e0..0a12fa3e4 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 d48c68af0..9e43b49e7 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 39deb5293..b81d2a077 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()