测试表

在核心测试逻辑重复时，将表驱动测试与子测试一起使用，以避免重复代码。

BadGood
  1. // func TestSplitHostPort(t testing.T)
  3. host, port, err := net.SplitHostPort("192.0.2.0:8000")
  4. require.NoError(t, err)
  5. assert.Equal(t, "192.0.2.0", host)
  6. assert.Equal(t, "8000", port)
  8. host, port, err = net.SplitHostPort("192.0.2.0:http")
  9. require.NoError(t, err)
  10. assert.Equal(t, "192.0.2.0", host)
  11. assert.Equal(t, "http", port)
  13. host, port, err = net.SplitHostPort(":8000")
  14. require.NoError(t, err)
  15. assert.Equal(t, "", host)
  16. assert.Equal(t, "8000", port)
  18. host, port, err = net.SplitHostPort("1:8")
  19. require.NoError(t, err)
  20. assert.Equal(t, "1", host)
  21. assert.Equal(t, "8", port)
 
  1. // func TestSplitHostPort(t testing.T)
  3. tests := []struct{
  4.   give     string
  5.   wantHost string
  6.   wantPort string
  7. }{
  8.   {
  9.     give:     "192.0.2.0:8000",
  10.     wantHost: "192.0.2.0",
  11.     wantPort: "8000",
  12.   },
  13.   {
  14.     give:     "192.0.2.0:http",
  15.     wantHost: "192.0.2.0",
  16.     wantPort: "http",
  17.   },
  18.   {
  19.     give:     ":8000",
  20.     wantHost: "",
  21.     wantPort: "8000",
  22.   },
  23.   {
  24.     give:     "1:8",
  25.     wantHost: "1",
  26.     wantPort: "8",
  27.   },
  28. }
  30. for _, tt := range tests {
  31.   t.Run(tt.give, func(t *testing.T) {
  32.     host, port, err := net.SplitHostPort(tt.give)
  33.     require.NoError(t, err)
  34.     assert.Equal(t, tt.wantHost, host)
  35.     assert.Equal(t, tt.wantPort, port)
  36.   })
  37. }

测试表使得向错误消息注入上下文信息，减少重复的逻辑，添加新的测试用例变得更加容易。

我们遵循这样的约定：将结构体切片称为 tests。 每个测试用例称为 tt 。此外，我们鼓励使用 give 和 want 前缀说明每个测试用例的输入和输出值。

  1. tests := []struct{
  2.   give     string
  3.   wantHost string
  4.   wantPort string
  5. }{
  6.   // ...
  7. }
  9. for _, tt := range tests {
  10.   // ...
  11. }