FixFewTests-part2 by xinlian12 · Pull Request #47933 · Azure/azure-sdk-for-java

xinlian12 · 2026-02-06T20:41:18Z

1. Test cleanup refactoring — bulk operations replace sequential deletes

The biggest change is in both TestSuiteBase classes (under rx/ and implementation/). The truncateCollection / cleanUpContainer methods are refactored to use Cosmos bulk operations (executeBulkOperations with CosmosBulkOperations.getDeleteItemOperation) instead of individual flatMap→deleteItem calls.

2. Bulk insert refactoring

The bulkInsert method is reworked to use the Cosmos bulk execution API (CosmosBulkOperations). A new insertAllItemsBlocking method is introduced, accepting a bulkEnabled parameter to choose between bulk and sequential insert.

3. Bulk insert concurrency reduced

DEFAULT_BULK_INSERT_CONCURRENCY_LEVEL dropped from 500 → 5, likely to reduce throttling in test environments.

4. Test call-site updates (20+ test files)

All test files that called truncateCollection or bulkInsertBlocking are updated to use the renamed methods (cleanUpContainer, insertAllItemsBlocking).

Copilot

Pull request overview

This PR updates Cosmos DB test infrastructure to use bulk execution APIs for container cleanup/truncation and adjusts several change feed tests to use the updated bulk insert helper.

Changes:

Switched test container cleanup/truncation from per-item deletes to executeBulkOperations.
Refactored bulkInsert in rx/TestSuiteBase to use bulk operations and updated change feed tests to match the new signature.
Minor test and code cleanup (helper rename, redundant exception catch removal, whitespace-only diff).

Reviewed changes

Copilot reviewed 7 out of 8 changed files in this pull request and generated 11 comments.

Show a summary per file

File	Description
sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/RxDocumentClientImpl.java	Whitespace-only diff in client initialization block.
sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/rx/changefeed/pkversion/IncrementalChangeFeedProcessorTest.java	Updated calls to `bulkInsert` to match new helper signature.
sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/rx/changefeed/epkversion/IncrementalChangeFeedProcessorTest.java	Updated calls to `bulkInsert` to match new helper signature.
sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/rx/changefeed/epkversion/FullFidelityChangeFeedProcessorTest.java	Updated calls to `bulkInsert` to match new helper signature.
sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/rx/TestSuiteBase.java	Reworked cleanup/truncation and bulk insert helpers to use bulk execution APIs.
sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/rx/OrderbyDocumentQueryTest.java	Renamed a local helper method to avoid confusion with updated `bulkInsert`.
sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/implementation/TestSuiteBase.java	Updated legacy (DocumentClient-based) truncation to delete documents via bulk execution.
sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/CosmosDiagnosticsTest.java	Removed redundant `JsonMappingException` catch (covered by `JsonProcessingException`).

sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/rx/TestSuiteBase.java

sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/implementation/TestSuiteBase.java

sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/RxDocumentClientImpl.java

sdk/cosmos/azure-cosmos-tests/src/test/java/com/azure/cosmos/rx/TestSuiteBase.java

- Add isSuccessStatusCode() validation for non-2xx bulk responses in all 3 sites (impl/TestSuiteBase truncateCollection, rx/TestSuiteBase cleanUpContainerInternal, rx/TestSuiteBase bulkInsert) - Keep 409/Conflict handling in bulkInsert but validate other non-2xx responses - Return server-created items from bulkInsertBlocking via getItem(clazz) - Use PartitionKey.NONE instead of new PartitionKey(null) in cleanUpContainerInternal

…agation Use BridgeInternal.createCosmosException() with status code and substatus code from CosmosBulkItemResponse instead of generic IllegalStateException.

xinlian12 · 2026-02-06T21:43:27Z

/azp run java - cosmos - tests

azure-pipelines · 2026-02-06T21:43:53Z

Azure Pipelines successfully started running 1 pipeline(s).

xinlian12 · 2026-02-06T23:19:04Z

/azp run java - cosmos - tests

azure-pipelines · 2026-02-06T23:19:28Z

