From d6a4513dc629c748b6f623c84ed6029218fc37a7 Mon Sep 17 00:00:00 2001
From: Purujit Saha <purujit.saha@snowflake.com>
Date: Thu, 31 Oct 2024 20:13:34 +0000
Subject: [PATCH] Deflake a test

---
 .../snowflake/ingest/streaming/internal/InternalStageTest.java  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/test/java/net/snowflake/ingest/streaming/internal/InternalStageTest.java b/src/test/java/net/snowflake/ingest/streaming/internal/InternalStageTest.java
index 150ec566a..3baac0058 100644
--- a/src/test/java/net/snowflake/ingest/streaming/internal/InternalStageTest.java
+++ b/src/test/java/net/snowflake/ingest/streaming/internal/InternalStageTest.java
@@ -488,7 +488,7 @@ public void testRefreshSnowflakeMetadataSynchronized() throws Exception {
           }
         });
 
-    workers.awaitTermination(150, TimeUnit.MILLISECONDS);
+    Assert.assertTrue(workers.awaitTermination(1, TimeUnit.SECONDS));
 
     Mockito.verify(mockClient).execute(Mockito.any());
   }