fix up tests

sydney-munro · sydney-munro · commit 023a99626a75 · 2024-10-25T12:42:08.000-07:00
diff --git a/google-cloud-storage/src/test/java/com/google/cloud/storage/ITGrpcOpenTelemetryTest.java b/google-cloud-storage/src/test/java/com/google/cloud/storage/ITGrpcOpenTelemetryTest.java
@@ -24,6 +24,7 @@
 import com.google.cloud.storage.it.runner.annotations.Backend;
 import com.google.cloud.storage.it.runner.annotations.Inject;
 import com.google.cloud.storage.it.runner.annotations.SingleBackend;
+import com.google.cloud.storage.it.runner.registry.Generator;
 import com.google.cloud.storage.it.runner.registry.TestBench;
 import io.opentelemetry.api.common.AttributeKey;
 import io.opentelemetry.sdk.OpenTelemetrySdk;
@@ -41,23 +42,17 @@
 @SingleBackend(Backend.TEST_BENCH)
 public class ITGrpcOpenTelemetryTest {
   @Inject public TestBench testBench;
-  private static final ApiClock TIME_SOURCE =
-      new ApiClock() {
-        @Override
-        public long nanoTime() {
-          return 42_000_000_000L;
-        }
-
-        @Override
-        public long millisTime() {
-          return 42_000L;
-        }
-      };
   private StorageOptions options;
-  private SpanExporter exporter = new TestExporter();
+  private SpanExporter exporter;
+  private Storage storage;
+  @Inject public Generator generator;
+  @Inject public BucketInfo testBucket;
+
+
 
   @Before
   public void setUp() {
+    exporter = new TestExporter();
     OpenTelemetrySdk openTelemetrySdk =
         OpenTelemetrySdk.builder()
             .setTracerProvider(
@@ -72,11 +67,12 @@ public void setUp() {
             .setCredentials(NoCredentials.getInstance())
             .setOpenTelemetrySdk(openTelemetrySdk)
             .build();
+    storage = options.getService();
   }
 
+
   @Test
   public void runCreateBucket() {
-    Storage storage = options.getService();
     String bucket = "random-bucket";
     storage.create(BucketInfo.of(bucket));
     TestExporter testExported = (TestExporter) exporter;
@@ -87,22 +83,19 @@ public void runCreateBucket() {
         "com.google.cloud.google-cloud-storage",
         getAttributeValue(spanData, "gcp.client.artifact"));
     Assert.assertEquals("grpc", getAttributeValue(spanData, "rpc.system"));
+
   }
 
   @Test
   public void runCreateBlob() {
-    Storage storage = options.getService();
-    String bucket = "random-bucket";
-    storage.create(BucketInfo.of(bucket));
     byte[] content = "Hello, World!".getBytes(UTF_8);
-    BlobId toCreate = BlobId.of(bucket, "blob");
+    BlobId toCreate = BlobId.of(testBucket.getName(), generator.randomObjectName());
     storage.create(BlobInfo.newBuilder(toCreate).build(), content);
     TestExporter testExported = (TestExporter) exporter;
     List<SpanData> spanData = testExported.getExportedSpans();
-    // (1) Span to create the bucket
-    // (2) Span when calling create
-    // (3) Span when passing call to internalDirectUpload
-    Assert.assertEquals(3, spanData.size());
+    // (1) Span when calling create
+    // (2) Span when passing call to internalDirectUpload
+    Assert.assertEquals(2, spanData.size());
     for(SpanData span : spanData) {
       Assert.assertEquals("Storage", getAttributeValue(span, "gcp.client.service"));
       Assert.assertEquals("googleapis/java-storage",
@@ -112,6 +105,7 @@ public void runCreateBlob() {
           getAttributeValue(span, "gcp.client.artifact"));
       Assert.assertEquals("grpc", getAttributeValue(span, "rpc.system"));
     }
+    Assert.assertEquals(spanData.get(1).getSpanContext(), spanData.get(0).getParentSpanContext());
   }
 
   private String getAttributeValue(SpanData spanData, String key) {
