mirror of
https://github.com/coder/coder.git
synced 2025-07-03 16:13:58 +00:00
113 lines
2.9 KiB
Go
113 lines
2.9 KiB
Go
package dbtestutil_test
|
|
|
|
import (
|
|
"database/sql"
|
|
"testing"
|
|
|
|
_ "github.com/lib/pq"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/goleak"
|
|
|
|
"github.com/coder/coder/v2/coderd/database/dbtestutil"
|
|
"github.com/coder/coder/v2/coderd/database/migrations"
|
|
"github.com/coder/coder/v2/testutil"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
goleak.VerifyTestMain(m, testutil.GoleakOptions...)
|
|
}
|
|
|
|
func TestOpen(t *testing.T) {
|
|
t.Parallel()
|
|
if !dbtestutil.WillUsePostgres() {
|
|
t.Skip("this test requires postgres")
|
|
}
|
|
|
|
connect, err := dbtestutil.Open(t)
|
|
require.NoError(t, err)
|
|
|
|
db, err := sql.Open("postgres", connect)
|
|
require.NoError(t, err)
|
|
err = db.Ping()
|
|
require.NoError(t, err)
|
|
err = db.Close()
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestOpen_InvalidDBFrom(t *testing.T) {
|
|
t.Parallel()
|
|
if !dbtestutil.WillUsePostgres() {
|
|
t.Skip("this test requires postgres")
|
|
}
|
|
|
|
_, err := dbtestutil.Open(t, dbtestutil.WithDBFrom("__invalid__"))
|
|
require.Error(t, err)
|
|
require.ErrorContains(t, err, "template database")
|
|
require.ErrorContains(t, err, "does not exist")
|
|
}
|
|
|
|
func TestOpen_ValidDBFrom(t *testing.T) {
|
|
t.Parallel()
|
|
if !dbtestutil.WillUsePostgres() {
|
|
t.Skip("this test requires postgres")
|
|
}
|
|
|
|
// first check if we can create a new template db
|
|
dsn, err := dbtestutil.Open(t, dbtestutil.WithDBFrom(""))
|
|
require.NoError(t, err)
|
|
|
|
db, err := sql.Open("postgres", dsn)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
require.NoError(t, db.Close())
|
|
})
|
|
|
|
err = db.Ping()
|
|
require.NoError(t, err)
|
|
|
|
templateDBName := "tpl_" + migrations.GetMigrationsHash()[:32]
|
|
tplDbExistsRes, err := db.Query("SELECT 1 FROM pg_database WHERE datname = $1", templateDBName)
|
|
if err != nil {
|
|
require.NoError(t, err)
|
|
}
|
|
require.True(t, tplDbExistsRes.Next())
|
|
require.NoError(t, tplDbExistsRes.Close())
|
|
|
|
// now populate the db with some data and use it as a new template db
|
|
// to verify that dbtestutil.Open respects WithDBFrom
|
|
_, err = db.Exec("CREATE TABLE my_wonderful_table (id serial PRIMARY KEY, name text)")
|
|
require.NoError(t, err)
|
|
_, err = db.Exec("INSERT INTO my_wonderful_table (name) VALUES ('test')")
|
|
require.NoError(t, err)
|
|
|
|
rows, err := db.Query("SELECT current_database()")
|
|
require.NoError(t, err)
|
|
require.True(t, rows.Next())
|
|
var freshTemplateDBName string
|
|
require.NoError(t, rows.Scan(&freshTemplateDBName))
|
|
require.NoError(t, rows.Close())
|
|
require.NoError(t, db.Close())
|
|
|
|
for i := 0; i < 10; i++ {
|
|
db, err := sql.Open("postgres", dsn)
|
|
require.NoError(t, err)
|
|
require.NoError(t, db.Ping())
|
|
require.NoError(t, db.Close())
|
|
}
|
|
|
|
// now create a new db from the template db
|
|
newDsn, err := dbtestutil.Open(t, dbtestutil.WithDBFrom(freshTemplateDBName))
|
|
require.NoError(t, err)
|
|
|
|
newDb, err := sql.Open("postgres", newDsn)
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() {
|
|
require.NoError(t, newDb.Close())
|
|
})
|
|
|
|
rows, err = newDb.Query("SELECT 1 FROM my_wonderful_table WHERE name = 'test'")
|
|
require.NoError(t, err)
|
|
require.True(t, rows.Next())
|
|
require.NoError(t, rows.Close())
|
|
}
|