chore: process the first 1000 rules only

1 files changed, 11 insertions, 13 deletions

diff --git a/internal/testhelpers/tmpdir.go b/internal/testhelpers/tmpdir.go
index 81f0781b..5765d308 100644
--- a/internal/testhelpers/tmpdir.go
+++ b/internal/testhelpers/tmpdir.go
@@ -15,24 +15,22 @@ import (
 
 var fs = vfs.Instrumented(&local.VFS{})
 
-func TmpDir(t *testing.T, pattern string) (vfs.Root, string, func()) {
+func TmpDir(tb testing.TB, pattern string) (vfs.Root, string) {
+	tb.Helper()
+
 	tmpDir, err := ioutil.TempDir("", pattern)
-	if t != nil {
-		require.NoError(t, err)
-	}
+	require.NoError(tb, err)
 
 	// On some systems `/tmp` can be a symlink
 	tmpDir, err = filepath.EvalSymlinks(tmpDir)
-	if t != nil {
-		require.NoError(t, err)
-	}
+	require.NoError(tb, err)
 
 	root, err := fs.Root(context.Background(), tmpDir)
-	if t != nil {
-		require.NoError(t, err)
-	}
+	require.NoError(tb, err)
+
+	tb.Cleanup(func() {
+		require.NoError(tb, os.RemoveAll(tmpDir))
+	})
 
-	return root, tmpDir, func() {
-		os.RemoveAll(tmpDir)
-	}
+	return root, tmpDir
 }