多次调用，顺序返回不同结果-北京尧图网络

gomonkey.ApplyMethodSeq()高频实用方法处理「结构体方法被多次调用需要返回不同结果」的核心方法完美解决方法第 1 次调用返回 A、第 2 次返回 B、第 N 次返回默认值 / 报错这类高频业务场景。这个也和上面2和3一样支持两种写法函数和结构体方法写法类似这里就只写一个示例了参数示例// 存储单次调用的返回值集合type Params []interface{}type OutputCell struct {Values Params // 本次要返回的参数个数/类型必须和原方法返回值完全一致Times int // 生效次数-1永久生效匹配后后续调用全用这个0生效指定次数}// 第一个参数要打桩的方法所属的类型通过 reflect.TypeOf(实例) 获取区分值接收者 / 指针接收者// 第二个参数要打桩的方法名字符串格式、大小写敏感必须和原方法名完全一致// 第三个参数打桩方法返回值切片入参和出参要保持和被打桩方法保持一致func (this *Patches) ApplyMethodSeq(target interface{}, methodName string, outputs []OutputCell) *Patches {m, ok : castRType(target).MethodByName(methodName)if !ok {panic(retrieve method by name failed)}d : getDoubleFunc(m.Type, outputs)return this.ApplyCore(m.Func, d)}product.gopackage method_demoimport fmt// 1. 定义业务结构体和待打桩的方法 type Product struct {Id intName stringStock int}// 结构体公有方法首字母大写库存查询返回【库存数量、错误信息】func (p *Product) GetStock() (int, error) {// 真实业务逻辑查询数据库/缓存获取库存return p.Stock, nil}// 2. 业务逻辑连续调用3次GetStock方法 // 模拟业务中多次调用结构体方法的场景func QueryStockMultiTimes(p *Product) {// 第1次调用stock1, err1 : p.GetStock()fmt.Printf(第1次查询库存: %d, err: %v\n, stock1, err1)// 第2次调用stock2, err2 : p.GetStock()fmt.Printf(第2次查询库存: %d, err: %v\n, stock2, err2)// 第3次调用stock3, err3 : p.GetStock()fmt.Printf(第3次查询库存: %d, err: %v\n, stock3, err3)}product_test.gopackage method_demoimport (errorsgithub.com/agiledragon/gomonkey/v2testing)func TestProduct(t *testing.T) {// 1. 创建patches实例必须写defer Reset()保证打桩还原无残留patches : gomonkey.NewPatches()defer patches.Reset()// 2. 初始化结构体实例prod : Product{Id: 1, Name: 苹果手机, Stock: 200}// 3. 核心定义序列打桩规则stockSeq : []gomonkey.OutputCell{{Values: gomonkey.Params{100, nil}, Times: 1}, // 规则1第1次调用返回100nil{Values: gomonkey.Params{0, nil}, Times: 1}, // 规则2第2次调用返回0nil{Values: gomonkey.Params{0, errors.New(库存不足无法下单)}, Times: 1}, // 规则3永久生效-1这个参数会导致严重bug避免使用}// 4. 执行方法序列打桩patches.ApplyMethodSeq(prod, GetStock, stockSeq)// 5. 执行业务逻辑触发多次调用QueryStockMultiTimes(prod)}命令行执行命令go test -run ^TestProduct$ -cover -gcflagsall-l -covermodeatomic结果PS D:\wyl\workspace\go\tracer\logic\method_demo go test -run ^TestProduct$ -cover -gcflagsall-l -covermodeatomic第1次查询库存: 100, err: nil第2次查询库存: 0, err: nil第3次查询库存: 0, err: 库存不足无法下单PASScoverage: 54.5% of statementsok tracer/logic/method_demo 0.330s5结构体私有方法打桩方法gomonkey.ApplyPrivateMethod()私有方法打桩方案方案一【最优推荐零侵入、Go 官方标准】测试文件和业务文件放在同一个包下首选无任何副作用方案二【妥协方案少量侵入】把「需要 mock 的私有方法」改为包级私有函数适合特殊场景方案三【不推荐侵入性大】把私有方法改为公有方法万不得已才用ApplyPrivateMethod和ApplyMethod打私有方法区别功能上完全一致同包下都能正常打桩、正常 mock 逻辑。唯一的区别是ApplyPrivateMethod语义精准一看就知道是打「私有方法」符合官方设计ApplyMethod语义模糊它的设计初衷是打「公有方法」打私有方法只是兼容生效。这个也和上面2和3一样支持两种写法函数和结构体方法写法类似这里就只写一个示例了参数示例// 第一个参数要打桩的方法所属的类型通过 reflect.TypeOf(实例) 获取区分值接收者 / 指针接收者// 第二个参数要打桩的方法名字符串格式、大小写敏感必须和原方法名完全一致// 第三个参数打桩方法入参和出参要保持和被打桩方法保持一致func (this *Patches) ApplyPrivateMethod(target interface{}, methodName string, double interface{}) *Patches {m, ok : creflect.MethodByName(castRType(target), methodName)if !ok {panic(retrieve method by name failed)}d : reflect.ValueOf(double)return this.ApplyCoreOnlyForPrivateMethod(m, d)}student.gopackage private_demotype Student struct{}// 公有方法对外暴露func (s Student) Study(course string) bool {// 内部调用了私有方法if s.checkCourse(course) { // 私有方法首字母小写s.recordLog(course) // 私有方法return true}return false}// 私有方法内部逻辑func (s Student) checkCourse(course string) bool {return course ! }func (s Student) recordLog(course string) {}student_test.gopackage private_demoimport (github.com/agiledragon/gomonkey/v2testing)// TODO: TestStudy有问题暂时不确定是因为不支持这种写法还是代码有问题func TestStudy(t *testing.T) {s : Student{}patches : gomonkey.NewPatches()defer patches.Reset()// ✅ 可以直接用gomonkey打桩【私有方法】checkCourse零权限问题patches.ApplyMethod(Student{}, checkCourse, func(_ *Student, course string) bool {return true // mock私有方法返回true})res : s.Study(math)if !res {t.Error(测试失败)}}// 测试代码指针接收者的私有方法打桩func TestStudyPrivate(t *testing.T) {s : Student{}patches : gomonkey.NewPatches()defer patches.Reset()patches.ApplyPrivateMethod((*Student)(nil), // ✅ 指针接收者传 (*结构体)(nil)checkCourse, // 私有方法名不变func(_ *Student, course string) bool { // ✅ 桩函数接收者是指针类型return true},)res : s.Study(math)if !res {t.Error(测试失败)}

