Files
hfb_sys/backend/internal/modules/csmanualentry/repository_test.go
T
yml2213 ee08e343f4 fix(csmanualentry): 编号搜索支持匹配带后缀的变体编号
- 按编号查询时同时命中本体与 -、/ 后缀变体(如 202611060124-0、202611060070/1)
- 补充变体匹配回归测试
2026-11-06 14:46:07 +08:00

281 lines
9.9 KiB
Go

package csmanualentry
import (
"errors"
"strings"
"testing"
"time"
"hfb_sys/backend/internal/database"
"hfb_sys/backend/internal/model"
"gorm.io/gorm"
)
func newManualEntryTestRepo(t *testing.T) (*Repository, *gorm.DB) {
t.Helper()
db := database.NewTestDB()
if err := database.AutoMigrateIfMissing(db,
&model.AdminUser{},
&model.CSManualEntry{},
&model.CSManualEntryOption{},
); err != nil {
t.Fatalf("迁移测试表失败: %v", err)
}
return NewRepository(db), db
}
func TestNormalizeOptionName(t *testing.T) {
tests := []struct {
name string
input string
want string
wantErr error
}{
{name: "trim", input: " 网易大神 ", want: "网易大神"},
{name: "empty allowed", input: " ", want: ""},
{name: "inner space kept", input: "网易 大神", want: "网易 大神"},
{name: "64 runes", input: strings.Repeat("游", 64), want: strings.Repeat("游", 64)},
{name: "65 runes", input: strings.Repeat("游", 65), wantErr: ErrOptionNameInvalid},
{name: "newline", input: "网易\n大神", wantErr: ErrOptionNameInvalid},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
got, err := normalizeOptionName(tt.input)
if !errors.Is(err, tt.wantErr) {
t.Fatalf("normalizeOptionName(%q) error = %v, want %v", tt.input, err, tt.wantErr)
}
if tt.wantErr == nil && got != tt.want {
t.Fatalf("normalizeOptionName(%q) = %q, want %q", tt.input, got, tt.want)
}
})
}
}
func TestRepositoryOptionCRUDAndEntryPlatformOperator(t *testing.T) {
repo, db := newManualEntryTestRepo(t)
ctx := t.Context()
admin := model.AdminUser{Username: "cs001", Nickname: "客服小张", PasswordHash: "x"}
if err := db.Create(&admin).Error; err != nil {
t.Fatalf("创建管理员失败: %v", err)
}
// 1. 选项 CRUD
platform, err := repo.CreateOption(ctx, OptionTypePlatform, "网易大神", "")
if err != nil {
t.Fatalf("创建平台选项失败: %v", err)
}
// 同类型第二个平台,用于验证重命名撞名(唯一键是 type+name,跨类型不算冲突)
if _, err := repo.CreateOption(ctx, OptionTypePlatform, "交易猫", ""); err != nil {
t.Fatalf("创建第二个平台选项失败: %v", err)
}
operator, err := repo.CreateOption(ctx, OptionTypeOperator, "技术测试", "")
if err != nil {
t.Fatalf("创建操作人选项失败: %v", err)
}
if _, err := repo.CreateOption(ctx, OptionTypePlatform, "网易大神", ""); !errors.Is(err, ErrOptionExists) {
t.Fatalf("重复平台应返回 ErrOptionExists, got %v", err)
}
if _, err := repo.CreateOption(ctx, "unknown", "x", ""); !errors.Is(err, ErrOptionTypeInvalid) {
t.Fatalf("非法类型应返回 ErrOptionTypeInvalid, got %v", err)
}
// 与平台同名的操作人是允许的(不同 type)
if _, err := repo.CreateOption(ctx, OptionTypeOperator, "网易大神", ""); err != nil {
t.Fatalf("跨类型同名应允许: %v", err)
}
if _, err := repo.UpdateOption(ctx, platform.ID, "网易", ""); err != nil {
t.Fatalf("重命名选项失败: %v", err)
}
if _, err := repo.UpdateOption(ctx, platform.ID, "交易猫", ""); !errors.Is(err, ErrOptionExists) {
t.Fatalf("重命名撞同类型已有名应返回 ErrOptionExists, got %v", err)
}
if _, err := repo.UpdateOption(ctx, 999999, "任何", ""); !errors.Is(err, ErrOptionNotFound) {
t.Fatalf("重命名不存在选项应返回 ErrOptionNotFound, got %v", err)
}
options, err := repo.ListAllOptions(ctx)
if err != nil {
t.Fatalf("读取选项列表失败: %v", err)
}
if len(options.Platforms) != 2 || options.Platforms[0].Name != "网易" {
t.Fatalf("平台选项 = %+v, want 2 条且首条为网易", options.Platforms)
}
if len(options.Operators) != 2 {
t.Fatalf("操作人选项 = %+v, want 2 条", options.Operators)
}
// 2. 录单写入平台 / 操作人
entry, err := repo.Create(ctx, admin.ID, "客服小张", "cs001", "202610130001", EntryGroupCollect, "网易", "技术测试", "备注1")
if err != nil {
t.Fatalf("录单失败: %v", err)
}
if entry.Platform != "网易" || entry.Operator != "技术测试" {
t.Fatalf("录单平台/操作人 = %q/%q", entry.Platform, entry.Operator)
}
// 修正录单的平台与操作人
updated, err := repo.Update(ctx, entry.ID, "202610130001", EntryGroupCollect, "网易", "客服小李", "备注2")
if err != nil {
t.Fatalf("修正录单失败: %v", err)
}
if updated.Operator != "客服小李" || updated.Remark != "备注2" {
t.Fatalf("修正后 = %+v", updated)
}
// 3. 关键字命中平台 / 操作人
list, err := repo.List(ctx, ListQuery{Keyword: "客服小李", Page: 1, PageSize: 20})
if err != nil {
t.Fatalf("按操作人查询失败: %v", err)
}
if list.Total != 1 {
t.Fatalf("按操作人查询命中 %d 条, want 1", list.Total)
}
list, err = repo.List(ctx, ListQuery{Keyword: "网易", Page: 1, PageSize: 20})
if err != nil {
t.Fatalf("按平台查询失败: %v", err)
}
if list.Total != 1 {
t.Fatalf("按平台查询命中 %d 条, want 1", list.Total)
}
// 4. 删除选项不影响历史录单(记录里是名称快照)
if err := repo.DeleteOption(ctx, operator.ID); err != nil {
t.Fatalf("删除操作人选项失败: %v", err)
}
if err := repo.DeleteOption(ctx, operator.ID); !errors.Is(err, ErrOptionNotFound) {
t.Fatalf("重复删除应返回 ErrOptionNotFound, got %v", err)
}
after, err := repo.FindByID(ctx, entry.ID)
if err != nil {
t.Fatalf("删除选项后读取录单失败: %v", err)
}
if after.Operator != "客服小李" {
t.Fatalf("删除选项后历史录单操作人 = %q, want 客服小李", after.Operator)
}
}
func TestRepositoryEntryGroup(t *testing.T) {
repo, db := newManualEntryTestRepo(t)
ctx := t.Context()
admin := model.AdminUser{Username: "cs002", Nickname: "客服小李", PasswordHash: "x"}
if err := db.Create(&admin).Error; err != nil {
t.Fatalf("创建管理员失败: %v", err)
}
// 同一编号:收号组登记一次
collect, err := repo.Create(ctx, admin.ID, "客服小李", "cs002", "202611050473/0", EntryGroupCollect, "", "", "")
if err != nil {
t.Fatalf("收号组录单失败: %v", err)
}
if collect.EntryGroup != EntryGroupCollect {
t.Fatalf("收号组别 = %q, want collect", collect.EntryGroup)
}
// 同组重复登记应冲突
if _, err := repo.Create(ctx, admin.ID, "客服小李", "cs002", "202611050473/0", EntryGroupCollect, "", "", ""); !errors.Is(err, ErrEntryNoExists) {
t.Fatalf("同组重复编号应返回 ErrEntryNoExists, got %v", err)
}
// 出号组登记同一编号应成功(编号绑定两个客服/两个组)
sell, err := repo.Create(ctx, admin.ID, "客服小李", "cs002", "202611050473/0", EntryGroupSell, "", "", "")
if err != nil {
t.Fatalf("出号组录单失败: %v", err)
}
if sell.EntryGroup != EntryGroupSell {
t.Fatalf("出号组别 = %q, want sell", sell.EntryGroup)
}
// 按组别筛选
list, err := repo.List(ctx, ListQuery{EntryNo: "202611050473/0", EntryGroup: EntryGroupSell, Page: 1, PageSize: 20})
if err != nil {
t.Fatalf("按组别查询失败: %v", err)
}
if list.Total != 1 || list.Items[0].EntryGroup != EntryGroupSell {
t.Fatalf("按出号组查询 = %+v, want 1 条 sell 记录", list.Items)
}
// 今日统计按组分列
stats, err := repo.TodayAgentStats(ctx, time.Now())
if err != nil {
t.Fatalf("今日统计失败: %v", err)
}
if len(stats) != 1 {
t.Fatalf("今日统计客服数 = %d, want 1", len(stats))
}
if stats[0].Count != 2 || stats[0].CollectCount != 1 || stats[0].SellCount != 1 {
t.Fatalf("今日统计 = %+v, want 总量2 收号1 出号1", stats[0])
}
// 修正组别:出号改成收号后与原收号记录冲突
if _, err := repo.Update(ctx, sell.ID, "202611050473/0", EntryGroupCollect, "", "", ""); !errors.Is(err, ErrEntryNoExists) {
t.Fatalf("改成冲突组别应返回 ErrEntryNoExists, got %v", err)
}
// 操作人选项组别标记
operator, err := repo.CreateOption(ctx, OptionTypeOperator, "测试操作", EntryGroupCollect)
if err != nil {
t.Fatalf("创建操作人选项失败: %v", err)
}
if operator.GroupTag != EntryGroupCollect {
t.Fatalf("操作人组别标记 = %q, want collect", operator.GroupTag)
}
updated, err := repo.UpdateOption(ctx, operator.ID, "测试操作", EntryGroupSell)
if err != nil {
t.Fatalf("更新操作人组别失败: %v", err)
}
if updated.GroupTag != EntryGroupSell {
t.Fatalf("更新后组别标记 = %q, want sell", updated.GroupTag)
}
}
func TestRepositoryListEntryNoWithSuffixVariants(t *testing.T) {
repo, db := newManualEntryTestRepo(t)
ctx := t.Context()
admin := model.AdminUser{Username: "cs003", Nickname: "客服小王", PasswordHash: "x"}
if err := db.Create(&admin).Error; err != nil {
t.Fatalf("创建管理员失败: %v", err)
}
// 基础编号 + 带 - / 后缀的变体
numbers := []string{"202611060124", "202611060124-0", "202611060124/1", "202611060157", "202611060124-0-2"}
for _, no := range numbers {
if _, err := repo.Create(ctx, admin.ID, "客服小王", "cs003", no, EntryGroupCollect, "", "", ""); err != nil {
t.Fatalf("录入 %s 失败: %v", no, err)
}
}
// 按基础编号搜索:应命中本体、-0、/1、-0-2,不命中无关编号
list, err := repo.List(ctx, ListQuery{EntryNo: "202611060124", Page: 1, PageSize: 20})
if err != nil {
t.Fatalf("按编号搜索失败: %v", err)
}
if list.Total != 4 {
t.Fatalf("按编号搜索命中 %d 条, want 4", list.Total)
}
got := make(map[string]bool)
for _, item := range list.Items {
got[item.EntryNo] = true
}
for _, want := range []string{"202611060124", "202611060124-0", "202611060124/1", "202611060124-0-2"} {
if !got[want] {
t.Fatalf("按编号搜索结果缺少 %s, got %v", want, got)
}
}
if got["202611060157"] {
t.Fatalf("不应命中无关编号 202611060157")
}
// 按后缀变体精确搜索仍可用
list, err = repo.List(ctx, ListQuery{EntryNo: "202611060124-0", Page: 1, PageSize: 20})
if err != nil {
t.Fatalf("按带后缀编号搜索失败: %v", err)
}
if list.Total != 2 {
t.Fatalf("按带后缀编号搜索命中 %d 条, want 2 (本体变体 -0 与 -0-2)", list.Total)
}
}