From b36eba0e23a6581b765fd8ccc4d2919c9d52b11a Mon Sep 17 00:00:00 2001
From: Piotr Fus <piotr.fus@snowflake.com>
Date: Tue, 22 Oct 2024 13:57:46 +0200
Subject: [PATCH] NO-SNOW Fix TestCancelQuery

---
 driver_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/driver_test.go b/driver_test.go
index 68bd156c5..2226edf15 100644
--- a/driver_test.go
+++ b/driver_test.go
@@ -1487,7 +1487,7 @@ func TestCancelQuery(t *testing.T) {
 		ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
 		defer cancel()
 
-		_, err := dbt.conn.QueryContext(ctx, "SELECT DISTINCT 1 FROM TABLE(GENERATOR(TIMELIMIT=> 100))")
+		_, err := dbt.conn.QueryContext(ctx, "CALL SYSTEM$WAIT(10, 'SECONDS')")
 		if err == nil {
 			dbt.Fatal("No timeout error returned")
 		}