test(sql_builder_test): add test for CreateInBatches sql statment

This commit is contained in:
Anton Kichaev 2023-09-02 12:38:59 +07:00
parent 28331bf198
commit df5d184baa

View File

@ -453,6 +453,28 @@ func TestToSQL(t *testing.T) {
}) })
}) })
assertEqualSQL(t, `SELECT * FROM users ORDER BY id DESC`, sql) assertEqualSQL(t, `SELECT * FROM users ORDER BY id DESC`, sql)
// create batch, models are equal batch size
bs := 2
m := gorm.Model{
CreatedAt: date,
UpdatedAt: date,
}
sql = DB.ToSQL(func(tx *gorm.DB) *gorm.DB {
return tx.Model(&User{}).
CreateInBatches(
[]User{
{
Name: "foo", Age: 20, Model: m,
},
{
Name: "bar", Age: 30, Model: m,
},
},
bs,
)
})
assertEqualSQL(t, `INSERT INTO "users" ("created_at","updated_at","deleted_at","name","age","birthday","company_id","manager_id","active") VALUES ('2021-10-18 00:00:00','2021-10-18 00:00:00',NULL,'foo',20,NULL,NULL,NULL,false),('2021-10-18 00:00:00','2021-10-18 00:00:00',NULL,'bar',30,NULL,NULL,NULL,false) RETURNING "id";`, sql)
} }
// assertEqualSQL for assert that the sql is equal, this method will ignore quote, and dialect specials. // assertEqualSQL for assert that the sql is equal, this method will ignore quote, and dialect specials.
@ -469,7 +491,7 @@ func assertEqualSQL(t *testing.T, expected string, actually string) {
expected = updatedAtRe.ReplaceAllString(expected, `"updated_at"=?`) expected = updatedAtRe.ReplaceAllString(expected, `"updated_at"=?`)
// ignore RETURNING "id" (only in PostgreSQL) // ignore RETURNING "id" (only in PostgreSQL)
returningRe := regexp.MustCompile(`(?i)RETURNING "id"`) returningRe := regexp.MustCompile(`\s?(?i)RETURNING "id"`)
actually = returningRe.ReplaceAllString(actually, ``) actually = returningRe.ReplaceAllString(actually, ``)
expected = returningRe.ReplaceAllString(expected, ``) expected = returningRe.ReplaceAllString(expected, ``)