@@ -288,6 +288,10 @@ def _create_mock_pipeline(self) -> SimulationPipeline:
288288 pipeline .spectrum_algorithm = Mock (spec = AbstractSpectrumAssigner )
289289 pipeline .snr_algorithm = Mock (spec = AbstractSNRMeasurer )
290290
291+ # Setup route_props for routing algorithm
292+ pipeline .routing_algorithm .route_props = Mock ()
293+ pipeline .routing_algorithm .route_props .paths_matrix = []
294+
291295 # Setup get_metrics
292296 pipeline .routing_algorithm .get_metrics .return_value = {}
293297 pipeline .spectrum_algorithm .get_metrics .return_value = {}
@@ -299,7 +303,8 @@ def test_process_request_with_successful_allocation(self) -> None:
299303 """Test process_request with successful spectrum allocation."""
300304 # Arrange
301305 pipeline = self ._create_mock_pipeline ()
302- pipeline .routing_algorithm .route .return_value = [1 , 2 , 3 ] # type: ignore[attr-defined]
306+ # Setup paths_matrix to contain the path found by routing
307+ pipeline .routing_algorithm .route_props .paths_matrix = [[1 , 2 , 3 ]]
303308 pipeline .spectrum_algorithm .assign .return_value = { # type: ignore[attr-defined]
304309 "start_slot" : 0 ,
305310 "end_slot" : 5 ,
@@ -356,6 +361,9 @@ def _create_mock_pipeline(self) -> SimulationPipeline:
356361 pipeline .routing_algorithm = Mock (spec = AbstractRoutingAlgorithm )
357362 pipeline .spectrum_algorithm = Mock (spec = AbstractSpectrumAssigner )
358363 pipeline .snr_algorithm = Mock (spec = AbstractSNRMeasurer )
364+ # Setup route_props for routing algorithm
365+ pipeline .routing_algorithm .route_props = Mock ()
366+ pipeline .routing_algorithm .route_props .paths_matrix = []
359367 pipeline .routing_algorithm .get_metrics .return_value = {}
360368 pipeline .spectrum_algorithm .get_metrics .return_value = {}
361369 pipeline .snr_algorithm .get_metrics .return_value = {}
@@ -365,7 +373,8 @@ def test_process_request_when_no_path_found(self) -> None:
365373 """Test process_request when no path is found."""
366374 # Arrange
367375 pipeline = self ._create_mock_pipeline ()
368- pipeline .routing_algorithm .route .return_value = None # type: ignore[attr-defined]
376+ # Empty paths_matrix indicates no path was found
377+ pipeline .routing_algorithm .route_props .paths_matrix = []
369378 request = Mock ()
370379
371380 # Act
@@ -380,7 +389,7 @@ def test_process_request_when_no_spectrum_available(self) -> None:
380389 """Test process_request when no spectrum is available."""
381390 # Arrange
382391 pipeline = self ._create_mock_pipeline ()
383- pipeline .routing_algorithm .route . return_value = [1 , 2 , 3 ] # type: ignore[attr-defined ]
392+ pipeline .routing_algorithm .route_props . paths_matrix = [[ 1 , 2 , 3 ]]
384393 pipeline .spectrum_algorithm .assign .return_value = None # type: ignore[attr-defined]
385394 request = Mock ()
386395
@@ -395,7 +404,7 @@ def test_process_request_when_snr_too_low(self) -> None:
395404 """Test process_request when SNR is below threshold."""
396405 # Arrange
397406 pipeline = self ._create_mock_pipeline ()
398- pipeline .routing_algorithm .route . return_value = [1 , 2 , 3 ] # type: ignore[attr-defined ]
407+ pipeline .routing_algorithm .route_props . paths_matrix = [[ 1 , 2 , 3 ]]
399408 pipeline .spectrum_algorithm .assign .return_value = { # type: ignore[attr-defined]
400409 "start_slot" : 0 ,
401410 "end_slot" : 5 ,
@@ -428,7 +437,7 @@ def test_process_request_when_allocation_fails(self) -> None:
428437 """Test process_request when spectrum allocation fails."""
429438 # Arrange
430439 pipeline = self ._create_mock_pipeline ()
431- pipeline .routing_algorithm .route . return_value = [1 , 2 , 3 ] # type: ignore[attr-defined ]
440+ pipeline .routing_algorithm .route_props . paths_matrix = [[ 1 , 2 , 3 ]]
432441 pipeline .spectrum_algorithm .assign .return_value = { # type: ignore[attr-defined]
433442 "start_slot" : 0 ,
434443 "end_slot" : 5 ,
@@ -660,7 +669,8 @@ def test_pipeline_handles_missing_topology_in_engine_props(self) -> None:
660669 }
661670 pipeline = SimulationPipeline (config )
662671 pipeline .routing_algorithm = Mock (spec = AbstractRoutingAlgorithm )
663- pipeline .routing_algorithm .route .return_value = [1 , 2 ]
672+ pipeline .routing_algorithm .route_props = Mock ()
673+ pipeline .routing_algorithm .route_props .paths_matrix = [[1 , 2 ]]
664674 pipeline .routing_algorithm .get_metrics .return_value = {}
665675 pipeline .spectrum_algorithm = Mock (spec = AbstractSpectrumAssigner )
666676 pipeline .spectrum_algorithm .assign .return_value = {
0 commit comments