Add in DefaultBlobWriteSession test

sydney-munro · sydney-munro · commit 347bb0fd8d12 · 2024-02-13T11:54:13.000-08:00
diff --git a/storage-shared-benchmarking/src/main/java/com/google/cloud/storage/benchmarking/StorageSharedBenchmarkingCli.java b/storage-shared-benchmarking/src/main/java/com/google/cloud/storage/benchmarking/StorageSharedBenchmarkingCli.java
@@ -109,9 +109,12 @@ public void run() {
       case "w1r3":
         runWorkload1();
         break;
-      case "bidi-w1r3":
+      case "bidi":
         runWorkloadBidi();
         break;
+      case "default-nobidi":
+        runWorkloadNoBidi();
+        break;
       default:
         throw new IllegalStateException("Specify a workload to run");
     }
@@ -171,6 +174,20 @@ private void runWorkloadBidi() {
     }
   }
 
+  private void runWorkloadNoBidi() {
+    StorageOptions options = StorageOptions.grpc()
+        .setProjectId(project)
+        .setBlobWriteSessionConfig(BlobWriteSessionConfigs.getDefault())
+        .build();
+    Storage storageClient = options.getService();
+    try {
+      runBidi(storageClient);
+    } catch (Exception e) {
+      System.err.println("Failed to run workload no bidi" + e.getMessage());
+      System.exit(1);
+    }
+  }
+
   private void runW1R3(Storage storageClient) throws ExecutionException, InterruptedException {
     ListeningExecutorService executorService =
         MoreExecutors.listeningDecorator(Executors.newFixedThreadPool(workers));
