From cafba0cf3891095a4122f7b8167fb5d7ed9adaf7 Mon Sep 17 00:00:00 2001 From: Benjamin Himes Date: Mon, 25 Nov 2024 15:03:40 +0200 Subject: [PATCH 1/2] Updates the Subtensor unit test object for mocking the websocket connection. --- tests/unit_tests/test_subtensor.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/tests/unit_tests/test_subtensor.py b/tests/unit_tests/test_subtensor.py index 73c63b30b9..669f15195e 100644 --- a/tests/unit_tests/test_subtensor.py +++ b/tests/unit_tests/test_subtensor.py @@ -256,6 +256,11 @@ def subtensor(mocker): mocker.patch.object( subtensor_module, "SubstrateInterface", return_value=fake_substrate ) + fake_websocket = mocker.MagicMock() + fake_websocket.client.connect.return_value = 0 + mocker.patch.object( + subtensor_module, "ws_client", return_value=fake_websocket + ) return Subtensor() From ba7aa4fe4367b24dd5420f7454a30f83f2f6c817 Mon Sep 17 00:00:00 2001 From: Benjamin Himes Date: Mon, 25 Nov 2024 16:02:27 +0200 Subject: [PATCH 2/2] Ruff --- tests/unit_tests/test_subtensor.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/unit_tests/test_subtensor.py b/tests/unit_tests/test_subtensor.py index 669f15195e..740df91884 100644 --- a/tests/unit_tests/test_subtensor.py +++ b/tests/unit_tests/test_subtensor.py @@ -258,9 +258,7 @@ def subtensor(mocker): ) fake_websocket = mocker.MagicMock() fake_websocket.client.connect.return_value = 0 - mocker.patch.object( - subtensor_module, "ws_client", return_value=fake_websocket - ) + mocker.patch.object(subtensor_module, "ws_client", return_value=fake_websocket) return Subtensor()