Adding long test variable

firebolt-db · Sep 10, 2024 · 074ca29 · 074ca29
1 parent 2d3e17b
commit 074ca29
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 1 deletion.
diff --git a/.github/workflows/integration-tests-v1.yml b/.github/workflows/integration-tests-v1.yml
@@ -48,5 +48,6 @@ jobs:
           ACCOUNT_NAME: "firebolt"
           SERVICE_ACCOUNT_CLIENT_ID: ${{ secrets.SERVICE_ID_STG }}
           SERVICE_ACCOUNT_CLIENT_SECRET: ${{ secrets.SERVICE_SECRET_STG }}
+          LONG_TEST_VALUE: ${{ vars.LONG_TEST_VALUE }}
         run: |
           go test . -timeout=30m -v --tags=integration_v0
diff --git a/.github/workflows/integration-tests-v2.yml b/.github/workflows/integration-tests-v2.yml
@@ -37,5 +37,6 @@ jobs:
           ACCOUNT_NAME: ${{ vars.FIREBOLT_ACCOUNT }}
           CLIENT_ID: ${{ secrets.FIREBOLT_CLIENT_ID_STG_NEW_IDN }}
           CLIENT_SECRET: ${{ secrets.FIREBOLT_CLIENT_SECRET_STG_NEW_IDN }}
+          LONG_TEST_VALUE: ${{ vars.LONG_TEST_VALUE }}
         run: |
           go test . -timeout=30m -v --tags=integration
diff --git a/.github/workflows/nightly-v1.yml b/.github/workflows/nightly-v1.yml
@@ -54,6 +54,7 @@ jobs:
           ACCOUNT_NAME: "firebolt"
           SERVICE_ACCOUNT_CLIENT_ID: ${{ secrets.FIREBOLT_CLIENT_ID_STAGING }}
           SERVICE_ACCOUNT_CLIENT_SECRET: ${{ secrets.FIREBOLT_CLIENT_SECRET_STAGING }}
+          LONG_TEST_VALUE: ${{ vars.LONG_TEST_VALUE }}
         run: |
           go test .  -timeout=30m -v -coverprofile cover.out --tags=integration_v0
 

diff --git a/.github/workflows/nightly-v2.yml b/.github/workflows/nightly-v2.yml
@@ -48,6 +48,7 @@ jobs:
           ACCOUNT_NAME: ${{ vars.FIREBOLT_ACCOUNT }}
           CLIENT_ID: ${{ secrets.FIREBOLT_CLIENT_ID_STG_NEW_IDN }}
           CLIENT_SECRET: ${{ secrets.FIREBOLT_CLIENT_SECRET_STG_NEW_IDN }}
+          LONG_TEST_VALUE: ${{ vars.LONG_TEST_VALUE }}
         run: |
           go test . -timeout=30m -v -coverprofile cover.out --tags=integration
 

diff --git a/connection_common_integration_test.go b/connection_common_integration_test.go
@@ -7,6 +7,7 @@ import (
 	"bytes"
 	"context"
 	"database/sql"
+	"os"
 	"reflect"
 	"testing"
 	"time"
@@ -19,6 +20,12 @@ const SCAN_STATEMENT_ERROR_MSG = "firebolt rows Scan() call failed with %v"
 const VALUES_ARE_NOT_EQUAL_ERROR_MSG = "values are not equal: %v and %v\n"
 const RESULTS_ARE_NOT_EQUAL_ERROR_MSG = "results are not equal "
 
+var longTestValue string
+
+func init() {
+	longTestValue = os.Getenv("LONG_TEST_VALUE")
+}
+
 // TestConnectionPrepareStatement, tests that prepare statement doesn't result into an error
 func TestConnectionSetStatement(t *testing.T) {
 	conn, err := sql.Open("firebolt", dsnMock)
@@ -363,7 +370,7 @@ func TestConnectionQueryByteaType(t *testing.T) {
 }
 
 func TestLongQuery(t *testing.T) {
-	var maxValue = 400000000000
+	var maxValue = longTestValue
 
 	finished_in := make(chan time.Duration, 1)
 	go func() {