NO-SNOW Fix TestCancelQuery (#1226)

sfc-gh-pfus · web-flow · commit 58c35d402f8e · 2024-10-24T12:59:24.000+02:00
diff --git 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")
 		}

Original file line number	Diff line number	Diff line change
`@@ -1487,7 +1487,7 @@ func TestCancelQuery(t *testing.T) {`
`1487`	`1487`	`ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)`
`1488`	`1488`	`defer cancel()`
`1489`	`1489`
`1490`		`- _, err := dbt.conn.QueryContext(ctx, "SELECT DISTINCT 1 FROM TABLE(GENERATOR(TIMELIMIT=> 100))")`
	`1490`	`+ _, err := dbt.conn.QueryContext(ctx, "CALL SYSTEM$WAIT(10, 'SECONDS')")`
`1491`	`1491`	`if err == nil {`
`1492`	`1492`	`dbt.Fatal("No timeout error returned")`
`1493`	`1493`	`}`