Merge branch 'RichardKnop-bugfix/hastable-string'

This commit is contained in:
Jinzhu 2016-02-14 14:14:56 +08:00
commit 3c9d2b1af6
2 changed files with 16 additions and 0 deletions

View File

@ -165,12 +165,24 @@ func TestHasTable(t *testing.T) {
Stuff string
}
DB.DropTable(&Foo{})
// Table should not exist at this point, HasTable should return false
if ok := DB.HasTable("foos"); ok {
t.Errorf("Table should not exist, but does")
}
if ok := DB.HasTable(&Foo{}); ok {
t.Errorf("Table should not exist, but does")
}
// We create the table
if err := DB.CreateTable(&Foo{}).Error; err != nil {
t.Errorf("Table should be created")
}
// And now it should exits, and HasTable should return true
if ok := DB.HasTable("foos"); !ok {
t.Errorf("Table should exist, but HasTable informs it does not")
}
if ok := DB.HasTable(&Foo{}); !ok {
t.Errorf("Table should exist, but HasTable informs it does not")
}

View File

@ -267,6 +267,10 @@ type dbTabler interface {
// TableName get table name
func (scope *Scope) TableName() string {
if strTableName, ok := scope.Value.(string); ok {
return strTableName
}
if scope.Search != nil && len(scope.Search.tableName) > 0 {
return scope.Search.tableName
}