Carlos Florêncio
59 lines
1.5 KiB
Go
59 lines
1.5 KiB
Go
package db_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/kandev/kandev/internal/db"
|
|
"github.com/kandev/kandev/internal/testutil"
|
|
)
|
|
|
|
func TestTableColumnsAndExistsPostgres(t *testing.T) {
|
|
database := testutil.OpenIsolatedPostgres(t, testutil.PostgresDSNFromEnv(t))
|
|
if _, err := database.Exec(`CREATE TABLE sample_schema (
|
|
id TEXT PRIMARY KEY,
|
|
created_at TIMESTAMPTZ NOT NULL
|
|
)`); err != nil {
|
|
t.Fatalf("create table: %v", err)
|
|
}
|
|
|
|
exists, err := db.TableExists(database, "sample_schema")
|
|
if err != nil {
|
|
t.Fatalf("table exists: %v", err)
|
|
}
|
|
if !exists {
|
|
t.Fatal("TableExists = false, want true")
|
|
}
|
|
|
|
columns, err := db.TableColumns(database, "sample_schema")
|
|
if err != nil {
|
|
t.Fatalf("table columns: %v", err)
|
|
}
|
|
if !columns["created_at"] || columns["missing"] {
|
|
t.Fatalf("columns = %#v, want created_at and not missing", columns)
|
|
}
|
|
|
|
got, err := db.ColumnExists(database, "sample_schema", "missing")
|
|
if err != nil {
|
|
t.Fatalf("missing column exists: %v", err)
|
|
}
|
|
if got {
|
|
t.Fatal("ColumnExists for missing column = true, want false")
|
|
}
|
|
|
|
missingTable, err := db.TableExists(database, "missing_schema")
|
|
if err != nil {
|
|
t.Fatalf("missing table exists: %v", err)
|
|
}
|
|
if missingTable {
|
|
t.Fatal("TableExists for missing table = true, want false")
|
|
}
|
|
|
|
missingColumns, err := db.TableColumns(database, "missing_schema")
|
|
if err != nil {
|
|
t.Fatalf("missing table columns: %v", err)
|
|
}
|
|
if len(missingColumns) != 0 {
|
|
t.Fatalf("missing table columns = %#v, want empty", missingColumns)
|
|
}
|
|
}
|