Update CreateItemBenchmarkOperation test.

Author: aavasthy

Microsoft.Azure.Cosmos.Samples/Tools/Benchmark/v3/CreateItemBenchmarkOperation.cs

@@ -5,47 +5,56 @@
 namespace CosmosBenchmark
 {
     using System;
-    using System.Net;
+    using System.Collections.Generic;
+    using System.IO;
     using System.Threading.Tasks;
     using Microsoft.Azure.Cosmos;
 
     internal class CreateItemBenchmarkOperation : IBenchmarkOperation
     {
         private readonly Container container;
-        private readonly string pk = "pk_benchmark";
-
-        public CreateItemBenchmarkOperation(CosmosClient client, string db, string containerName)
+        private readonly string partitionKeyPath;
+        private readonly Dictionary<string, object> sampleJObject;
+        private readonly string databaseName;
+        private readonly string containerName;
+
+        public CreateItemBenchmarkOperation(
+            CosmosClient cosmosClient,
+            string dbName,
+            string containerName,
+            string partitionKeyPath,
+            string sampleJson)
         {
-            this.container = client.GetContainer(db, containerName);
+            this.databaseName = dbName;
+            this.containerName = containerName;
+            this.container = cosmosClient.GetContainer(this.databaseName, this.containerName);
+            this.partitionKeyPath = partitionKeyPath.Replace("/", "");
+            this.sampleJObject = JsonHelper.Deserialize<Dictionary<string, object>>(sampleJson);
         }
 
-        public async Task PrepareAsync() { }
+        public BenchmarkOperationType OperationType => BenchmarkOperationType.Insert;
 
-        public async Task<OperationResult> ExecuteOnceAsync()
+        public Task PrepareAsync()
         {
-            CosmosIntegrationTestObject item = new CosmosIntegrationTestObject
-            {
-                Id = Guid.NewGuid().ToString(),
-                Pk = this.pk,
-                Other = "Create Test"
-            };
-
-            ItemResponse<CosmosIntegrationTestObject> response = await this.container.CreateItemAsync(item, new PartitionKey(item.Pk));
-            if (response.StatusCode != HttpStatusCode.Created)
-            {
-                throw new Exception($"Failed to create item: {item.Id}");
-            }
-
-            return new OperationResult { OperationType = BenchmarkOperationType.Insert };
+            this.sampleJObject["id"] = Guid.NewGuid().ToString();
+            this.sampleJObject[this.partitionKeyPath] = Guid.NewGuid().ToString();
+            return Task.CompletedTask;
         }
 
-        public BenchmarkOperationType OperationType => BenchmarkOperationType.Insert;
-
-        public class CosmosIntegrationTestObject
+        public async Task<OperationResult> ExecuteOnceAsync()
         {
-            public string Id { get; set; }
-            public string Pk { get; set; }
-            public string Other { get; set; }
+            PartitionKey partitionKey = new PartitionKey(this.sampleJObject[this.partitionKeyPath].ToString());
+            ItemResponse<Dictionary<string, object>> response = await this.container.CreateItemAsync(this.sampleJObject, partitionKey: partitionKey);
+
+            return new OperationResult
+            {
+                DatabseName = this.databaseName,
+                ContainerName = this.containerName,
+                OperationType = this.OperationType,
+                RuCharges = response.RequestCharge,
+                CosmosDiagnostics = response.Diagnostics,
+                LazyDiagnostics = () => response.Diagnostics.ToString(),
+            };
         }
     }
-}
+}