65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package satellitedbtest_test
|
|
|
|
import (
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap"
|
|
|
|
"storj.io/storj/private/dbutil"
|
|
"storj.io/storj/private/dbutil/pgutil/pgtest"
|
|
"storj.io/storj/private/testcontext"
|
|
dbx "storj.io/storj/satellite/satellitedb/dbx"
|
|
"storj.io/storj/satellite/satellitedb/satellitedbtest"
|
|
)
|
|
|
|
func TestNewCockroach(t *testing.T) {
|
|
ctx := testcontext.New(t)
|
|
defer ctx.Cleanup()
|
|
|
|
if *pgtest.CrdbConnStr == "" {
|
|
t.Skip("Cockroachdb flag missing")
|
|
}
|
|
namespacedDBName := "name#spaced/Test/DB"
|
|
testdb, err := satellitedbtest.NewCockroach(zap.L(), namespacedDBName)
|
|
require.NoError(t, err)
|
|
|
|
// assert new test db exists
|
|
driver, source, _, err := dbutil.SplitConnStr(*pgtest.CrdbConnStr)
|
|
require.NoError(t, err)
|
|
|
|
db, err := dbx.Open(driver, source)
|
|
require.NoError(t, err)
|
|
defer ctx.Check(db.Close)
|
|
|
|
// NewCockroach removes all non-alphanumeric characters from the name. We need it to match.
|
|
r := regexp.MustCompile(`\W`)
|
|
formattedName := r.ReplaceAllString(namespacedDBName, "")
|
|
|
|
var exists *bool
|
|
row := db.QueryRow(`SELECT EXISTS (
|
|
SELECT datname FROM pg_catalog.pg_database WHERE lower(datname) = lower($1)
|
|
);`, formattedName,
|
|
)
|
|
err = row.Scan(&exists)
|
|
require.NoError(t, err)
|
|
assert.True(t, *exists)
|
|
|
|
err = testdb.Close()
|
|
require.NoError(t, err)
|
|
|
|
// assert new test db was deleted
|
|
row = db.QueryRow(`SELECT EXISTS (
|
|
SELECT datname FROM pg_catalog.pg_database WHERE lower(datname) = lower($1)
|
|
);`, formattedName,
|
|
)
|
|
err = row.Scan(&exists)
|
|
require.NoError(t, err)
|
|
assert.False(t, *exists)
|
|
}
|