Fixed unit tests

thewhaleking · thewhaleking · commit b5ccbef42325 · 2025-07-18T17:00:48.000+02:00
diff --git a/tests/unit_tests/test_async_subtensor.py b/tests/unit_tests/test_async_subtensor.py
@@ -3570,52 +3570,75 @@ async def test_get_liquidity_list_happy_path(subtensor, fake_wallet, mocker):
         return_value=(Balance.from_tao(0.0), Balance.from_tao(0.0, netuid)),
     )
 
-    mocked_substrate_query = mocker.AsyncMock(
+    mocked_substrate_query_multi = mocker.AsyncMock(
         side_effect=[
-            # for gather
-            {"bits": 0},
-            {"bits": 0},
-            {"bits": 18446744073709551616},
-            # for loop
-            {
-                "liquidity_net": 1000000000000,
-                "liquidity_gross": 1000000000000,
-                "fees_out_tao": {"bits": 0},
-                "fees_out_alpha": {"bits": 0},
-            },
-            {
-                "liquidity_net": -1000000000000,
-                "liquidity_gross": 1000000000000,
-                "fees_out_tao": {"bits": 0},
-                "fees_out_alpha": {"bits": 0},
-            },
-            {
-                "liquidity_net": 1000000000000,
-                "liquidity_gross": 1000000000000,
-                "fees_out_tao": {"bits": 0},
-                "fees_out_alpha": {"bits": 0},
-            },
-            {
-                "liquidity_net": -1000000000000,
-                "liquidity_gross": 1000000000000,
-                "fees_out_tao": {"bits": 0},
-                "fees_out_alpha": {"bits": 0},
-            },
-            {
-                "liquidity_net": 1000000000000,
-                "liquidity_gross": 1000000000000,
-                "fees_out_tao": {"bits": 0},
-                "fees_out_alpha": {"bits": 0},
-            },
-            {
-                "liquidity_net": -1000000000000,
-                "liquidity_gross": 1000000000000,
-                "fees_out_tao": {"bits": 0},
-                "fees_out_alpha": {"bits": 0},
-            },
+            [
+                (None, {"bits": 0}),
+                (None, {"bits": 0}),
+                (None, {"bits": 18446744073709551616}),
+            ],
+            [
+                (
+                    None,
+                    {
+                        "liquidity_net": 1000000000000,
+                        "liquidity_gross": 1000000000000,
+                        "fees_out_tao": {"bits": 0},
+                        "fees_out_alpha": {"bits": 0},
+                    },
+                ),
+                (
+                    None,
+                    {
+                        "liquidity_net": -1000000000000,
+                        "liquidity_gross": 1000000000000,
+                        "fees_out_tao": {"bits": 0},
+                        "fees_out_alpha": {"bits": 0},
+                    },
+                ),
+                (
+                    None,
+                    {
+                        "liquidity_net": 1000000000000,
+                        "liquidity_gross": 1000000000000,
+                        "fees_out_tao": {"bits": 0},
+                        "fees_out_alpha": {"bits": 0},
+                    },
+                ),
+                (
+                    None,
+                    {
+                        "liquidity_net": -1000000000000,
+                        "liquidity_gross": 1000000000000,
+                        "fees_out_tao": {"bits": 0},
+                        "fees_out_alpha": {"bits": 0},
+                    },
+                ),
+                (
+                    None,
+                    {
+                        "liquidity_net": 1000000000000,
+                        "liquidity_gross": 1000000000000,
+                        "fees_out_tao": {"bits": 0},
+                        "fees_out_alpha": {"bits": 0},
+                    },
+                ),
+                (
+                    None,
+                    {
+                        "liquidity_net": -1000000000000,
+                        "liquidity_gross": 1000000000000,
+                        "fees_out_tao": {"bits": 0},
+                        "fees_out_alpha": {"bits": 0},
+                    },
+                ),
+            ],
         ]
     )
-    mocker.patch.object(subtensor.substrate, "query", mocked_substrate_query)
+
+    mocker.patch.object(
+        subtensor.substrate, "query_multi", mocked_substrate_query_multi
+    )
 
     fake_positions = [
         [
diff --git a/tests/unit_tests/test_subtensor.py b/tests/unit_tests/test_subtensor.py
@@ -954,8 +954,7 @@ def test_state_call(subtensor, mocker):
 
     # Asserts
     subtensor.substrate.rpc_request.assert_called_once_with(
-        method="state_call",
-        params=[fake_method, fake_data],
+        method="state_call", params=[fake_method, fake_data], block_hash=None
     )
     assert result == subtensor.substrate.rpc_request.return_value
 

Original file line number	Diff line number	Diff line change
`@@ -954,8 +954,7 @@ def test_state_call(subtensor, mocker):`
`954`	`954`
`955`	`955`	`# Asserts`
`956`	`956`	`subtensor.substrate.rpc_request.assert_called_once_with(`
`957`		`- method="state_call",`
`958`		`- params=[fake_method, fake_data],`
	`957`	`+ method="state_call", params=[fake_method, fake_data], block_hash=None`
`959`	`958`	`)`
`960`	`959`	`assert result == subtensor.substrate.rpc_request.return_value`
`961`	`960`