Fix testing type on new helper funcs

hashicorp · Jun 11, 2021 · 27730d4 · 27730d4
1 parent 6f15007
commit 27730d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/testutil/wait.go b/testutil/wait.go
@@ -232,7 +232,7 @@ func WaitForRunning(t testing.TB, rpc rpcFn, job *structs.Job) []*structs.AllocL
 }
 
 // WaitForFiles blocks until all the files in the slice are present
-func WaitForFiles(t testing.T, files []string) {
+func WaitForFiles(t testing.TB, files []string) {
 	WaitForResult(func() (bool, error) {
 		return FilesExist(files), nil
 	}, func(err error) {
@@ -241,7 +241,7 @@ func WaitForFiles(t testing.T, files []string) {
 }
 
 // WaitForFilesUntil blocks until duration or all the files in the slice are present
-func WaitForFilesUntil(t testing.T, files []string, until time.Duration) {
+func WaitForFilesUntil(t testing.TB, files []string, until time.Duration) {
 	WaitForResultUntil(until, func() (bool, error) {
 		return FilesExist(files), nil
 	}, func(err error) {