mirror of
https://github.com/hanzoai/migrate.git
synced 2026-09-30 22:55:10 +00:00
Eng Zer Jun
1a301f2273
The directory created by `T.TempDir` is automatically removed when the test and all its subtests complete. Reference: https://pkg.go.dev/testing#T.TempDir Signed-off-by: Eng Zer Jun <[email protected]>
64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package ql
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/golang-migrate/migrate/v4"
|
|
dt "github.com/golang-migrate/migrate/v4/database/testing"
|
|
_ "github.com/golang-migrate/migrate/v4/source/file"
|
|
_ "modernc.org/ql/driver"
|
|
)
|
|
|
|
func Test(t *testing.T) {
|
|
dir := t.TempDir()
|
|
t.Logf("DB path : %s\n", filepath.Join(dir, "ql.db"))
|
|
p := &Ql{}
|
|
addr := fmt.Sprintf("ql://%s", filepath.Join(dir, "ql.db"))
|
|
d, err := p.Open(addr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
db, err := sql.Open("ql", filepath.Join(dir, "ql.db"))
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer func() {
|
|
if err := db.Close(); err != nil {
|
|
return
|
|
}
|
|
}()
|
|
dt.Test(t, d, []byte("CREATE TABLE t (Qty int, Name string);"))
|
|
}
|
|
|
|
func TestMigrate(t *testing.T) {
|
|
dir := t.TempDir()
|
|
t.Logf("DB path : %s\n", filepath.Join(dir, "ql.db"))
|
|
|
|
db, err := sql.Open("ql", filepath.Join(dir, "ql.db"))
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer func() {
|
|
if err := db.Close(); err != nil {
|
|
return
|
|
}
|
|
}()
|
|
|
|
driver, err := WithInstance(db, &Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
m, err := migrate.NewWithDatabaseInstance(
|
|
"file://./examples/migrations",
|
|
"ql", driver)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dt.TestMigrate(t, m)
|
|
}
|