Add test for missing bucket or key

trinodb · Sep 18, 2024 · df4167d · df4167d
1 parent 3c9b92e
commit df4167d
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/trino-aws-proxy/src/test/java/io/trino/aws/proxy/server/AbstractTestProxiedRequests.java b/trino-aws-proxy/src/test/java/io/trino/aws/proxy/server/AbstractTestProxiedRequests.java
@@ -44,6 +44,7 @@
 import java.time.Duration;
 import java.util.Comparator;
 import java.util.List;
+import java.util.UUID;
 import java.util.concurrent.CopyOnWriteArrayList;
 import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
@@ -54,6 +55,7 @@
 import static com.google.common.collect.ImmutableList.toImmutableList;
 import static com.google.common.util.concurrent.MoreExecutors.shutdownAndAwaitTermination;
 import static io.trino.aws.proxy.server.testing.TestingUtil.TEST_FILE;
+import static io.trino.aws.proxy.server.testing.TestingUtil.assertFileNotInS3;
 import static io.trino.aws.proxy.server.testing.TestingUtil.getFileFromStorage;
 import static io.trino.aws.proxy.server.testing.TestingUtil.headObjectInStorage;
 import static io.trino.aws.proxy.server.testing.TestingUtil.listFilesInS3Bucket;
@@ -249,6 +251,17 @@ public void testPathsNeedingEscaping()
  internalClient.deleteBucket(r -> r.bucket(bucket));
  }
 
+ @Test
+ public void testKeyOrBucketDoesNotExist()
+ {
+ assertFileNotInS3(internalClient, UUID.randomUUID().toString(), UUID.randomUUID().toString());
+
+ String newBucketName = "new-bucket";
+ remoteClient.createBucket(r -> r.bucket(newBucketName));
+
+ assertFileNotInS3(internalClient, newBucketName, UUID.randomUUID().toString());
+ }
+
  private static String buildLine(int partNumber)
  {
  // min multi-part is 5MB