mirror of
https://github.com/status-im/migrate.git
synced 2025-02-23 08:18:07 +00:00
480a5a634a
* Consistently lock in ensureVersionTable and do not call ensureVersionTable from Drop across all database implementations * Add test for dropping postgres databases * Fix failing database tests * Fix CockroachDb test, lock table should be created before versionTable * Add Initialize() to Driver interface, and add integration tests for Drop() between database implementations and migrate * Remove Initialize, document breaking behaviour of Drop * Revert introduction of Initialize method * Removed Initialize in Stub as well * Remove call to non-existent Initialize and make sure to close re-initialized database connections * Revert changes to TestDrop in database/testing * Split Test and TestMigrate into different test entrypoints * Remove unused import in migrate_testing * Remove erroneous code to fix tests * Add stub source imports to database tests * Add Stub source to migrate tests * Use example migrations for tests * Add file driver to database tests * Align database directory layout * Add file source driver to Cassandra * Review changes * Minor syntactic change for cleaner diff
122 lines
2.4 KiB
Go
122 lines
2.4 KiB
Go
package sqlite3
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"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"
|
|
_ "github.com/mattn/go-sqlite3"
|
|
)
|
|
|
|
func Test(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "sqlite3-driver-test")
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer func() {
|
|
os.RemoveAll(dir)
|
|
}()
|
|
t.Logf("DB path : %s\n", filepath.Join(dir, "sqlite3.db"))
|
|
p := &Sqlite{}
|
|
addr := fmt.Sprintf("sqlite3://%s", filepath.Join(dir, "sqlite3.db"))
|
|
d, err := p.Open(addr)
|
|
if err != nil {
|
|
t.Fatalf("%v", err)
|
|
}
|
|
dt.Test(t, d, []byte("CREATE TABLE t (Qty int, Name string);"))
|
|
}
|
|
|
|
func TestMigrate(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "sqlite3-driver-test")
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer func() {
|
|
os.RemoveAll(dir)
|
|
}()
|
|
t.Logf("DB path : %s\n", filepath.Join(dir, "sqlite3.db"))
|
|
p := &Sqlite{}
|
|
addr := fmt.Sprintf("sqlite3://%s", filepath.Join(dir, "sqlite3.db"))
|
|
d, err := p.Open(addr)
|
|
if err != nil {
|
|
t.Fatalf("%v", err)
|
|
}
|
|
|
|
db, err := sql.Open("sqlite3", filepath.Join(dir, "sqlite3.db"))
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer func() {
|
|
if err := db.Close(); err != nil {
|
|
return
|
|
}
|
|
}()
|
|
driver, err := WithInstance(db, &Config{})
|
|
if err != nil {
|
|
t.Fatalf("%v", err)
|
|
}
|
|
if err := d.Drop(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
m, err := migrate.NewWithDatabaseInstance(
|
|
"file://./examples/migrations",
|
|
"ql", driver)
|
|
if err != nil {
|
|
t.Fatalf("%v", err)
|
|
}
|
|
dt.TestMigrate(t, m, []byte("CREATE TABLE t (Qty int, Name string);"))
|
|
}
|
|
|
|
func TestMigrationTable(t *testing.T) {
|
|
dir, err := ioutil.TempDir("", "sqlite3-driver-test-migration-table")
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer func() {
|
|
os.RemoveAll(dir)
|
|
}()
|
|
|
|
t.Logf("DB path : %s\n", filepath.Join(dir, "sqlite3.db"))
|
|
|
|
db, err := sql.Open("sqlite3", filepath.Join(dir, "sqlite3.db"))
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer func() {
|
|
if err := db.Close(); err != nil {
|
|
return
|
|
}
|
|
}()
|
|
|
|
config := &Config{
|
|
MigrationsTable: "my_migration_table",
|
|
}
|
|
driver, err := WithInstance(db, config)
|
|
if err != nil {
|
|
t.Fatalf("%v", err)
|
|
}
|
|
m, err := migrate.NewWithDatabaseInstance(
|
|
"file://./examples/migrations",
|
|
"ql", driver)
|
|
if err != nil {
|
|
t.Fatalf("%v", err)
|
|
}
|
|
t.Log("UP")
|
|
err = m.Up()
|
|
if err != nil {
|
|
t.Fatalf("%v", err)
|
|
}
|
|
|
|
_, err = db.Query(fmt.Sprintf("SELECT * FROM %s", config.MigrationsTable))
|
|
if err != nil {
|
|
t.Fatalf("%v", err)
|
|
}
|
|
}
|