File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 2020 NodeUserUsage ,
2121 User ,
2222)
23- from app .models .node import NodeCreate , NodeResponse , NodeSettings
23+ from app .models .node import NodeCreate , NodeResponse , NodesResponse , NodeSettings
2424from app .models .stats import (
2525 NodeRealtimeStats ,
2626 NodeStats ,
@@ -210,7 +210,8 @@ def test_get_usage_passes_filters(access_token, node_operator_mock):
210210
211211def 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
You can’t perform that action at this time.
0 commit comments