Carlos Florêncio
73 lines
1.7 KiB
Go
73 lines
1.7 KiB
Go
package db
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
func TestTableColumnsAndExistsSQLite(t *testing.T) {
|
|
rawDB, err := OpenSQLite(filepath.Join(t.TempDir(), "schema.db"))
|
|
if err != nil {
|
|
t.Fatalf("open sqlite: %v", err)
|
|
}
|
|
database := sqlx.NewDb(rawDB, "sqlite3")
|
|
t.Cleanup(func() { _ = database.Close() })
|
|
|
|
if _, err := database.Exec(`CREATE TABLE sample_schema (
|
|
id TEXT PRIMARY KEY,
|
|
created_at DATETIME NOT NULL
|
|
)`); err != nil {
|
|
t.Fatalf("create table: %v", err)
|
|
}
|
|
|
|
exists, err := TableExists(database, "sample_schema")
|
|
if err != nil {
|
|
t.Fatalf("table exists: %v", err)
|
|
}
|
|
if !exists {
|
|
t.Fatal("TableExists = false, want true")
|
|
}
|
|
|
|
columns, err := 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)
|
|
}
|
|
|
|
for _, column := range []struct {
|
|
name string
|
|
want bool
|
|
}{
|
|
{name: "id", want: true},
|
|
{name: "missing", want: false},
|
|
} {
|
|
got, err := ColumnExists(database, "sample_schema", column.name)
|
|
if err != nil {
|
|
t.Fatalf("column %s exists: %v", column.name, err)
|
|
}
|
|
if got != column.want {
|
|
t.Errorf("ColumnExists(%q) = %v, want %v", column.name, got, column.want)
|
|
}
|
|
}
|
|
|
|
missingTable, err := 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 := 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)
|
|
}
|
|
}
|