64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
|
// Copyright (C) 2018 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package satellitedb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"flag"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
|
||
|
|
"storj.io/storj/internal/testcontext"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
// this connstring is expected to work under the storj-test docker-compose instance
|
||
|
|
defaultPostgresConn = "postgres://storj:storj-pass@test-postgres/teststorj?sslmode=disable"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
testPostgres = flag.String("postgres-test-db", os.Getenv("STORJ_POSTGRES_TEST"), "PostgreSQL test database connection string")
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDatabase(t *testing.T) {
|
||
|
|
testDrivers(t, func(ctx *testcontext.Context, t *testing.T, db *DB) {
|
||
|
|
err := db.CreateTables()
|
||
|
|
assert.NoError(t, err)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func testDrivers(t *testing.T, fn func(ctx *testcontext.Context, t *testing.T, db *DB)) {
|
||
|
|
t.Run("Sqlite", func(t *testing.T) {
|
||
|
|
ctx := testcontext.New(t)
|
||
|
|
defer ctx.Cleanup()
|
||
|
|
|
||
|
|
// creating in-memory db and opening connection
|
||
|
|
db, err := NewDB("sqlite3://file::memory:?mode=memory&cache=shared")
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer ctx.Check(db.Close)
|
||
|
|
|
||
|
|
fn(ctx, t, db)
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("Postgres", func(t *testing.T) {
|
||
|
|
if *testPostgres == "" {
|
||
|
|
t.Skipf("postgres flag missing, example:\n-postgres-test-db=%s", defaultPostgresConn)
|
||
|
|
}
|
||
|
|
|
||
|
|
ctx := testcontext.New(t)
|
||
|
|
defer ctx.Cleanup()
|
||
|
|
|
||
|
|
db, err := NewDB(*testPostgres)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
defer ctx.Check(db.Close)
|
||
|
|
|
||
|
|
fn(ctx, t, db)
|
||
|
|
})
|
||
|
|
}
|