Azure Pipelines successfully started running 1 pipeline(s).

- Add insertUsingPointOperations for one-by-one createItem path - Add bulkEnabled parameter to insertAllItemsBlocking/voidInsertAllItemsBlocking - Rename bulkInsertBlocking -> insertAllItemsBlocking - Rename voidBulkInsertBlocking -> voidInsertAllItemsBlocking - Update all callers with explicit bulkEnabled=true

xinlian12 · 2026-02-07T03:37:40Z

/azp run java - cosmos - tests

azure-pipelines · 2026-02-07T03:38:05Z

Azure Pipelines successfully started running 1 pipeline(s).

xinlian12 · 2026-02-08T21:35:20Z

/azp run java - cosmos - tests

azure-pipelines · 2026-02-08T21:35:47Z

Azure Pipelines successfully started running 1 pipeline(s).

kushagraThapar

LGTM, thanks @xinlian12

xinlian12 · 2026-02-09T18:24:39Z

/check-enforcer override

* fix few tests part 2 --------- Co-authored-by: Annie Liang <anniemac@Annies-MacBook-Pro.local>

Annie Liang added 3 commits February 6, 2026 10:57

fix test pipeline

cc4689f

fix few tests part 2

54dbd8b

fix some tests

d2f3e17

Copilot AI review requested due to automatic review settings February 6, 2026 20:41

xinlian12 requested review from a team and kirankumarkolli as code owners February 6, 2026 20:41

github-actions bot added the Cosmos label Feb 6, 2026

xinlian12 changed the title ~~Fix few tests part2~~ [NO REVIEW]FixFewTests-part2 Feb 6, 2026

Copilot started reviewing on behalf of xinlian12 February 6, 2026 20:42 View session

Copilot AI reviewed Feb 6, 2026

View reviewed changes

rename

47c7aa1

xinlian12 mentioned this pull request Feb 6, 2026

Leverage x-ms-cosmos-hub-region-processing-only for 404 Read Session Not Available cross-region retry scenarios. #47631

Merged

6 tasks

Annie Liang added 3 commits February 6, 2026 13:09

simplify code

6e960fc

Replace IllegalStateException with CosmosException in bulk error prop…

fab4fd0

…agation Use BridgeInternal.createCosmosException() with status code and substatus code from CosmosBulkItemResponse instead of generic IllegalStateException.

test

6df45b3

xinlian12 changed the title ~~[NO REVIEW]FixFewTests-part2~~ FixFewTests-part2 Feb 8, 2026

Annie Liang added 2 commits February 8, 2026 11:13

fix test pipeline

4e8b099

merge from main and resolve conflicts

e87a2e2

kushagraThapar approved these changes Feb 9, 2026

View reviewed changes

xinlian12 merged commit 97bed4f into Azure:main Feb 9, 2026
95 of 97 checks passed

jeet1995 pushed a commit to jeet1995/azure-sdk-for-java that referenced this pull request Feb 23, 2026

FixFewTests-part2 (Azure#47933)

610ad49

* fix few tests part 2 --------- Co-authored-by: Annie Liang <anniemac@Annies-MacBook-Pro.local>

Conversation

xinlian12 commented Feb 6, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

Copilot AI left a comment

Choose a reason for hiding this comment

Pull request overview

Reviewed changes

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

xinlian12 commented Feb 6, 2026

Uh oh!

azure-pipelines bot commented Feb 6, 2026

Uh oh!

xinlian12 commented Feb 6, 2026

Uh oh!

azure-pipelines bot commented Feb 6, 2026

Uh oh!

xinlian12 commented Feb 7, 2026

Uh oh!

azure-pipelines bot commented Feb 7, 2026

Uh oh!

xinlian12 commented Feb 8, 2026

Uh oh!

azure-pipelines bot commented Feb 8, 2026

Uh oh!

kushagraThapar left a comment

Choose a reason for hiding this comment

Uh oh!

xinlian12 commented Feb 9, 2026

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

xinlian12 commented Feb 6, 2026 •

edited

Loading