refactor: try moving skip for bq tests around

PeerDB-io · Jan 18, 2025 · 28cc355 · 28cc355
1 parent 32d14b9
commit 28cc355
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/flow/e2e/bigquery/bigquery.go b/flow/e2e/bigquery/bigquery.go
@@ -2,6 +2,7 @@ package e2e_bigquery
 
 import (
 	"fmt"
+	"os"
 	"strings"
 	"testing"
 	"time"
@@ -88,6 +89,9 @@ func (s PeerFlowE2ETestSuiteBQ) Teardown() {
 }
 
 func SetupSuite(t *testing.T) PeerFlowE2ETestSuiteBQ {
+	if val, ok := os.LookupEnv("CI_PG_VERSION"); ok && val != "16" {
+		t.Skip("Only running in PG16 to reduce flakiness from high concurrency")
+	}
 	t.Helper()
 
 	suffix := shared.RandomString(8)

diff --git a/flow/e2e/bigquery/peer_flow_bq_test.go b/flow/e2e/bigquery/peer_flow_bq_test.go
@@ -21,10 +21,6 @@ import (
 )
 
 func TestPeerFlowE2ETestSuiteBQ(t *testing.T) {
-	if val, ok := os.LookupEnv("CI_PG_VERSION"); ok && val != "16" {
-		t.Skip("Only running in PG16 to reduce flakiness from high concurrency")
-	}
-
 	e2eshared.RunSuite(t, SetupSuite)
 }