weijuesen
132 lines
3.5 KiB
Go
132 lines
3.5 KiB
Go
package database
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
"silk-server-go/migrations"
|
|
|
|
"github.com/golang-migrate/migrate/v4/source/iofs"
|
|
"gorm.io/driver/postgres"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func newMockGormDB(t *testing.T) (*gorm.DB, sqlmock.Sqlmock) {
|
|
t.Helper()
|
|
|
|
sqlDB, mock, err := sqlmock.New()
|
|
if err != nil {
|
|
t.Fatalf("create sqlmock: %v", err)
|
|
}
|
|
|
|
gdb, err := gorm.Open(postgres.New(postgres.Config{Conn: sqlDB}), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatalf("open gorm with sqlmock: %v", err)
|
|
}
|
|
|
|
return gdb, mock
|
|
}
|
|
|
|
func TestCheckSchemaVersionRejectsMismatch(t *testing.T) {
|
|
db, mock := newMockGormDB(t)
|
|
defer func() {
|
|
sqlDB, _ := db.DB()
|
|
_ = sqlDB.Close()
|
|
}()
|
|
|
|
mock.ExpectQuery("SELECT version, dirty FROM schema_migrations").
|
|
WillReturnRows(sqlmock.NewRows([]string{"version", "dirty"}).AddRow(1, false))
|
|
|
|
if err := CheckSchemaVersion(db, "2"); err == nil {
|
|
t.Fatal("expected schema mismatch error")
|
|
}
|
|
}
|
|
|
|
func TestCheckSchemaVersionAcceptsMatch(t *testing.T) {
|
|
db, mock := newMockGormDB(t)
|
|
defer func() {
|
|
sqlDB, _ := db.DB()
|
|
_ = sqlDB.Close()
|
|
}()
|
|
|
|
mock.ExpectQuery("SELECT version, dirty FROM schema_migrations").
|
|
WillReturnRows(sqlmock.NewRows([]string{"version", "dirty"}).AddRow(1, false))
|
|
|
|
if err := CheckSchemaVersion(db, "1"); err != nil {
|
|
t.Fatalf("expected schema version match, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestCheckSchemaVersionRejectsMissingVersion(t *testing.T) {
|
|
db, mock := newMockGormDB(t)
|
|
defer func() {
|
|
sqlDB, _ := db.DB()
|
|
_ = sqlDB.Close()
|
|
}()
|
|
|
|
mock.ExpectQuery("SELECT version, dirty FROM schema_migrations").
|
|
WillReturnRows(sqlmock.NewRows([]string{"version", "dirty"}))
|
|
|
|
if err := CheckSchemaVersion(db, "1"); err == nil {
|
|
t.Fatal("expected missing schema version error")
|
|
}
|
|
}
|
|
|
|
func TestCheckSchemaVersionRejectsDirty(t *testing.T) {
|
|
db, mock := newMockGormDB(t)
|
|
defer func() {
|
|
sqlDB, _ := db.DB()
|
|
_ = sqlDB.Close()
|
|
}()
|
|
|
|
mock.ExpectQuery("SELECT version, dirty FROM schema_migrations").
|
|
WillReturnRows(sqlmock.NewRows([]string{"version", "dirty"}).AddRow(1, true))
|
|
|
|
if err := CheckSchemaVersion(db, "1"); err == nil {
|
|
t.Fatal("expected dirty migration error")
|
|
}
|
|
}
|
|
|
|
func TestEmbeddedMigrationsIncludeBaseline(t *testing.T) {
|
|
driver, err := iofs.New(migrations.FS, ".")
|
|
if err != nil {
|
|
t.Fatalf("load embedded migrations: %v", err)
|
|
}
|
|
|
|
version, err := driver.First()
|
|
if err != nil {
|
|
t.Fatalf("read first migration: %v", err)
|
|
}
|
|
if version != 1 {
|
|
t.Fatalf("expected baseline migration version 1, got %d", version)
|
|
}
|
|
next, err := driver.Next(version)
|
|
if err != nil || next != 2 {
|
|
t.Fatalf("expected risk assessment migration version 2, got %d (err %v)", next, err)
|
|
}
|
|
next, err = driver.Next(next)
|
|
if err != nil || next != 4 {
|
|
t.Fatalf("expected notifications/outbox migration version 4, got %d (err %v)", next, err)
|
|
}
|
|
next, err = driver.Next(next)
|
|
if err != nil || next != 5 {
|
|
t.Fatalf("expected detection/disease migration version 5, got %d (err %v)", next, err)
|
|
}
|
|
next, err = driver.Next(next)
|
|
if err != nil || next != 6 {
|
|
t.Fatalf("expected biosecurity migration version 6, got %d (err %v)", next, err)
|
|
}
|
|
next, err = driver.Next(next)
|
|
if err != nil || next != 7 {
|
|
t.Fatalf("expected inspection idempotency migration version 7, got %d (err %v)", next, err)
|
|
}
|
|
next, err = driver.Next(next)
|
|
if err != nil || next != 8 {
|
|
t.Fatalf("expected governance migration version 8, got %d (err %v)", next, err)
|
|
}
|
|
next, err = driver.Next(next)
|
|
if err != nil || next != 9 {
|
|
t.Fatalf("expected object authority migration version 9, got %d (err %v)", next, err)
|
|
}
|
|
}
|