Test Save with specfied table name

This commit is contained in:
Jinzhu 2019-06-11 17:30:14 +08:00
parent c44c6027fb
commit 153ce22c99
2 changed files with 13 additions and 4 deletions

View File

@ -466,7 +466,7 @@ func (s *DB) Save(value interface{}) *DB {
if !scope.PrimaryKeyZero() { if !scope.PrimaryKeyZero() {
newDB := scope.callCallbacks(s.parent.callbacks.updates).db newDB := scope.callCallbacks(s.parent.callbacks.updates).db
if newDB.Error == nil && newDB.RowsAffected == 0 { if newDB.Error == nil && newDB.RowsAffected == 0 {
return s.FirstOrCreate(value) return s.New().Table(scope.TableName()).FirstOrCreate(value)
} }
return newDB return newDB
} }

View File

@ -44,13 +44,13 @@ func OpenTestConnection() (db *gorm.DB, err error) {
case "mysql": case "mysql":
fmt.Println("testing mysql...") fmt.Println("testing mysql...")
if dbDSN == "" { if dbDSN == "" {
dbDSN = "gorm:gorm@tcp(localhost:9910)/gorm?charset=utf8&parseTime=True" dbDSN = "gorm:gorm@tcp(localhost:3306)/gorm?charset=utf8&parseTime=True"
} }
db, err = gorm.Open("mysql", dbDSN) db, err = gorm.Open("mysql", dbDSN)
case "postgres": case "postgres":
fmt.Println("testing postgres...") fmt.Println("testing postgres...")
if dbDSN == "" { if dbDSN == "" {
dbDSN = "user=gorm password=gorm DB.name=gorm port=9920 sslmode=disable" dbDSN = "user=gorm password=gorm DB.name=gorm port=5432 sslmode=disable"
} }
db, err = gorm.Open("postgres", dbDSN) db, err = gorm.Open("postgres", dbDSN)
case "mssql": case "mssql":
@ -61,7 +61,7 @@ func OpenTestConnection() (db *gorm.DB, err error) {
// sp_changedbowner 'gorm'; // sp_changedbowner 'gorm';
fmt.Println("testing mssql...") fmt.Println("testing mssql...")
if dbDSN == "" { if dbDSN == "" {
dbDSN = "sqlserver://gorm:LoremIpsum86@localhost:9930?database=gorm" dbDSN = "sqlserver://gorm:LoremIpsum86@localhost:1433?database=gorm"
} }
db, err = gorm.Open("mssql", dbDSN) db, err = gorm.Open("mssql", dbDSN)
default: default:
@ -178,6 +178,15 @@ func TestSetTable(t *testing.T) {
t.Errorf("Query from specified table") t.Errorf("Query from specified table")
} }
var user User
DB.Table("deleted_users").First(&user, "name = ?", "DeletedUser")
user.Age = 20
DB.Table("deleted_users").Save(&user)
if DB.Table("deleted_users").First(&user, "name = ? AND age = ?", "DeletedUser", 20).RecordNotFound() {
t.Errorf("Failed to found updated user")
}
DB.Save(getPreparedUser("normal_user", "reset_table")) DB.Save(getPreparedUser("normal_user", "reset_table"))
DB.Table("deleted_users").Save(getPreparedUser("deleted_user", "reset_table")) DB.Table("deleted_users").Save(getPreparedUser("deleted_user", "reset_table"))
var user1, user2, user3 User var user1, user2, user3 User