Skip to content

Commit fc0286d

Browse files
committed
fix(node): get nodes test
1 parent f0a1039 commit fc0286d

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/api/test_node.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
NodeUserUsage,
2121
User,
2222
)
23-
from app.models.node import NodeCreate, NodeResponse, NodeSettings
23+
from app.models.node import NodeCreate, NodeResponse, NodesResponse, NodeSettings
2424
from app.models.stats import (
2525
NodeRealtimeStats,
2626
NodeStats,
@@ -210,7 +210,8 @@ def test_get_usage_passes_filters(access_token, node_operator_mock):
210210

211211
def test_get_nodes_forwards_query_params(access_token, node_operator_mock):
212212
node_response = sample_node_response()
213-
node_operator_mock.get_db_nodes.return_value = [node_response]
213+
nodes_response = NodesResponse(nodes=[node_response], total=1)
214+
node_operator_mock.get_db_nodes.return_value = nodes_response
214215
params = [
215216
("core_id", "2"),
216217
("offset", "5"),
@@ -224,7 +225,7 @@ def test_get_nodes_forwards_query_params(access_token, node_operator_mock):
224225
]
225226
response = client.get("/api/nodes", headers=auth_headers(access_token), params=params)
226227
assert response.status_code == status.HTTP_200_OK
227-
assert response.json() == [node_response.model_dump()]
228+
assert response.json() == nodes_response.model_dump(mode="json")
228229

229230
awaited_kwargs = node_operator_mock.get_db_nodes.await_args.kwargs
230231
assert awaited_kwargs["core_id"] == 2

0 commit comments

Comments
 (0)