forked from mirror/gorm
Test Save with specfied table name
This commit is contained in:
parent
c44c6027fb
commit
153ce22c99
2
main.go
2
main.go
|
@ -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
|
||||||
}
|
}
|
||||||
|
|
15
main_test.go
15
main_test.go
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue