Skip to content

Commit 80ec777

Browse files
jawad-arb123jawad-arb123
committed
fix the Query Tests
1 parent 71b193d commit 80ec777

File tree

1 file changed

+26
-26
lines changed

1 file changed

+26
-26
lines changed

‎tests/query_test.go‎

Lines changed: 26 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ func TestFind(t *testing.T) {
7171

7272
t.Run("First", func(t *testing.T) {
7373
var first User
74-
if err := DB.Where("name = ?", "find").First(&first).Error; err != nil {
74+
if err := DB.Where("\"name\" = ?", "find").First(&first).Error; err != nil {
7575
t.Errorf("errors happened when query first: %v", err)
7676
} else {
7777
CheckUser(t, first, users[0])
@@ -80,15 +80,15 @@ func TestFind(t *testing.T) {
8080

8181
t.Run("Last", func(t *testing.T) {
8282
var last User
83-
if err := DB.Where("name = ?", "find").Last(&last).Error; err != nil {
83+
if err := DB.Where("\"name\" = ?", "find").Last(&last).Error; err != nil {
8484
t.Errorf("errors happened when query last: %v", err)
8585
} else {
8686
CheckUser(t, last, users[2])
8787
}
8888
})
8989

9090
var all []User
91-
if err := DB.Where("name = ?", "find").Find(&all).Error; err != nil || len(all) != 3 {
91+
if err := DB.Where("\"name\" = ?", "find").Find(&all).Error; err != nil || len(all) != 3 {
9292
t.Errorf("errors happened when query find: %v, length: %v", err, len(all))
9393
} else {
9494
for idx, user := range users {
@@ -100,7 +100,7 @@ func TestFind(t *testing.T) {
100100

101101
t.Run("FirstMap", func(t *testing.T) {
102102
first := map[string]interface{}{}
103-
if err := DB.Model(&User{}).Where("name = ?", "find").First(first).Error; err != nil {
103+
if err := DB.Model(&User{}).Where("\"name\" = ?", "find").First(first).Error; err != nil {
104104
t.Errorf("errors happened when query first: %v", err)
105105
} else {
106106
for _, name := range []string{"Name", "Age", "Birthday"} {
@@ -131,7 +131,7 @@ func TestFind(t *testing.T) {
131131

132132
t.Run("FirstMapWithTable", func(t *testing.T) {
133133
first := map[string]interface{}{}
134-
if err := DB.Table("users").Where("name = ?", "find").Find(first).Error; err != nil {
134+
if err := DB.Table("users").Where("\"name\" = ?", "find").Find(first).Error; err != nil {
135135
t.Errorf("errors happened when query first: %v", err)
136136
} else {
137137
for _, name := range []string{"Name", "Age", "Birthday"} {
@@ -163,7 +163,7 @@ func TestFind(t *testing.T) {
163163

164164
t.Run("FirstPtrMap", func(t *testing.T) {
165165
first := map[string]interface{}{}
166-
if err := DB.Model(&User{}).Where("name = ?", "find").First(&first).Error; err != nil {
166+
if err := DB.Model(&User{}).Where("\"name\" = ?", "find").First(&first).Error; err != nil {
167167
t.Errorf("errors happened when query first: %v", err)
168168
} else {
169169
for _, name := range []string{"Name", "Age", "Birthday"} {
@@ -178,7 +178,7 @@ func TestFind(t *testing.T) {
178178

179179
t.Run("FirstSliceOfMap", func(t *testing.T) {
180180
allMap := []map[string]interface{}{}
181-
if err := DB.Model(&User{}).Where("name = ?", "find").Find(&allMap).Error; err != nil {
181+
if err := DB.Model(&User{}).Where("\"name\" = ?", "find").Find(&allMap).Error; err != nil {
182182
t.Errorf("errors happened when query find: %v", err)
183183
} else {
184184
for idx, user := range users {
@@ -213,7 +213,7 @@ func TestFind(t *testing.T) {
213213

214214
t.Run("FindSliceOfMapWithTable", func(t *testing.T) {
215215
allMap := []map[string]interface{}{}
216-
if err := DB.Table("users").Where("name = ?", "find").Find(&allMap).Error; err != nil {
216+
if err := DB.Table("users").Where("\"name\" = ?", "find").Find(&allMap).Error; err != nil {
217217
t.Errorf("errors happened when query find: %v", err)
218218
} else {
219219
for idx, user := range users {
@@ -248,7 +248,7 @@ func TestFind(t *testing.T) {
248248
})
249249

250250
var models []User
251-
if err := DB.Where("name in (?)", []string{"find"}).Find(&models).Error; err != nil || len(models) != 3 {
251+
if err := DB.Where("\"name\" in (?)", []string{"find"}).Find(&models).Error; err != nil || len(models) != 3 {
252252
t.Errorf("errors happened when query find with in clause: %v, length: %v", err, len(models))
253253
} else {
254254
for idx, user := range users {
@@ -260,7 +260,7 @@ func TestFind(t *testing.T) {
260260

261261
// test array
262262
var models2 [3]User
263-
if err := DB.Where("name in (?)", []string{"find"}).Find(&models2).Error; err != nil {
263+
if err := DB.Where("\"name\" in (?)", []string{"find"}).Find(&models2).Error; err != nil {
264264
t.Errorf("errors happened when query find with in clause: %v, length: %v", err, len(models2))
265265
} else {
266266
for idx, user := range users {
@@ -272,7 +272,7 @@ func TestFind(t *testing.T) {
272272

273273
// test smaller array
274274
var models3 [2]User
275-
if err := DB.Where("name in (?)", []string{"find"}).Find(&models3).Error; err != nil {
275+
if err := DB.Where("\"name\" in (?)", []string{"find"}).Find(&models3).Error; err != nil {
276276
t.Errorf("errors happened when query find with in clause: %v, length: %v", err, len(models3))
277277
} else {
278278
for idx, user := range users[:2] {
@@ -283,7 +283,7 @@ func TestFind(t *testing.T) {
283283
}
284284

285285
var none []User
286-
if err := DB.Where("name in (?)", []string{}).Find(&none).Error; err != nil || len(none) != 0 {
286+
if err := DB.Where("\"name\" in (?)", []string{}).Find(&none).Error; err != nil || len(none) != 0 {
287287
t.Errorf("errors happened when query find with in clause and zero length parameter: %v, length: %v", err, len(none))
288288
}
289289
}
@@ -323,7 +323,7 @@ func TestFindInBatches(t *testing.T) {
323323
totalBatch int
324324
)
325325

326-
if result := DB.Table("\"users\" u").Where("\"name\" = ?", users[0].Name).FindInBatches(&results, 2, func(tx *gorm.DB, batch int) error {
326+
if result := DB.Table("users").Where("\"name\" = ?", users[0].Name).FindInBatches(&results, 2, func(tx *gorm.DB, batch int) error {
327327
totalBatch += batch
328328

329329
if tx.RowsAffected != 2 {
@@ -380,7 +380,7 @@ func TestFindInBatchesWithOffsetLimit(t *testing.T) {
380380
)
381381

382382
// offset limit
383-
if result := DB.Offset(3).Limit(5).Where("name = ?", users[0].Name).FindInBatches(&sub, 2, func(tx *gorm.DB, batch int) error {
383+
if result := DB.Offset(3).Limit(5).Where("\"name\" = ?", users[0].Name).FindInBatches(&sub, 2, func(tx *gorm.DB, batch int) error {
384384
results = append(results, sub...)
385385
lastBatch = batch
386386
return nil
@@ -398,22 +398,22 @@ func TestFindInBatchesWithOffsetLimit(t *testing.T) {
398398

399399
var sub1 []User
400400
// limit < batchSize
401-
if result := DB.Limit(5).Where("name = ?", users[0].Name).FindInBatches(&sub1, 10, func(tx *gorm.DB, batch int) error {
401+
if result := DB.Limit(5).Where("\"name\" = ?", users[0].Name).FindInBatches(&sub1, 10, func(tx *gorm.DB, batch int) error {
402402
return nil
403403
}); result.Error != nil || result.RowsAffected != 5 {
404404
t.Errorf("Failed to batch find, got error %v, rows affected: %v", result.Error, result.RowsAffected)
405405
}
406406

407407
var sub2 []User
408408
// only offset
409-
if result := DB.Offset(3).Where("name = ?", users[0].Name).FindInBatches(&sub2, 2, func(tx *gorm.DB, batch int) error {
409+
if result := DB.Offset(3).Where("\"name\" = ?", users[0].Name).FindInBatches(&sub2, 2, func(tx *gorm.DB, batch int) error {
410410
return nil
411411
}); result.Error != nil || result.RowsAffected != 7 {
412412
t.Errorf("Failed to batch find, got error %v, rows affected: %v", result.Error, result.RowsAffected)
413413
}
414414

415415
var sub3 []User
416-
if result := DB.Limit(4).Where("name = ?", users[0].Name).FindInBatches(&sub3, 2, func(tx *gorm.DB, batch int) error {
416+
if result := DB.Limit(4).Where("\"name\" = ?", users[0].Name).FindInBatches(&sub3, 2, func(tx *gorm.DB, batch int) error {
417417
return nil
418418
}); result.Error != nil || result.RowsAffected != 4 {
419419
t.Errorf("Failed to batch find, got error %v, rows affected: %v", result.Error, result.RowsAffected)
@@ -437,7 +437,7 @@ func TestFindInBatchesWithError(t *testing.T) {
437437
totalBatch int
438438
)
439439

440-
if result := DB.Table("wrong_table").Where("name = ?", users[0].Name).FindInBatches(&results, 2, func(tx *gorm.DB, batch int) error {
440+
if result := DB.Table("wrong_table").Where("\"name\" = ?", users[0].Name).FindInBatches(&results, 2, func(tx *gorm.DB, batch int) error {
441441
totalBatch += batch
442442
return nil
443443
}); result.Error == nil || result.RowsAffected > 0 {
@@ -447,7 +447,7 @@ func TestFindInBatchesWithError(t *testing.T) {
447447
t.Fatalf("incorrect total batch, expected: %v, got: %v", 0, totalBatch)
448448
}
449449

450-
if result := DB.Omit("id").Where("name = ?", users[0].Name).FindInBatches(&results, 2, func(tx *gorm.DB, batch int) error {
450+
if result := DB.Omit("id").Where("\"name\" = ?", users[0].Name).FindInBatches(&results, 2, func(tx *gorm.DB, batch int) error {
451451
totalBatch += batch
452452
return nil
453453
}); result.Error != gorm.ErrPrimaryKeyRequired {
@@ -863,7 +863,7 @@ func TestOmit(t *testing.T) {
863863
DB.Save(&user)
864864

865865
var result User
866-
DB.Where("name = ?", user.Name).Omit("name").Find(&result)
866+
DB.Where("\"name\"= ?", user.Name).Omit("name").Find(&result)
867867
if result.ID == 0 {
868868
t.Errorf("Should not have ID because only selected name, %+v", result.ID)
869869
}
@@ -878,7 +878,7 @@ func TestOmitWithAllFields(t *testing.T) {
878878
DB.Save(&user)
879879

880880
var userResult User
881-
DB.Session(&gorm.Session{QueryFields: true}).Where("users.name = ?", user.Name).Omit("name").Find(&userResult)
881+
DB.Session(&gorm.Session{QueryFields: true}).Where("\"users\".\"name\" = ?", user.Name).Omit("name").Find(&userResult)
882882
if userResult.ID == 0 {
883883
t.Errorf("Should not have ID because only selected name, %+v", userResult.ID)
884884
}
@@ -907,7 +907,7 @@ func TestMapColumns(t *testing.T) {
907907
Age uint
908908
}
909909
var res result
910-
DB.Table("users").Where("name = ?", user.Name).MapColumns(map[string]string{"name": "nickname"}).Scan(&res)
910+
DB.Table("users").Where("\"name\" = ?", user.Name).MapColumns(map[string]string{"name": "nickname"}).Scan(&res)
911911
if res.Nickname != user.Name {
912912
t.Errorf("Expected res.Nickname to be %s, but got %s", user.Name, res.Nickname)
913913
}
@@ -1304,7 +1304,7 @@ func TestScanNullValue(t *testing.T) {
13041304
}
13051305

13061306
var result User
1307-
if err := DB.First(&result, "id = ?", user.ID).Error; err != nil {
1307+
if err := DB.First(&result, "\"id\" = ?", user.ID).Error; err != nil {
13081308
t.Fatalf("failed to query struct data with null age, got error %v", err)
13091309
}
13101310

@@ -1322,7 +1322,7 @@ func TestScanNullValue(t *testing.T) {
13221322
}
13231323

13241324
var results []User
1325-
if err := DB.Find(&results, "name like ?", "scan_null_value_for_slice%").Error; err != nil {
1325+
if err := DB.Find(&results, "\"name\" like ?", "scan_null_value_for_slice%").Error; err != nil {
13261326
t.Fatalf("failed to query slice data with null age, got error %v", err)
13271327
}
13281328
}
@@ -1365,15 +1365,15 @@ func TestQueryScannerWithSingleColumn(t *testing.T) {
13651365
DB.Create(&user)
13661366

13671367
var result1 DoubleInt64
1368-
if err := DB.Model(&User{}).Where("name LIKE ?", "scanner_raw_%").Limit(1).Pluck(
1368+
if err := DB.Model(&User{}).Where("\"name\" LIKE ?", "scanner_raw_%").Limit(1).Pluck(
13691369
"age", &result1).Error; err != nil {
13701370
t.Errorf("Failed, got error: %v", err)
13711371
}
13721372

13731373
tests.AssertEqual(t, result1.data, 20)
13741374

13751375
var result2 DoubleInt64
1376-
if err := DB.Model(&User{}).Where("name LIKE ?", "scanner_raw_%").Limit(1).Select(
1376+
if err := DB.Model(&User{}).Where("\"name\" LIKE ?", "scanner_raw_%").Limit(1).Select(
13771377
"age").Scan(&result2).Error; err != nil {
13781378
t.Errorf("Failed, got error: %v", err)
13791379
}

0 commit comments

Comments
 (0)