129 lines
3.6 KiB
Go
129 lines
3.6 KiB
Go
package core
|
||
|
||
import (
|
||
"fmt"
|
||
"testing"
|
||
)
|
||
|
||
// TestQueryBuilder_Omit 测试 Omit 方法
|
||
func TestQueryBuilder_Omit(t *testing.T) {
|
||
// 创建测试模型
|
||
type UserModel struct {
|
||
ID int64 `json:"id" db:"id"`
|
||
Name string `json:"name" db:"name"`
|
||
Email string `json:"email" db:"email"`
|
||
Password string `json:"password" db:"password"`
|
||
Status int `json:"status" db:"status"`
|
||
}
|
||
|
||
// 创建 Database 实例（使用 nil 连接，只测试 SQL 生成）
|
||
db := &Database{}
|
||
|
||
t.Run("排除单个字段", func(t *testing.T) {
|
||
qb := db.Model(&UserModel{}).Omit("password").(*QueryBuilder)
|
||
sql, args := qb.BuildSelect()
|
||
|
||
fmt.Printf("排除单个字段 SQL: %s\n", sql)
|
||
fmt.Printf("参数：%v\n", args)
|
||
|
||
// 验证 SQL 不包含 password 字段
|
||
if containsString(sql, "password") {
|
||
t.Errorf("SQL 不应该包含 password 字段：%s", sql)
|
||
}
|
||
|
||
// 验证 SQL 包含其他字段
|
||
expectedFields := []string{"id", "name", "email", "status"}
|
||
for _, field := range expectedFields {
|
||
if !containsString(sql, field) {
|
||
t.Errorf("SQL 应该包含字段 %s: %s", field, sql)
|
||
}
|
||
}
|
||
})
|
||
|
||
t.Run("排除多个字段", func(t *testing.T) {
|
||
qb := db.Model(&UserModel{}).Omit("password", "status").(*QueryBuilder)
|
||
sql, args := qb.BuildSelect()
|
||
|
||
fmt.Printf("排除多个字段 SQL: %s\n", sql)
|
||
fmt.Printf("参数：%v\n", args)
|
||
|
||
// 验证 SQL 不包含 password 和 status 字段
|
||
if containsString(sql, "password") {
|
||
t.Errorf("SQL 不应该包含 password 字段：%s", sql)
|
||
}
|
||
if containsString(sql, "status") {
|
||
t.Errorf("SQL 不应该包含 status 字段：%s", sql)
|
||
}
|
||
|
||
// 验证 SQL 包含其他字段
|
||
expectedFields := []string{"id", "name", "email"}
|
||
for _, field := range expectedFields {
|
||
if !containsString(sql, field) {
|
||
t.Errorf("SQL 应该包含字段 %s: %s", field, sql)
|
||
}
|
||
}
|
||
})
|
||
|
||
t.Run("Omit 与 Select 优先级 - Select 优先", func(t *testing.T) {
|
||
qb := db.Model(&UserModel{}).Select("id", "name").Omit("password").(*QueryBuilder)
|
||
sql, args := qb.BuildSelect()
|
||
|
||
fmt.Printf("Select 优先 SQL: %s\n", sql)
|
||
fmt.Printf("参数：%v\n", args)
|
||
|
||
// 当同时使用 Select 和 Omit 时，Select 优先
|
||
expectedFields := []string{"id", "name"}
|
||
for _, field := range expectedFields {
|
||
if !containsString(sql, field) {
|
||
t.Errorf("SQL 应该包含字段 %s: %s", field, sql)
|
||
}
|
||
}
|
||
})
|
||
|
||
t.Run("链式调用 Omit", func(t *testing.T) {
|
||
qb := db.Model(&UserModel{}).Omit("password").Omit("status").(*QueryBuilder)
|
||
sql, args := qb.BuildSelect()
|
||
|
||
fmt.Printf("链式调用 Omit SQL: %s\n", sql)
|
||
fmt.Printf("参数：%v\n", args)
|
||
|
||
// 验证 SQL 不包含 password 和 status 字段
|
||
if containsString(sql, "password") {
|
||
t.Errorf("SQL 不应该包含 password 字段：%s", sql)
|
||
}
|
||
if containsString(sql, "status") {
|
||
t.Errorf("SQL 不应该包含 status 字段：%s", sql)
|
||
}
|
||
})
|
||
|
||
t.Run("不设置 Omit - 默认行为", func(t *testing.T) {
|
||
qb := db.Model(&UserModel{}).(*QueryBuilder)
|
||
sql, args := qb.BuildSelect()
|
||
|
||
fmt.Printf("默认行为 SQL: %s\n", sql)
|
||
fmt.Printf("参数：%v\n", args)
|
||
|
||
// 默认应该查询所有字段（使用 *）
|
||
if sql != "SELECT * FROM user_model" {
|
||
t.Errorf("默认应该使用 SELECT *: %s", sql)
|
||
}
|
||
})
|
||
}
|
||
|
||
// containsString 检查字符串是否包含子串
|
||
func containsString(s, substr string) bool {
|
||
return len(s) >= len(substr) && (s == substr ||
|
||
s[:len(substr)] == substr ||
|
||
s[len(s)-len(substr):] == substr ||
|
||
findSubstring(s, substr))
|
||
}
|
||
|
||
func findSubstring(s, substr string) bool {
|
||
for i := 0; i <= len(s)-len(substr); i++ {
|
||
if s[i:i+len(substr)] == substr {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|