@@ -421,12 +421,42 @@ async def test_collect_all_components(self):
421421 """Test collecting statistics from all components."""
422422 mock_suite = Mock ()
423423
424- # Mock order manager
424+ # Mock order manager - get_order_statistics returns a dict/TypedDict
425425 mock_suite .orders = Mock ()
426426 mock_suite .orders .get_order_statistics .return_value = {
427427 "orders_placed" : 10 ,
428428 "orders_filled" : 8 ,
429+ "orders_cancelled" : 1 ,
430+ "orders_rejected" : 1 ,
431+ "orders_modified" : 2 ,
429432 "fill_rate" : 0.8 ,
433+ "avg_fill_time_ms" : 150.0 ,
434+ "total_volume_traded" : 1000 ,
435+ "total_commission" : 25.0 ,
436+ "api_calls" : 20 ,
437+ "cached_responses" : 5 ,
438+ "realtime_connected" : True ,
439+ "active_orders" : 2 ,
440+ "pending_orders" : 1 ,
441+ "position_orders" : 3 ,
442+ "bracket_orders" : 1 ,
443+ "oco_orders" : 0 ,
444+ "avg_order_size" : 50 ,
445+ "max_order_size" : 100 ,
446+ "min_order_size" : 10 ,
447+ "cancel_rate" : 0.1 ,
448+ "reject_rate" : 0.1 ,
449+ "modify_rate" : 0.2 ,
450+ "avg_slippage" : 0.5 ,
451+ "positive_slippage_rate" : 0.3 ,
452+ "order_latency_p50" : 100.0 ,
453+ "order_latency_p95" : 200.0 ,
454+ "order_latency_p99" : 300.0 ,
455+ "memory_usage_mb" : 5.2 ,
456+ "error_count" : 0 ,
457+ "last_order_time" : None ,
458+ "last_fill_time" : None ,
459+ "tracking_accuracy" : 0.95 ,
430460 }
431461
432462 # Mock position manager
@@ -438,12 +468,14 @@ async def test_collect_all_components(self):
438468 }
439469 )
440470
441- # Mock data manager
471+ # Mock data manager - get_memory_stats is async
442472 mock_suite .data = Mock ()
443- mock_suite .data .get_memory_stats .return_value = {
444- "bars_processed" : 1000 ,
445- "memory_usage_mb" : 5.2 ,
446- }
473+ mock_suite .data .get_memory_stats = AsyncMock (
474+ return_value = {
475+ "bars_processed" : 1000 ,
476+ "memory_usage_mb" : 5.2 ,
477+ }
478+ )
447479
448480 # Mock components that don't exist
449481 mock_suite .orderbook = None
0 commit comments