1 files changed, 9 insertions, 0 deletions
diff --git a/test/test_helper.rb b/test/test_helper.rb
index b9a2c9e..f7ec6d9 100644
--- a/test/test_helper.rb
+++ b/test/test_helper.rb
@@ -4,6 +4,7 @@ require 'bundler/setup'
 require 'minitest/autorun'
 require 'leap_cli'
 require 'gli'
+require 'fileutils'
 
 DEBUG = true
 
@@ -49,6 +50,14 @@ class Minitest::Test
     "#{base_path}/test/provider"
   end
 
+  def cleanup_files(*args)
+    Dir.chdir(test_provider_path) do
+      args.each do |file|
+        FileUtils.rm_r(file) if File.exist?(file)
+      end
+    end
+  end
+
   def with_multiple_rubies(&block)
     if ENV["RUBY"]
       ENV["RUBY"].split(',').each do |ruby|