jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(tests): resolve order dependency in disable_segments_from_index_task tests (#25737)

Browse source
This commit is contained in:
-LAN- 2025-09-16 08:26:52 +08:00 • committed by GitHub
commit b98b389baf
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
api/tests/test_containers_integration_tests/tasks/test_disable_segments_from_index_task.py
View file

@ -290,9 +290,9 @@ class TestDisableSegmentsFromIndexTask:
# 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] == [ assert sorted(call_args[0][1]) == sorted(
segment.index_node_id for segment in segments [segment.index_node_id for segment in segments]
] # Second argument should be 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
@ -719,7 +719,9 @@ class TestDisableSegmentsFromIndexTask:
# 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