From 27730d41ee078eb0d7ebb434910ecf5035aebeb4 Mon Sep 17 00:00:00 2001
From: davemay99 <dmay@hashicorp.com>
Date: Fri, 11 Jun 2021 17:39:26 -0400
Subject: [PATCH] Fix testing type on new helper funcs

---
 testutil/wait.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/testutil/wait.go b/testutil/wait.go
index 9af6d073576c..19215d0ced0a 100644
--- 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) {