Function TestOpen_ValidDBFrom

coderd/database/dbtestutil/postgres_test.go:44–104

(t *testing.T)

}

func TestOpen_ValidDBFrom(t *testing.T) {
	t.Parallel()

	// 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)