mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-10-25 19:44:10 +00:00
164 lines
3.2 KiB
Go
164 lines
3.2 KiB
Go
package migrate
|
|
|
|
import (
|
|
"database/sql"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/russross/meddler"
|
|
)
|
|
|
|
type Sample struct {
|
|
ID int64 `meddler:"id,pk"`
|
|
Imel string `meddler:"imel"`
|
|
Name string `meddler:"name"`
|
|
}
|
|
|
|
type RenameSample struct {
|
|
ID int64 `meddler:"id,pk"`
|
|
Email string `meddler:"email"`
|
|
Name string `meddler:"name"`
|
|
}
|
|
|
|
type AddColumnSample struct {
|
|
ID int64 `meddler:"id,pk"`
|
|
Imel string `meddler:"imel"`
|
|
Name string `meddler:"name"`
|
|
Num int64 `meddler:"num"`
|
|
}
|
|
|
|
type RemoveColumnSample struct {
|
|
ID int64 `meddler:"id,pk"`
|
|
Name string `meddler:"name"`
|
|
}
|
|
|
|
// ---------- revision 1
|
|
|
|
type revision1 struct{}
|
|
|
|
func (r *revision1) Up(op Operation) error {
|
|
_, err := op.CreateTable("samples", []string{
|
|
"id INTEGER PRIMARY KEY AUTOINCREMENT",
|
|
"imel VARCHAR(255) UNIQUE",
|
|
"name VARCHAR(255)",
|
|
})
|
|
return err
|
|
}
|
|
|
|
func (r *revision1) Down(op Operation) error {
|
|
_, err := op.DropTable("samples")
|
|
return err
|
|
}
|
|
|
|
func (r *revision1) Revision() int64 {
|
|
return 1
|
|
}
|
|
|
|
// ---------- end of revision 1
|
|
|
|
// ---------- revision 2
|
|
|
|
type revision2 struct{}
|
|
|
|
func (r *revision2) Up(op Operation) error {
|
|
_, err := op.RenameTable("samples", "examples")
|
|
return err
|
|
}
|
|
|
|
func (r *revision2) Down(op Operation) error {
|
|
_, err := op.RenameTable("examples", "samples")
|
|
return err
|
|
}
|
|
|
|
func (r *revision2) Revision() int64 {
|
|
return 2
|
|
}
|
|
|
|
// ---------- end of revision 2
|
|
|
|
var db *sql.DB
|
|
|
|
var testSchema = `
|
|
CREATE TABLE samples (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
imel VARCHAR(255) UNIQUE,
|
|
name VARCHAR(255),
|
|
);
|
|
`
|
|
|
|
var dataDump = []string{
|
|
`INSERT INTO samples (imel, name) VALUES ('test@example.com', 'Test Tester');`,
|
|
`INSERT INTO samples (imel, name) VALUES ('foo@bar.com', 'Foo Bar');`,
|
|
`INSERT INTO samples (imel, name) VALUES ('crash@bandicoot.io', 'Crash Bandicoot');`,
|
|
}
|
|
|
|
func TestMigrateCreateTable(t *testing.T) {
|
|
defer tearDown()
|
|
if err := setUp(); err != nil {
|
|
t.Fatalf("Error preparing database: %q", err)
|
|
}
|
|
|
|
Driver = SQLite
|
|
|
|
mgr := New(db)
|
|
if err := mgr.Add(&revision1{}).Migrate(); err != nil {
|
|
t.Errorf("Can not migrate: %q", err)
|
|
}
|
|
|
|
sample := Sample{
|
|
ID: 1,
|
|
Imel: "test@example.com",
|
|
Name: "Test Tester",
|
|
}
|
|
if err := meddler.Save(db, "samples", &sample); err != nil {
|
|
t.Errorf("Can not save data: %q", err)
|
|
}
|
|
}
|
|
|
|
func TestMigrateRenameTable(t *testing.T) {
|
|
defer tearDown()
|
|
if err := setUp(); err != nil {
|
|
t.Fatalf("Error preparing database: %q", err)
|
|
}
|
|
|
|
Driver = SQLite
|
|
|
|
mgr := New(db)
|
|
if err := mgr.Add(&revision1{}).Migrate(); err != nil {
|
|
t.Errorf("Can not migrate: %q", err)
|
|
}
|
|
|
|
loadFixture(t)
|
|
|
|
if err := mgr.Add(&revision2{}).Migrate(); err != nil {
|
|
t.Errorf("Can not migrate: %q", err)
|
|
}
|
|
|
|
sample := Sample{}
|
|
if err := meddler.QueryRow(db, &sample, `SELECT * FROM examples WHERE id = ?`, 2); err != nil {
|
|
t.Errorf("Can not fetch data: %q", err)
|
|
}
|
|
|
|
if sample.Imel != "foo@bar.com" {
|
|
t.Errorf("Column doesn't match\n\texpect:\t%s\n\tget:\t%s", "foo@bar.com", sample.Imel)
|
|
}
|
|
}
|
|
|
|
func setUp() error {
|
|
var err error
|
|
db, err = sql.Open("sqlite3", "migration_tests.sqlite")
|
|
return err
|
|
}
|
|
|
|
func tearDown() {
|
|
db.Close()
|
|
os.Remove("migration_tests.sqlite")
|
|
}
|
|
|
|
func loadFixture(t *testing.T) {
|
|
for _, sql := range dataDump {
|
|
if _, err := db.Exec(sql); err != nil {
|
|
t.Errorf("Can not insert into database: %q", err)
|
|
}
|
|
}
|
|
}
|