Test: Refactor test concurrency handling and add SDK chunk management tests (#8112)

### What problem does this PR solve?

- Improve concurrent test cases by using as_completed for better
reliability
- Rename variables for clarity (chunk_num -> count)
- Add new SDK API test suite for chunk management operations
- Update HTTP API tests with consistent concurrency patterns

### Type of change

- [x] Add test cases
- [x] Refactoring
This commit is contained in:
Liu An
2025-06-06 19:43:14 +08:00
committed by GitHub
parent 157cd8b1b0
commit 5825a24d26
21 changed files with 946 additions and 55 deletions

View File

@ -85,7 +85,7 @@ class TestCapability:
futures = [executor.submit(create_dataset, api_key, {"name": f"dataset_{i}"}) for i in range(count)]
responses = list(as_completed(futures))
assert len(responses) == count, responses
assert all(futures.result()["code"] == 0 for futures in futures)
assert all(future.result()["code"] == 0 for future in futures)
@pytest.mark.usefixtures("clear_datasets")

View File

@ -93,7 +93,7 @@ class TestCapability:
futures = [executor.submit(delete_datasets, api_key, {"ids": ids[i : i + 1]}) for i in range(count)]
responses = list(as_completed(futures))
assert len(responses) == count, responses
assert all(futures.result()["code"] == 0 for futures in futures)
assert all(future.result()["code"] == 0 for future in futures)
class TestDatasetsDelete:

View File

@ -49,7 +49,7 @@ class TestCapability:
futures = [executor.submit(list_datasets, api_key) for i in range(count)]
responses = list(as_completed(futures))
assert len(responses) == count, responses
assert all(futures.result()["code"] == 0 for futures in futures)
assert all(future.result()["code"] == 0 for future in futures)
@pytest.mark.usefixtures("add_datasets")

View File

@ -95,7 +95,7 @@ class TestCapability:
futures = [executor.submit(update_dataset, api_key, dataset_id, {"name": f"dataset_{i}"}) for i in range(count)]
responses = list(as_completed(futures))
assert len(responses) == count, responses
assert all(futures.result()["code"] == 0 for futures in futures)
assert all(future.result()["code"] == 0 for future in futures)
class TestDatasetUpdate: