diff --git a/main_test.go b/main_test.go index e0b39e9f..0555ef67 100644 --- a/main_test.go +++ b/main_test.go @@ -4,13 +4,14 @@ import ( "database/sql" "database/sql/driver" "fmt" + "strconv" + testdb "github.com/erikstmartin/go-testdb" _ "github.com/go-sql-driver/mysql" "github.com/jinzhu/gorm" "github.com/jinzhu/now" _ "github.com/lib/pq" _ "github.com/mattn/go-sqlite3" - "strconv" "os" "testing" @@ -127,6 +128,22 @@ func (c Cart) TableName() string { return "shopping_cart" } +func TestHasTable(t *testing.T) { + type Foo struct { + Id int + Stuff string + } + if table_ok := db.HasTable(&Foo{}); table_ok { + t.Errorf("Table should not exist, but does") + } + if err := db.CreateTable(&Foo{}).Error; err != nil { + t.Errorf("Table should be created") + } + if table_ok := db.HasTable(&Foo{}); !table_ok { + t.Errorf("Table should exist, but HasTable informs it does not") + } +} + func TestTableName(t *testing.T) { db := db.Model("") if db.NewScope(Order{}).TableName() != "orders" {