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())