chore: add tests for object contexts samples

Author: ShreyasSinha

samples/snippets/src/test/java/com/example/storage/ITObjectSnippets.java

@@ -43,7 +43,9 @@
 import com.example.storage.object.GenerateEncryptionKey;
 import com.example.storage.object.GenerateV4GetObjectSignedUrl;
 import com.example.storage.object.GenerateV4PutObjectSignedUrl;
+import com.example.storage.object.GetObjectContexts;
 import com.example.storage.object.GetObjectMetadata;
+import com.example.storage.object.ListObjectContexts;
 import com.example.storage.object.ListObjects;
 import com.example.storage.object.ListObjectsWithOldVersions;
 import com.example.storage.object.ListObjectsWithPrefix;
@@ -52,6 +54,7 @@
 import com.example.storage.object.MakeObjectPublic;
 import com.example.storage.object.RestoreSoftDeletedObject;
 import com.example.storage.object.RotateObjectEncryptionKey;
+import com.example.storage.object.SetObjectContexts;
 import com.example.storage.object.SetObjectMetadata;
 import com.example.storage.object.SetObjectRetentionPolicy;
 import com.example.storage.object.StreamObjectDownload;
@@ -65,6 +68,8 @@
 import com.google.cloud.storage.Blob;
 import com.google.cloud.storage.BlobId;
 import com.google.cloud.storage.BlobInfo;
+import com.google.cloud.storage.BlobInfo.ObjectContexts;
+import com.google.cloud.storage.BlobInfo.ObjectCustomContextPayload;
 import com.google.cloud.storage.Bucket;
 import com.google.cloud.storage.BucketInfo;
 import com.google.cloud.storage.BucketInfo.IamConfiguration;
@@ -79,6 +84,7 @@
 import com.google.cloud.storage.it.runner.annotations.Inject;
 import com.google.cloud.storage.it.runner.registry.KmsFixture;
 import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.Maps;
 import com.google.common.io.BaseEncoding;
 import java.io.File;
 import java.io.IOException;
@@ -89,6 +95,7 @@
 import java.nio.file.Path;
 import java.time.Duration;
 import java.util.Date;
+import java.util.Map;
 import java.util.Random;
 import javax.net.ssl.HttpsURLConnection;
 import org.junit.Assert;
@@ -633,4 +640,73 @@ public void testRestoreSoftDeletedObject() throws Exception {
       assertNotNull(storage.get(BlobId.of(bucketName, blob)));
     }
   }
+
+  @Test
+  public void testSetObjectContexts() throws Exception {
+    String blobName = generator.randomObjectName();
+    String key = "test-key-get";
+    String value = "test-value-get";
+
+    Blob initialBlob = storage.create(info(blobName), CONTENT, BlobTargetOption.doesNotExist());
+
+    SetObjectContexts.setObjectContexts(
+        GOOGLE_CLOUD_PROJECT, bucket.getName(), blobName, key, value);
+    String setOutput = stdOut.getCapturedOutputAsUtf8String();
+    assertThat(setOutput).contains("Updated custom contexts for object " + blobName);
+
+    Blob updatedBlob = storage.get(bucket.getName(), blobName);
+    assertThat(updatedBlob.getContexts().getCustom().get(key).getValue()).isEqualTo(value);
+  }
+
+  @Test
+  public void testGetObjectContexts() throws Exception {
+    String blobName = generator.randomObjectName();
+    String key = "test-key-get";
+    String value = "test-value-get";
+
+    storage.create(info(blobName), CONTENT, BlobTargetOption.doesNotExist());
+
+    ObjectCustomContextPayload payload =
+        ObjectCustomContextPayload.newBuilder().setValue(value).build();
+    Map<String, ObjectCustomContextPayload> custom = Maps.newHashMap();
+    custom.put(key, payload);
+    ObjectContexts contexts = ObjectContexts.newBuilder().setCustom(custom).build();
+    BlobInfo pendingUpdate =
+        storage.get(bucket.getName(), blobName).toBuilder().setContexts(contexts).build();
+    storage.update(pendingUpdate);
+
+    GetObjectContexts.getObjectContexts(GOOGLE_CLOUD_PROJECT, bucket.getName(), blobName);
+
+    String getOutput = stdOut.getCapturedOutputAsUtf8String();
+
+    assertThat(getOutput).contains("Custom Contexts:");
+    assertThat(getOutput).contains(key + "=ObjectCustomContextPayload{");
+    assertThat(getOutput).contains("value=" + value);
+  }
+
+  @Test
+  public void testListObjectContexts() throws Exception {
+    String blobName = generator.randomObjectName();
+    String key = "test-key-list";
+    String value = "test-value-list";
+
+    storage.create(info(blobName), CONTENT, BlobTargetOption.doesNotExist());
+
+    ObjectCustomContextPayload payload =
+        ObjectCustomContextPayload.newBuilder().setValue(value).build();
+    Map<String, ObjectCustomContextPayload> custom = Maps.newHashMap();
+    custom.put(key, payload);
+    ObjectContexts contexts = ObjectContexts.newBuilder().setCustom(custom).build();
+    BlobInfo pendingUpdate =
+        storage.get(bucket.getName(), blobName).toBuilder().setContexts(contexts).build();
+    storage.update(pendingUpdate);
+
+    ListObjectContexts.listObjectContexts(GOOGLE_CLOUD_PROJECT, bucket.getName(), key);
+    String listOutput = stdOut.getCapturedOutputAsUtf8String();
+
+    assertThat(listOutput).contains("gs://" + bucket.getName() + "/" + blobName);
+
+    assertThat(listOutput)
+        .contains("Listing objects for bucket: " + bucket.getName() + "with context key: " + key);
+  }
 }