migrate.go 356 B

12345678910111213141516171819
  1. package tests
  2. import (
  3. "testing"
  4. "github.com/jinzhu/gorm"
  5. )
  6. func TestMigrate(t *testing.T, db *gorm.DB) {
  7. allModels := []interface{}{&User{}, &Account{}, &Pet{}, &Toy{}, &Company{}, &Language{}}
  8. db.AutoMigrate(allModels...)
  9. for _, m := range allModels {
  10. if !db.Migrator().HasTable(m) {
  11. t.Errorf("Failed to create table for %+v", m)
  12. }
  13. }
  14. }