peachy-go/db_test.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
			})
	}
}