diff --git a/main_test.go b/main_test.go index 5058590c..1aac506d 100644 --- a/main_test.go +++ b/main_test.go @@ -1264,12 +1264,22 @@ func TestCountWithQueryOption(t *testing.T) { func TestQueryHint(t *testing.T) { db := DB.New() - var count int + db.Delete(User{}) + defer db.Delete(User{}) - err := db.Set("gorm:query_hint", "/*master*/").Raw("select ?", "1").Count(&count).Error + DB.Create(&User{Name: "user1"}) + + var count int + err := db.Model(User{}).Select("users.id"). + Set("gorm:query_hint", "/*master*/"). + Count(&count).Error if err != nil { - t.Error("Unexpected error on query_hint") + t.Error("Unexpected error on query count with query_option") + } + + if count != 1 { + t.Error("Unexpected result on query count with query_option") } }