Go 表驱动测试实战:用 t.Run 子测试组织可维护的单元测试

Go 表驱动测试实战:用 t.Run 子测试组织可维护的单元测试你给一个函数写单测,第一版是复制粘贴:测输入 A 断言一段,测输入 B 再抄一遍断言,测边界值又抄一遍。三个用例还好,等长到十几个,某天函数签名改了,你得挨个改十几段几乎一样的代码。更糟的是,一个断言t.Fatal挂掉,后面的用例根本没跑,你不知道到底错了几个。Go 社区的标准答案是表驱动测试:把用例做成一张表,循环跑,配合t.Run拆成独立子测试。这篇手把手带你从「复制粘贴」重构到「一张表搞定」。被测函数拿一个简单的场景:把秒数格式化成人类可读的时长。packagetimeutilimportfmt// FormatDuration 把秒数格式化成 1h2m3s 这样的字符串funcFormatDuration(secint)(string,error){ifsec0{return,fmt.Errorf(秒数不能为负: %d,sec)}h,m,s:sec/3600,(sec%3600)/60,sec%60returnfmt.Sprintf(%dh%dm%ds,h,m,s),nil}朴素写法:复制粘贴,又臭又长第一版新手写法,每个用例抄一遍:funcTestFormatDuration_Naive(t*testing.T){got,_:FormatDuration(0)ifgot!0h0m0s{t.Fatalf(0 秒: 期望 0h0m0s, 得到 %s,got)}got,_FormatDuration(3661)ifgot!1h1m1s{t.Fatalf(3661 秒: 期望 1h1m1s, 得到 %s,got)}_,err:FormatDuration(-1)iferrnil{t.Fatal(负数应该报错)}}三个问题:断言逻辑重复三遍;t.Fatalf一旦触发,后面的用例直接不跑了;加用例只能继续往下抄。表驱动重构:用例是数据,断言只写一次核心思路是把「输入 期望输出」抽成一个结构体切片(就是那张「表」),断言逻辑写一遍,循环套用:funcTestFormatDuration(t*testing.T){// 这张表就是全部用例,加用例 加一行tests:[]struct{namestring// 用例名,出错时能一眼看出是哪个inputintwantstringwantErrbool}{{name:零秒,input:0,want:0h0m0s,wantErr:false},{name:一分钟,input:60,want:0h1m0s,wantErr:false},{name:时分秒齐全,input:3661,want:1h1m1s,wantErr:false},{name:负数报错,input:-1,want:,wantErr:true},}for_,tt:rangetests{got,err:FormatDuration(tt.input)if(err!nil)!tt.wantErr{t.Errorf(%s: 是否报错 %v, 期望 wantErr %v,tt.name,err!nil,tt.wantErr)continue}ifgot!tt.want{t.Errorf(%s: 得到 %q, 期望 %q,tt.name,got,tt.want)}}}注意这里用t.Errorf而不是t.Fatalf——Errorf标记失败但继续跑,所以一次能看到所有失败的用例,而不是碰到第一个就停。加新用例只需在表里加一行,断言逻辑一个字都不用动。升级:t.Run 拆成独立子测试上面的版本还有个小缺点:所有用例挤在一个TestFormatDuration里,测试输出看不清哪个子用例过了、哪个挂了,也没法单独跑某一个。用t.Run给每个用例开一个子测试就解决了:funcTestFormatDuration(t*testing.T){tests:[]struct{namestringinputintwantstringwantErrbool}{{name:零秒,input:0,want:0h0m0s},{name:一分钟,input:60,want:0h1m0s},{name:时分秒齐全,input:3661,want:1h1m1s},{name:负数报错,input:-1,wantErr:true},}for_,tt:rangetests{// 每个用例一个子测试,名字就是 tt.namet.Run(tt.name,func(t*testing.T){got,err:FormatDuration(tt.input)if(err!nil)!tt.wantErr{t.Fatalf(是否报错 %v, 期望 %v,err!nil,tt.wantErr)}ifgot!tt.want{t.Errorf(得到 %q, 期望 %q,got,tt.want)}})}}t.Run的好处:输出里每个子测试独立显示,--- FAIL: TestFormatDuration/负数报错一眼定位。可以只跑某一个:go test -run TestFormatDuration/时分秒齐全。子测试之间隔离,某个子测试t.Fatalf只终止它自己,不影响其余子测试继续跑。跑一下看输出:$ gotest-v-runTestFormatDurationRUN TestFormatDurationRUN TestFormatDuration/零秒RUN TestFormatDuration/一分钟RUN TestFormatDuration/时分秒齐全RUN TestFormatDuration/负数报错 --- PASS: TestFormatDuration(0.00s)--- PASS: TestFormatDuration/零秒(0.00s)--- PASS: TestFormatDuration/一分钟(0.00s)--- PASS: TestFormatDuration/时分秒齐全(0.00s)--- PASS: TestFormatDuration/负数报错(0.00s)PASS一个必须知道的坑:循环变量与 t.Parallel如果你想让子测试并行跑(t.Parallel()),在Go 1.22 之前有个经典陷阱:所有子测试闭包共享同一个循环变量tt,并行执行时它们读到的都是循环结束后的最后一个值。老代码里你会看到这种「重新赋值」的写法来规避:for_,tt:rangetests{tt:tt// Go 1.22 前必须这行,给每次迭代一份独立副本t.Run(tt.name,func(t*testing.T){t.Parallel()// 标记为并行,和其他并行子测试一起跑got,_:FormatDuration(tt.input)ifgot!tt.want{t.Errorf(得到 %q, 期望 %q,got,tt.want)}})}Go 1.22 起循环变量每次迭代都是新的,tt : tt这行可以删掉了。但如果你的项目还跑在老版本,或者你不确定别人的环境,留着它最保险——它无害。不加t.Parallel()时子测试是串行的,不受这个坑影响,但加了并行又忘了处理循环变量,就会得到「所有用例都在测最后一行数据」的诡异结果。小结别复制粘贴断言:把用例抽成结构体切片(表),断言逻辑只写一遍,循环套用。表里给每个用例一个name字段,出错能一眼定位是哪个用例。用t.Run(tt.name, ...)拆成子测试:输出清晰、可单独跑、子测试之间隔离。收集所有失败用t.Errorf(继续跑),需要立即终止当前子测试才用t.Fatalf。并行子测试注意循环变量陷阱:Go 1.22 前要写tt : tt,1.22 起可省;不确定就留着。一句话记忆:用例是数据、断言是代码,一张表 t.Run就是 Go 测试的标准姿势。