diff --git a/graph/db/test_kvdb.go b/graph/db/test_kvdb.go index 569e0949f..22904f858 100644 --- a/graph/db/test_kvdb.go +++ b/graph/db/test_kvdb.go @@ -9,6 +9,9 @@ import ( "github.com/stretchr/testify/require" ) +// isSQLDB indicates that this build does not use a SQL database. +var isSQLDB = false + // NewTestDB is a helper function that creates an BBolt database for testing. func NewTestDB(t testing.TB) Store { backend, backendCleanup, err := kvdb.GetTestBackend(t.TempDir(), "cgr") diff --git a/graph/db/test_postgres.go b/graph/db/test_postgres.go index 716210afa..a1bd61a2c 100644 --- a/graph/db/test_postgres.go +++ b/graph/db/test_postgres.go @@ -11,6 +11,9 @@ import ( "github.com/stretchr/testify/require" ) +// isSQLDB indicates that this build uses a SQL database. +var isSQLDB = true + // NewTestDB is a helper function that creates a SQLStore backed by a SQL // database for testing. func NewTestDB(t testing.TB) Store { diff --git a/graph/db/test_sqlite.go b/graph/db/test_sqlite.go index 68fef6eb9..137ab7aa9 100644 --- a/graph/db/test_sqlite.go +++ b/graph/db/test_sqlite.go @@ -11,6 +11,9 @@ import ( "github.com/stretchr/testify/require" ) +// isSQLDB indicates that this build uses a SQL database. +var isSQLDB = true + // NewTestDB is a helper function that creates a SQLStore backed by a SQL // database for testing. func NewTestDB(t testing.TB) Store {