| # Verify the call arguments (checking by attributes rather than object identity) | # Verify the call arguments (checking by attributes rather than object identity) | ||||
| call_args = mock_processor.clean.call_args | call_args = mock_processor.clean.call_args | ||||
| assert call_args[0][0].id == dataset.id # First argument should be the dataset | assert call_args[0][0].id == dataset.id # First argument should be the dataset | ||||
| assert call_args[0][1] == [ | |||||
| segment.index_node_id for segment in segments | |||||
| ] # Second argument should be node IDs | |||||
| assert sorted(call_args[0][1]) == sorted( | |||||
| [segment.index_node_id for segment in segments] | |||||
| ) # Compare sorted lists to handle any order while preserving duplicates | |||||
| assert call_args[1]["with_keywords"] is True | assert call_args[1]["with_keywords"] is True | ||||
| assert call_args[1]["delete_child_chunks"] is False | assert call_args[1]["delete_child_chunks"] is False | ||||
| # Verify the call arguments | # Verify the call arguments | ||||
| call_args = mock_processor.clean.call_args | call_args = mock_processor.clean.call_args | ||||
| assert call_args[0][0].id == dataset.id # First argument should be the dataset | assert call_args[0][0].id == dataset.id # First argument should be the dataset | ||||
| assert call_args[0][1] == expected_node_ids # Second argument should be node IDs | |||||
| assert sorted(call_args[0][1]) == sorted( | |||||
| expected_node_ids | |||||
| ) # Compare sorted lists to handle any order while preserving duplicates | |||||
| assert call_args[1]["with_keywords"] is True | assert call_args[1]["with_keywords"] is True | ||||
| assert call_args[1]["delete_child_chunks"] is False | assert call_args[1]["delete_child_chunks"] is False | ||||