72 lines
2.1 KiB
Go
72 lines
2.1 KiB
Go
// 功能描述：验证三个管理后台的工作人员写入入口均接入公共错误转换链路。
|
||
// 版本：v1.0
|
||
package common
|
||
|
||
import (
|
||
"go/ast"
|
||
"go/parser"
|
||
"go/token"
|
||
"path/filepath"
|
||
"runtime"
|
||
"testing"
|
||
)
|
||
|
||
// TestStaffHandlersUseWriteErrorConversion 防止任一后台重新绕过工作人员错误转换器。
|
||
// 参数：t 为 Go 测试上下文。
|
||
// 返回值：无。
|
||
func TestStaffHandlersUseWriteErrorConversion(t *testing.T) {
|
||
_, currentFile, _, ok := runtime.Caller(0)
|
||
if !ok {
|
||
t.Fatal("无法定位工作人员接线测试文件")
|
||
}
|
||
logicDirectory := filepath.Dir(filepath.Dir(currentFile))
|
||
targets := []struct {
|
||
name string
|
||
path string
|
||
}{
|
||
{name: "平台总后台", path: filepath.Join(logicDirectory, "platform", "staff", "staff.go")},
|
||
{name: "气站后台", path: filepath.Join(logicDirectory, "gas", "staff.go")},
|
||
{name: "配送点后台", path: filepath.Join(logicDirectory, "delivery", "staff.go")},
|
||
}
|
||
|
||
for _, target := range targets {
|
||
t.Run(target.name, func(t *testing.T) {
|
||
calls := commonCallsInFile(t, target.path)
|
||
if calls["CreateStaffRecord"] == 0 {
|
||
t.Fatalf("%s工作人员创建入口未接入 CreateStaffRecord", target.name)
|
||
}
|
||
if calls["UpdateAllowedByIdentityWithError"] == 0 {
|
||
t.Fatalf("%s工作人员编辑入口未接入错误转换更新函数", target.name)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
// commonCallsInFile 统计源文件中调用 common 包函数的次数。
|
||
// 参数：t 为测试上下文，path 为待检查的 Go 源文件。
|
||
// 返回值：以函数名为键、调用次数为值的映射。
|
||
func commonCallsInFile(t *testing.T, path string) map[string]int {
|
||
t.Helper()
|
||
file, err := parser.ParseFile(token.NewFileSet(), path, nil, 0)
|
||
if err != nil {
|
||
t.Fatalf("解析工作人员处理器失败：%v", err)
|
||
}
|
||
calls := map[string]int{}
|
||
ast.Inspect(file, func(node ast.Node) bool {
|
||
call, ok := node.(*ast.CallExpr)
|
||
if !ok {
|
||
return true
|
||
}
|
||
selector, ok := call.Fun.(*ast.SelectorExpr)
|
||
if !ok {
|
||
return true
|
||
}
|
||
packageName, ok := selector.X.(*ast.Ident)
|
||
if ok && packageName.Name == "common" {
|
||
calls[selector.Sel.Name]++
|
||
}
|
||
return true
|
||
})
|
||
return calls
|
||
}
|