contrib/go-pg/pg.v10: add INTEGRATION flag check for tests (#921)

DataDog · May 12, 2021 · e19dac6 · e19dac6
1 parent 4ba66f6
commit e19dac6
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/contrib/go-pg/pg.v10/pg_go_test.go b/contrib/go-pg/pg.v10/pg_go_test.go
@@ -7,6 +7,8 @@ package pg
 
 import (
 	"context"
+	"fmt"
+	"os"
 	"testing"
 
 	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/ext"
@@ -18,6 +20,15 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
+func TestMain(m *testing.M) {
+	_, ok := os.LookupEnv("INTEGRATION")
+	if !ok {
+		fmt.Println("--- SKIP: to enable integration test, set the INTEGRATION environment variable")
+		os.Exit(0)
+	}
+	os.Exit(m.Run())
+}
+
 func TestImplementsHook(t *testing.T) {
 	var _ pg.QueryHook = (*queryHook)(nil)
 }