diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/ResourceResponse.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/ResourceResponse.java index e624e686eb09..270ab645c81f 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/ResourceResponse.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/ResourceResponse.java @@ -274,7 +274,10 @@ public T getResource() { } /** - * Gets the number of index paths (terms) generated by the operation. + * Gets the request charge as request units (RU) consumed by the operation. + *

+ * For more information about the RU and factors that can impact the effective charges please visit + * Request Units in Azure Cosmos DB * * @return the request charge. */ diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/StoredProcedureResponse.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/StoredProcedureResponse.java index 42a7c8ae05b1..90c50a5b0a17 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/StoredProcedureResponse.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/StoredProcedureResponse.java @@ -73,7 +73,10 @@ public String getCurrentResourceQuotaUsage() { } /** - * Gets the number of normalized requests charged. + * Gets the request charge as request units (RU) consumed by the operation. + *

+ * For more information about the RU and factors that can impact the effective charges please visit + * Request Units in Azure Cosmos DB * * @return the request charge. */ diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/query/metrics/ClientSideMetrics.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/query/metrics/ClientSideMetrics.java index 34a80d28d88a..7c86f6157207 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/query/metrics/ClientSideMetrics.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/implementation/query/metrics/ClientSideMetrics.java @@ -64,9 +64,12 @@ public long getRetries() { } /** - * Gets the request charge for this continuation of the query. + * Gets the request charge as request units (RU) consumed by the operation. + *

+ * For more information about the RU and factors that can impact the effective charges please visit + * Request Units in Azure Cosmos DB * - * @return the requestCharge + * @return the request charge. */ public double getRequestCharge() { return requestCharge; diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosAsyncItemResponse.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosAsyncItemResponse.java index 40329cbd5f9b..67e1fa9496ce 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosAsyncItemResponse.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosAsyncItemResponse.java @@ -127,7 +127,10 @@ public String getActivityId() { } /** - * Gets the number of index paths (terms) generated by the operation. + * Gets the request charge as request units (RU) consumed by the operation. + *

+ * For more information about the RU and factors that can impact the effective charges please visit + * Request Units in Azure Cosmos DB * * @return the request charge. */ diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosConflictProperties.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosConflictProperties.java index f347343cb1fe..7c3bf9ee3fc0 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosConflictProperties.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosConflictProperties.java @@ -80,7 +80,7 @@ public CosmosConflictProperties setId(String id) { * * @return the ID associated with the resource. */ - public String getResourceId() { + String getResourceId() { return this.conflict.getResourceId(); } diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosItemResponse.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosItemResponse.java index ef2f1940ab1e..f009ca2e2154 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosItemResponse.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosItemResponse.java @@ -68,7 +68,10 @@ public String getActivityId() { } /** - * Gets the number of index paths (terms) generated by the operation. + * Gets the request charge as request units (RU) consumed by the operation. + *

+ * For more information about the RU and factors that can impact the effective charges please visit + * Request Units in Azure Cosmos DB * * @return the request charge. */ diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosPermissionProperties.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosPermissionProperties.java index 3f00bf86d5c9..06e919b922b0 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosPermissionProperties.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosPermissionProperties.java @@ -125,7 +125,7 @@ public String getId() { * * @return the ID associated with the resource. */ - public String getResourceId() { + String getResourceId() { return this.permission.getResourceId(); } diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosResponse.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosResponse.java index 605b3f665e6a..c2f5617c177a 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosResponse.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosResponse.java @@ -79,7 +79,10 @@ public String getActivityId() { } /** - * Gets the number of index paths (terms) generated by the operation. + * Gets the request charge as request units (RU) consumed by the operation. + *

+ * For more information about the RU and factors that can impact the effective charges please visit + * Request Units in Azure Cosmos DB * * @return the request charge. */ diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosStoredProcedureProperties.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosStoredProcedureProperties.java index 40eeef57f758..af5866bd02bf 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosStoredProcedureProperties.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosStoredProcedureProperties.java @@ -22,7 +22,7 @@ public final class CosmosStoredProcedureProperties { /** * Constructor. */ - public CosmosStoredProcedureProperties() { + CosmosStoredProcedureProperties() { this.storedProcedure = new StoredProcedure(); } @@ -96,7 +96,7 @@ public String getId() { * * @return the ID associated with the resource. */ - public String getResourceId() { + String getResourceId() { return this.storedProcedure.getResourceId(); } diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosTriggerProperties.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosTriggerProperties.java index 6ddd95b53db3..d14a31896561 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosTriggerProperties.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosTriggerProperties.java @@ -18,10 +18,22 @@ public final class CosmosTriggerProperties { /** * Constructor */ - public CosmosTriggerProperties() { + CosmosTriggerProperties() { this.trigger = new Trigger(); } + /** + * Constructor. + * + * @param id the id of the Cosmos trigger. + * @param body the body of the Cosmos trigger. + */ + public CosmosTriggerProperties(String id, String body) { + this.trigger = new Trigger(); + trigger.setId(id); + trigger.setBody(body); + } + /** * Constructor. * @@ -120,7 +132,7 @@ public String getId() { * * @return the ID associated with the resource. */ - public String getResourceId() { + String getResourceId() { return this.trigger.getResourceId(); } diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosUserDefinedFunctionProperties.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosUserDefinedFunctionProperties.java index 1e546686b361..8ec9bf2fdab4 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosUserDefinedFunctionProperties.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosUserDefinedFunctionProperties.java @@ -17,16 +17,28 @@ public final class CosmosUserDefinedFunctionProperties { private UserDefinedFunction userDefinedFunction; /** - * Constructor + * Constructor. + */ + CosmosUserDefinedFunctionProperties() { + this.userDefinedFunction = new UserDefinedFunction(); + } + + /** + * Constructor. + * + * @param id the id of the Cosmos user defined function. + * @param body the body of the Cosmos user defined function. */ - public CosmosUserDefinedFunctionProperties() { + public CosmosUserDefinedFunctionProperties(String id, String body) { this.userDefinedFunction = new UserDefinedFunction(); + userDefinedFunction.setId(id); + userDefinedFunction.setBody(body); } /** * Constructor. * - * @param jsonString the json string that represents the cosmos user defined function properties. + * @param jsonString the JSON string that represents the cosmos user defined function properties. */ CosmosUserDefinedFunctionProperties(String jsonString) { this.userDefinedFunction = new UserDefinedFunction(jsonString); @@ -81,7 +93,7 @@ public CosmosUserDefinedFunctionProperties setId(String id) { * * @return the ID associated with the resource. */ - public String getResourceId() { + String getResourceId() { return this.userDefinedFunction.getResourceId(); } diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosUserProperties.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosUserProperties.java index 88521bb68d3e..61e127824c3f 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosUserProperties.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/CosmosUserProperties.java @@ -16,6 +16,7 @@ public final class CosmosUserProperties { private User user; + /** * Initialize a user object. */ @@ -80,7 +81,7 @@ public String getId() { * * @return the ID associated with the resource. */ - public String getResourceId() { + String getResourceId() { return this.user.getResourceId(); } diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/FeedResponse.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/FeedResponse.java index 28c9828c2458..f5dd75360342 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/FeedResponse.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/FeedResponse.java @@ -244,7 +244,10 @@ public String getCurrentResourceQuotaUsage() { } /** - * Gets the number of index paths (terms) generated by the operation. + * Gets the request charge as request units (RU) consumed by the operation. + *

+ * For more information about the RU and factors that can impact the effective charges please visit + * Request Units in Azure Cosmos DB * * @return the request charge. */ diff --git a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/ThroughputProperties.java b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/ThroughputProperties.java index b1e475c3ca6e..8d1a0a0a9337 100644 --- a/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/ThroughputProperties.java +++ b/sdk/cosmos/azure-cosmos/src/main/java/com/azure/cosmos/models/ThroughputProperties.java @@ -29,7 +29,7 @@ public static ThroughputProperties createManualThroughput(int throughput) { } /** - * Create auto scale throughput properties. + * Create auto-scale throughput properties. * * @param autoScaleMaxThroughput the max auto scale throughput * @param autoUpgradethroughputIncrementPercentage the auto upgrade max throughput increment percentage @@ -63,7 +63,7 @@ public Integer getManualThroughput() { } /** - * Gets offer autoscale properties. + * Gets offer auto-scale properties. * * @return the offer autoscale properties */ @@ -72,9 +72,9 @@ OfferAutoscaleSettings getOfferAutoscaleProperties() { } /** - * Gets max autoscale throughput. + * Gets max auto-scale throughput. * - * @return the max autoscale throughput + * @return the max auto-scale throughput */ public int getAutoscaleMaxThroughput() { return this.offer.getAutoscaleMaxThroughput(); @@ -122,7 +122,7 @@ public ThroughputProperties setId(String id) { * * @return the ID associated with the resource. */ - public String getResourceId() { + String getResourceId() { return this.offer.getResourceId(); } diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosSyncStoredProcTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosSyncStoredProcTest.java index 9341c50ef8ed..258201249e66 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosSyncStoredProcTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosSyncStoredProcTest.java @@ -127,9 +127,11 @@ public void replaceStoredProcedure() throws Exception { } private CosmosStoredProcedureProperties getCosmosStoredProcedureProperties() { - CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties(); - storedProcedureDef.setId(UUID.randomUUID().toString()); - storedProcedureDef.setBody("function() {var x = 10;}"); + CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); + return storedProcedureDef; } @@ -146,9 +148,10 @@ public void deleteStoredProcedure() throws Exception { } @Test(groups = {"simple"}, timeOut = TIMEOUT) public void executeStoredProcedure() throws Exception { - CosmosStoredProcedureProperties sproc = new CosmosStoredProcedureProperties() - .setId(UUID.randomUUID().toString()); - sproc.setBody("function() {var x = 10;}"); + CosmosStoredProcedureProperties sproc = new CosmosStoredProcedureProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); CosmosStoredProcedureResponse response = container.getScripts().createStoredProcedure(sproc); CosmosStoredProcedureRequestOptions options = new CosmosStoredProcedureRequestOptions(); diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosSyncUDFTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosSyncUDFTest.java index 70f439941d6c..c24c867547a1 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosSyncUDFTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosSyncUDFTest.java @@ -94,9 +94,11 @@ public void deleteUDF() throws Exception { } private CosmosUserDefinedFunctionProperties getCosmosUserDefinedFunctionProperties() { - CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties(); - udf.setId(UUID.randomUUID().toString()); - udf.setBody("function() {var x = 10;}"); + CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); + return udf; } diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosTriggerTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosTriggerTest.java index 5d6a877d1400..891c54ab4d5f 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosTriggerTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/CosmosTriggerTest.java @@ -101,11 +101,13 @@ public void readAllTriggers() throws Exception { } private CosmosTriggerProperties getCosmosTriggerProperties() { - CosmosTriggerProperties trigger = new CosmosTriggerProperties(); - trigger.setId(UUID.randomUUID().toString()); - trigger.setBody("function() {var x = 10;}"); + CosmosTriggerProperties trigger = new CosmosTriggerProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); trigger.setTriggerOperation(TriggerOperation.CREATE); trigger.setTriggerType(TriggerType.PRE); + return trigger; } diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/implementation/FeedResponseListValidator.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/implementation/FeedResponseListValidator.java index bf881756dccb..b3854b02c96d 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/implementation/FeedResponseListValidator.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/implementation/FeedResponseListValidator.java @@ -143,6 +143,23 @@ public void validate(List> feedList) { return this; } + public Builder exactlyContainsIdsInAnyOrder(List expectedIds) { + validators.add(new FeedResponseListValidator() { + @Override + public void validate(List> feedList) { + List actualIds = feedList + .stream() + .flatMap(f -> f.getResults().stream()) + .map(r -> getResource(r).getId()) + .collect(Collectors.toList()); + assertThat(actualIds) + .describedAs("IDs of results") + .containsOnlyElementsOf(expectedIds); + } + }); + return this; + } + public Builder numberOfPages(int expectedNumberOfPages) { validators.add(new FeedResponseListValidator() { @Override diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/PermissionQueryTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/PermissionQueryTest.java index 8a667c6469b5..96d1359cd635 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/PermissionQueryTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/PermissionQueryTest.java @@ -58,7 +58,7 @@ public void queryWithFilter() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(expectedDocs.size()) - .exactlyContainsInAnyOrder(expectedDocs.stream().map(d -> d.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder(expectedDocs.stream().map(CosmosPermissionProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .pageSatisfy(0, new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) @@ -94,9 +94,9 @@ public void queryAll() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator .Builder() - .exactlyContainsInAnyOrder(createdPermissions + .exactlyContainsIdsInAnyOrder(createdPermissions .stream() - .map(d -> d.getResourceId()) + .map(CosmosPermissionProperties::getId) .collect(Collectors.toList())) .numberOfPages(expectedPageSize) .allPagesSatisfy(new FeedResponseValidator.Builder() diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedPermissionsTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedPermissionsTest.java index cab95edfdbe9..2e75428234be 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedPermissionsTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedPermissionsTest.java @@ -50,8 +50,8 @@ public void readPermissions() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(createdPermissions.size()) .numberOfPages(expectedPageSize) - .exactlyContainsInAnyOrder(createdPermissions.stream().map( - p -> p.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder(createdPermissions.stream().map( + CosmosPermissionProperties::getId).collect(Collectors.toList())) .allPagesSatisfy(new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) .build(); diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedStoredProceduresTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedStoredProceduresTest.java index 7159d7f15bcb..9d847d65a2a6 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedStoredProceduresTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedStoredProceduresTest.java @@ -45,8 +45,8 @@ public void readStoredProcedures() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(createdStoredProcedures.size()) - .exactlyContainsInAnyOrder( - createdStoredProcedures.stream().map(d -> d.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder( + createdStoredProcedures.stream().map(CosmosStoredProcedureProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .allPagesSatisfy(new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) @@ -73,9 +73,11 @@ public void afterClass() { } public CosmosStoredProcedureProperties createStoredProcedures(CosmosAsyncContainer cosmosContainer) { - CosmosStoredProcedureProperties sproc = new CosmosStoredProcedureProperties(); - sproc.setId(UUID.randomUUID().toString()); - sproc.setBody("function() {var x = 10;}"); + CosmosStoredProcedureProperties sproc = new CosmosStoredProcedureProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); + return cosmosContainer.getScripts().createStoredProcedure(sproc, new CosmosStoredProcedureRequestOptions()) .block().getProperties(); } diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedTriggersTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedTriggersTest.java index 7ffa47304200..86c06fcc4686 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedTriggersTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedTriggersTest.java @@ -43,8 +43,8 @@ public void readTriggers() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(createdTriggers.size()) - .exactlyContainsInAnyOrder( - createdTriggers.stream().map(d -> d.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder( + createdTriggers.stream().map(CosmosTriggerProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .allPagesSatisfy(new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) @@ -71,11 +71,13 @@ public void afterClass() { } public CosmosTriggerProperties createTriggers(CosmosAsyncContainer cosmosContainer) { - CosmosTriggerProperties trigger = new CosmosTriggerProperties(); - trigger.setId(UUID.randomUUID().toString()); - trigger.setBody("function() {var x = 10;}"); + CosmosTriggerProperties trigger = new CosmosTriggerProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); trigger.setTriggerOperation(TriggerOperation.CREATE); trigger.setTriggerType(TriggerType.PRE); + return cosmosContainer.getScripts().createTrigger(trigger).block().getProperties(); } } diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedUdfsTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedUdfsTest.java index 3aa6605fedea..875d1174ea86 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedUdfsTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedUdfsTest.java @@ -47,8 +47,8 @@ public void readUserDefinedFunctions() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(createdUserDefinedFunctions.size()) - .exactlyContainsInAnyOrder( - createdUserDefinedFunctions.stream().map(d -> d.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder( + createdUserDefinedFunctions.stream().map(CosmosUserDefinedFunctionProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .allPagesSatisfy(new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) @@ -75,9 +75,11 @@ public void afterClass() { } public CosmosUserDefinedFunctionProperties createUserDefinedFunctions(CosmosAsyncContainer cosmosContainer) { - CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties(); - udf.setId(UUID.randomUUID().toString()); - udf.setBody("function() {var x = 10;}"); + CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); + return cosmosContainer.getScripts().createUserDefinedFunction(udf).block() .getProperties(); } diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedUsersTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedUsersTest.java index 8dd72d0ef20e..0102e111e5da 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedUsersTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/ReadFeedUsersTest.java @@ -45,7 +45,7 @@ public void readUsers() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(createdUsers.size()) - .exactlyContainsInAnyOrder(createdUsers.stream().map(d -> d.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder(createdUsers.stream().map(CosmosUserProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .pageSatisfy(0, new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureCrudTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureCrudTest.java index f7c4063f5916..21e52120dbf8 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureCrudTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureCrudTest.java @@ -36,9 +36,10 @@ public StoredProcedureCrudTest(CosmosClientBuilder clientBuilder) { @Test(groups = { "simple" }, timeOut = TIMEOUT) public void createStoredProcedure() throws Exception { - CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties(); - storedProcedureDef.setId(UUID.randomUUID().toString()); - storedProcedureDef.setBody("function() {var x = 10;}"); + CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); Mono createObservable = container.getScripts().createStoredProcedure(storedProcedureDef, new CosmosStoredProcedureRequestOptions()); @@ -54,9 +55,11 @@ public void createStoredProcedure() throws Exception { @Test(groups = { "simple" }, timeOut = TIMEOUT) public void readStoredProcedure() throws Exception { - CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties(); - storedProcedureDef.setId(UUID.randomUUID().toString()); - storedProcedureDef.setBody("function() {var x = 10;}"); + CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); + CosmosStoredProcedureResponse storedProcedureResponse = container.getScripts().createStoredProcedure(storedProcedureDef, new CosmosStoredProcedureRequestOptions()).block(); CosmosAsyncStoredProcedure storedProcedure = container.getScripts().getStoredProcedure(storedProcedureResponse.getProperties().getId()); @@ -75,9 +78,10 @@ public void readStoredProcedure() throws Exception { @Test(groups = { "simple" }, timeOut = TIMEOUT) public void deleteStoredProcedure() throws Exception { - CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties(); - storedProcedureDef.setId(UUID.randomUUID().toString()); - storedProcedureDef.setBody("function() {var x = 10;}"); + CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); CosmosStoredProcedureResponse storedProcedureResponse = this.container.getScripts().createStoredProcedure(storedProcedureDef, new CosmosStoredProcedureRequestOptions()).block(); CosmosAsyncStoredProcedure storedProcedure = diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureQueryTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureQueryTest.java index 59cb84463469..884c608a92d1 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureQueryTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureQueryTest.java @@ -55,7 +55,7 @@ public void queryWithFilter() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(expectedDocs.size()) - .exactlyContainsInAnyOrder(expectedDocs.stream().map(d -> d.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder(expectedDocs.stream().map(CosmosStoredProcedureProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .pageSatisfy(0, new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) @@ -96,7 +96,7 @@ public void queryAll() throws Exception { int expectedPageSize = (expectedDocs.size() + maxItemCount - 1) / maxItemCount; FeedResponseListValidator validator = new FeedResponseListValidator.Builder() - .exactlyContainsInAnyOrder(expectedDocs.stream().map(d -> d.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder(expectedDocs.stream().map(CosmosStoredProcedureProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .allPagesSatisfy(new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) @@ -142,9 +142,11 @@ public void afterClass() { } private static CosmosStoredProcedureProperties getStoredProcedureDef() { - CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties(); - storedProcedureDef.setId(UUID.randomUUID().toString()); - storedProcedureDef.setBody("function() {var x = 10;}"); + CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); + return storedProcedureDef; } } diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureUpsertReplaceTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureUpsertReplaceTest.java index fbe0b7f5bd12..fc32d95fb501 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureUpsertReplaceTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/StoredProcedureUpsertReplaceTest.java @@ -37,9 +37,11 @@ public StoredProcedureUpsertReplaceTest(CosmosClientBuilder clientBuilder) { public void replaceStoredProcedure() throws Exception { // create a stored procedure - CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties(); - storedProcedureDef.setId(UUID.randomUUID().toString()); - storedProcedureDef.setBody("function() {var x = 10;}"); + CosmosStoredProcedureProperties storedProcedureDef = new CosmosStoredProcedureProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); + CosmosStoredProcedureProperties readBackSp = createdCollection.getScripts() .createStoredProcedure(storedProcedureDef, new CosmosStoredProcedureRequestOptions()).block() .getProperties(); diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerCrudTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerCrudTest.java index f25620843159..a66d745b176e 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerCrudTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerCrudTest.java @@ -34,9 +34,10 @@ public TriggerCrudTest(CosmosClientBuilder clientBuilder) { public void createTrigger() throws Exception { // create a trigger - CosmosTriggerProperties trigger = new CosmosTriggerProperties(); - trigger.setId(UUID.randomUUID().toString()); - trigger.setBody("function() {var x = 10;}"); + CosmosTriggerProperties trigger = new CosmosTriggerProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); trigger.setTriggerOperation(TriggerOperation.CREATE); trigger.setTriggerType(TriggerType.PRE); @@ -55,9 +56,10 @@ public void createTrigger() throws Exception { @Test(groups = { "simple" }, timeOut = TIMEOUT) public void readTrigger() throws Exception { // create a trigger - CosmosTriggerProperties trigger = new CosmosTriggerProperties(); - trigger.setId(UUID.randomUUID().toString()); - trigger.setBody("function() {var x = 10;}"); + CosmosTriggerProperties trigger = new CosmosTriggerProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); trigger.setTriggerOperation(TriggerOperation.CREATE); trigger.setTriggerType(TriggerType.PRE); CosmosAsyncTrigger readBackTrigger = createdCollection.getScripts().createTrigger(trigger).block().getTrigger(); @@ -79,9 +81,10 @@ public void readTrigger() throws Exception { @Test(groups = { "simple" }, timeOut = TIMEOUT) public void deleteTrigger() throws Exception { // create a trigger - CosmosTriggerProperties trigger = new CosmosTriggerProperties(); - trigger.setId(UUID.randomUUID().toString()); - trigger.setBody("function() {var x = 10;}"); + CosmosTriggerProperties trigger = new CosmosTriggerProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); trigger.setTriggerOperation(TriggerOperation.CREATE); trigger.setTriggerType(TriggerType.PRE); CosmosAsyncTrigger readBackTrigger = createdCollection.getScripts().createTrigger(trigger).block().getTrigger(); diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerQueryTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerQueryTest.java index 38f52d03c70a..9ff5a10038a0 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerQueryTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerQueryTest.java @@ -58,7 +58,7 @@ public void queryWithFilter() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(expectedDocs.size()) - .exactlyContainsInAnyOrder(expectedDocs.stream().map(doc -> doc.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder(expectedDocs.stream().map(CosmosTriggerProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .pageSatisfy(0, new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) @@ -93,7 +93,8 @@ public void queryAll() throws Exception { CosmosPagedFlux queryObservable = createdCollection.getScripts().queryTriggers(query, options); - createdTriggers.forEach(cosmosTriggerSettings -> logger.info("Created trigger in method: {}", cosmosTriggerSettings.getResourceId())); + createdTriggers.forEach(cosmosTriggerSettings -> logger.info("Created trigger in method: {}", + cosmosTriggerSettings.getId())); List expectedDocs = createdTriggers; @@ -101,9 +102,9 @@ public void queryAll() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator .Builder() - .exactlyContainsInAnyOrder(expectedDocs + .exactlyContainsIdsInAnyOrder(expectedDocs .stream() - .map(doc -> doc.getResourceId()) + .map(CosmosTriggerProperties::getId) .collect(Collectors.toList())) .numberOfPages(expectedPageSize) .allPagesSatisfy(new FeedResponseValidator.Builder() @@ -152,11 +153,13 @@ public void afterClass() { } private static CosmosTriggerProperties getTriggerDef() { - CosmosTriggerProperties trigger = new CosmosTriggerProperties(); - trigger.setId(UUID.randomUUID().toString()); - trigger.setBody("function() {var x = 10;}"); + CosmosTriggerProperties trigger = new CosmosTriggerProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); trigger.setTriggerOperation(TriggerOperation.CREATE); trigger.setTriggerType(TriggerType.PRE); + return trigger; } } diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerUpsertReplaceTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerUpsertReplaceTest.java index 4f1a03e6c950..6ceb511f346f 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerUpsertReplaceTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/TriggerUpsertReplaceTest.java @@ -34,9 +34,10 @@ public TriggerUpsertReplaceTest(CosmosClientBuilder clientBuilder) { public void replaceTrigger() throws Exception { // create a trigger - CosmosTriggerProperties trigger = new CosmosTriggerProperties(); - trigger.setId(UUID.randomUUID().toString()); - trigger.setBody("function() {var x = 10;}"); + CosmosTriggerProperties trigger = new CosmosTriggerProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); trigger.setTriggerOperation(TriggerOperation.CREATE); trigger.setTriggerType(TriggerType.PRE); CosmosTriggerProperties readBackTrigger = createdCollection.getScripts().createTrigger(trigger).block().getProperties(); diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionCrudTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionCrudTest.java index 9e88f0f7fbb1..14feec11ef38 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionCrudTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionCrudTest.java @@ -30,9 +30,10 @@ public UserDefinedFunctionCrudTest(CosmosClientBuilder clientBuilder) { @Test(groups = { "simple" }, timeOut = TIMEOUT) public void createUserDefinedFunction() throws Exception { // create udf - CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties(); - udf.setId(UUID.randomUUID().toString()); - udf.setBody("function() {var x = 10;}"); + CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); Mono createObservable = createdCollection.getScripts().createUserDefinedFunction(udf); @@ -48,13 +49,14 @@ public void createUserDefinedFunction() throws Exception { @Test(groups = { "simple" }, timeOut = TIMEOUT) public void readUserDefinedFunction() throws Exception { // create a udf - CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties(); - udf.setId(UUID.randomUUID().toString()); - udf.setBody("function() {var x = 10;}"); + CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); createdCollection.getScripts().createUserDefinedFunction(udf).block(); CosmosAsyncUserDefinedFunction readBackUdf = createdCollection.getScripts().getUserDefinedFunction(udf.getId()); - // read udf + // read udf waitIfNeededForReplicasToCatchUp(getClientBuilder()); Mono readObservable = readBackUdf.read(); @@ -70,9 +72,10 @@ public void readUserDefinedFunction() throws Exception { @Test(groups = { "simple" }, timeOut = TIMEOUT) public void deleteUserDefinedFunction() throws Exception { // create a udf - CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties(); - udf.setId(UUID.randomUUID().toString()); - udf.setBody("function() {var x = 10;}"); + CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); createdCollection.getScripts().createUserDefinedFunction(udf).block(); CosmosAsyncUserDefinedFunction readBackUdf = createdCollection.getScripts().getUserDefinedFunction(udf.getId()); diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionQueryTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionQueryTest.java index ead18c7738b0..eadfa717974f 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionQueryTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionQueryTest.java @@ -60,7 +60,7 @@ public void queryWithFilter() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(expectedDocs.size()) - .exactlyContainsInAnyOrder(expectedDocs.stream().map(d -> d.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder(expectedDocs.stream().map(CosmosUserDefinedFunctionProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .pageSatisfy(0, new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) @@ -102,9 +102,9 @@ public void queryAll() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator .Builder() - .exactlyContainsInAnyOrder(expectedDocs + .exactlyContainsIdsInAnyOrder(expectedDocs .stream() - .map(d -> d.getResourceId()) + .map(CosmosUserDefinedFunctionProperties::getId) .collect(Collectors.toList())) .numberOfPages(expectedPageSize) .allPagesSatisfy(new FeedResponseValidator.Builder() @@ -152,9 +152,11 @@ public void afterClass() { } private static CosmosUserDefinedFunctionProperties getUserDefinedFunctionDef() { - CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties(); - udf.setId(UUID.randomUUID().toString()); - udf.setBody("function() {var x = 10;}"); + CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); + return udf; } } diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionUpsertReplaceTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionUpsertReplaceTest.java index 5d562c0ff7a5..5aea657b6ea7 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionUpsertReplaceTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserDefinedFunctionUpsertReplaceTest.java @@ -31,9 +31,10 @@ public UserDefinedFunctionUpsertReplaceTest(CosmosClientBuilder clientBuilder) { public void replaceUserDefinedFunction() throws Exception { // create a udf - CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties(); - udf.setId(UUID.randomUUID().toString()); - udf.setBody("function() {var x = 10;}"); + CosmosUserDefinedFunctionProperties udf = new CosmosUserDefinedFunctionProperties( + UUID.randomUUID().toString(), + "function() {var x = 10;}" + ); CosmosUserDefinedFunctionProperties readBackUdf = null; diff --git a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserQueryTest.java b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserQueryTest.java index 9d74cd971df2..b7b60c34c97b 100644 --- a/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserQueryTest.java +++ b/sdk/cosmos/azure-cosmos/src/test/java/com/azure/cosmos/rx/UserQueryTest.java @@ -59,7 +59,7 @@ public void queryUsersWithFilter() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(expectedUsers.size()) - .exactlyContainsInAnyOrder(expectedUsers.stream().map(d -> d.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder(expectedUsers.stream().map(CosmosUserProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .pageSatisfy(0, new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build()) @@ -86,7 +86,7 @@ public void queryAllUsers() throws Exception { FeedResponseListValidator validator = new FeedResponseListValidator.Builder() .totalSize(expectedUsers.size()) - .exactlyContainsInAnyOrder(expectedUsers.stream().map(d -> d.getResourceId()).collect(Collectors.toList())) + .exactlyContainsIdsInAnyOrder(expectedUsers.stream().map(CosmosUserProperties::getId).collect(Collectors.toList())) .numberOfPages(expectedPageSize) .pageSatisfy(0, new FeedResponseValidator.Builder() .requestChargeGreaterThanOrEqualTo(1.0).build())