多次调用，顺序返回不同结果

相关推荐

Python上下文管理器：with语句与enter/exit完全指南

SetDPI：3步掌握Windows命令行DPI调整的终极方案

Win11Debloat：3分钟免费优化Windows系统，让电脑重获新生

更多资讯

嵌入式固件烧录与OTA升级实战：从编译到远程更新的完整链路

幼儿园守时习惯培养有必要吗？深度解析幼儿守时教育的核心价值与培养逻辑

CCNP ENCOR实战：PDF配置手册的ENSP转化与VLAN/Trunk排错指南

海誓山盟景区情侣浪漫景点推荐用户力荐

Linux部署TModLoader服务器：原理、避坑与systemd实战

改进遗传算法优化神经网络结构与超参

今日更新

MATLAB+Yalmip+CPLEX实战：综合能源系统优化调度全流程解析

I3C比I2C快10倍？RK3576实战：速率、DTS配置与混合总线避坑指南

字符串转对象：JSON.parse、new Function与URLSearchParams

本周精选

如何彻底离线使用 Laya-CoreML：Hugging Face 下载、缓存机制与 local_files_only 详解

基于1Panel 部署MySQL MCP 服务：TaoToken 统一 Key 接入与 config.toml 配置实战

崇州园区营销网站建设避坑指南：5大注意事项保备案安全

本月热门

基于YOLO的焊缝缺陷检测：从数据标注到工业部署全流程实践

分布式服务的决策记录

创业产品的首版范围

读完文章，想聊聊您的网站？