From 119f72bde30066c47a389a5a9c88a9a6e37556eb Mon Sep 17 00:00:00 2001
From: Dominik Przybysz <132913826+sfc-gh-dprzybysz@users.noreply.github.com>
Date: Fri, 24 Nov 2023 08:50:12 +0100
Subject: [PATCH] SNOW-565154: Test download stream should fail fast on not
 existing file (#1566)

---
 .../storage/CloudStorageClientLatestIT.java   | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 src/test/java/net/snowflake/client/jdbc/cloud/storage/CloudStorageClientLatestIT.java

diff --git a/src/test/java/net/snowflake/client/jdbc/cloud/storage/CloudStorageClientLatestIT.java b/src/test/java/net/snowflake/client/jdbc/cloud/storage/CloudStorageClientLatestIT.java
new file mode 100644
index 000000000..20a070a02
--- /dev/null
+++ b/src/test/java/net/snowflake/client/jdbc/cloud/storage/CloudStorageClientLatestIT.java
@@ -0,0 +1,46 @@
+package net.snowflake.client.jdbc.cloud.storage;
+
+import static org.hamcrest.CoreMatchers.instanceOf;
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.Assert.fail;
+
+import java.io.InputStream;
+import java.sql.Connection;
+import java.sql.SQLException;
+import java.sql.Statement;
+import java.util.UUID;
+import net.snowflake.client.category.TestCategoryOthers;
+import net.snowflake.client.jdbc.BaseJDBCTest;
+import net.snowflake.client.jdbc.SnowflakeConnection;
+import org.junit.Test;
+import org.junit.experimental.categories.Category;
+
+@Category(TestCategoryOthers.class)
+public class CloudStorageClientLatestIT extends BaseJDBCTest {
+
+  /**
+   * Test for SNOW-565154 - it was waiting for ~5 minutes so the test is waiting much shorter time
+   */
+  @Test(timeout = 30000L)
+  public void testDownloadStreamShouldFailFastOnNotExistingFile() throws Throwable {
+    String stageName =
+        "testDownloadStream_stage_" + UUID.randomUUID().toString().replaceAll("-", "_");
+    try (Connection connection = getConnection();
+        Statement statement = connection.createStatement()) {
+      try {
+        statement.execute("CREATE OR REPLACE TEMP STAGE " + stageName);
+
+        try (InputStream out =
+            connection
+                .unwrap(SnowflakeConnection.class)
+                .downloadStream("@" + stageName, "/fileNotExist.gz", true)) {
+          fail("file should not exist");
+        } catch (Throwable e) {
+          assertThat(e, instanceOf(SQLException.class));
+        }
+      } finally {
+        statement.execute("DROP STAGE IF EXISTS " + stageName);
+      }
+    }
+  }
+}