如何在 go 测试中仅重新运行失败的测试

Question

我有一个带有多个文件和测试的 package。 目前要重新运行我正在使用的失败测试：

func TestMain(m *testing.M) {
    var r int = -1
    maxRuns := 3
    for i := 0; i < maxRuns && r != 0; i++ {
        r = m.Run()
    }
    os.Exit(r)
}

实际上它有效，但是它在单个测试失败场景中重新运行所有 package 测试。 所以有时它会导致下一个序列：通过、通过、失败。 这样测试标记为失败。 我的目标是仅在 package 中重新运行失败的测试，在第一次成功时将其标记为通过，仅在三次失败的情况下将测试标记为失败。 有什么建议吗？

Answer 1

这可能已经回答了很长时间，但是我看到的集成测试的方式是将重试代码包装在 function 中，该代码对您的测试实现进行回调。 如果您正在拨打 API 电话并且在context.Done中需要检查挂断电话或超时，这将特别有用。 例如： https://pkg.go.dev/k8s.io/client-go/util/retry

这是一个简单的例子。

func MyPolicy(ctx context.Context, t *testing.T, retries int, testImpl func(context.Context, t *testing.T)) error {
  for i := 0; i < retries; i++ {
    err := testImpl(ctx, t)
    if err != nil {
      t.Logf("Test error: %v", err)
      continue
    }
    return nil
  }
  return fmt.Errorf("Retry budget of %d attempts exhausted", retries)
}


func TestIntegration(t *testing.T) {
  impl := func(ctx context.Context, t *testing.T) error {
   // implement here calling t.Logf when needed not t.Fatal/Error
  }

  if err := MyPolicy(context.Background(), t, 3, impl); err != nil {
    t.Fatalf("Test failed with: %v", err)
  } 
}