peachy-go/db_test.go

92 lines
2.1 KiB
Go

package peachy_test
import (
"os"
"path/filepath"
"testing"
"git.codemonkeysoftware.net/b/peachy-go"
"github.com/shoenig/test"
"github.com/shoenig/test/must"
"zombiezen.com/go/sqlite"
)
func pending(t *testing.T) {
t.Fatalf("test not implemented")
}
func closeIfExists(db *peachy.DB) {
if db != nil {
db.Close()
}
}
func TestOpen(t *testing.T) {
t.Run("Open succeeds on DB created by Create", func(t *testing.T) {
dir := t.TempDir()
dbPath := filepath.Join(dir, "test.db")
db, err := peachy.Create(dbPath)
must.NoError(t, err)
db.Close()
db, err = peachy.Open(dbPath)
defer closeIfExists(db)
must.NotNil(t, db)
must.NoError(t, err)
// TODO Make sure the DB works once it has any methods.
})
t.Run("Open fails on alien SQLite DB", func(t *testing.T) {
dir := t.TempDir()
dbPath := filepath.Join(dir, "test.db")
conn, err := sqlite.OpenConn(dbPath)
must.NoError(t, err)
conn.Close()
db, err := peachy.Open(dbPath)
defer closeIfExists(db)
must.Nil(t, db)
must.ErrorIs(t, err, peachy.ErrInvalidDB)
})
t.Run("Open fails on nonexistent file", func(t *testing.T) {
dir := t.TempDir()
dbPath := filepath.Join(dir, "test.db")
db, err := peachy.Open(dbPath)
defer closeIfExists(db)
must.Nil(t, db)
must.ErrorIs(t, err, peachy.ErrFileNotExist)
must.FileNotExists(t, dbPath)
})
t.Run("Open fails on non-SQLite file", func(t *testing.T) {
path := filepath.Join(t.TempDir(), "test.db")
err := os.WriteFile(path, []byte("I'm not a database."), os.ModePerm)
must.NoError(t, err)
db, err := peachy.Open(path)
defer closeIfExists(db)
must.Nil(t, db)
must.ErrorIs(t, err, peachy.ErrInvalidDB, must.Sprint(sqlite.ErrCode(err)))
})
}
func TestFieldTypes(t *testing.T) {
db, err := peachy.Create(":memory:")
must.NoError(t, err)
defer db.Close()
names := []string{"alpha", `"bravo`, " charl!e"}
for _, name := range names {
err := db.AddFieldType(name)
must.NoError(t, err)
}
fieldTypes, err := db.GetFieldTypes()
must.NoError(t, err)
for _, name := range names {
test.SliceContainsFunc(t, fieldTypes, name,
func(ft peachy.FieldType, s string) bool {
return ft.Name == s
})
}
}