From 178d0199ca15a520cfdbf0f578ee5f1a6122783d Mon Sep 17 00:00:00 2001
From: Karl Baum <karl.baum@gmail.com>
Date: Fri, 7 Jun 2019 19:33:58 -0400
Subject: [PATCH] Add ability to benchmark ton of files

---
 test/benchmarks/benchmark.rake | 11 +++++++++++
 test/unique_files.rb           |  6 ++++--
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/test/benchmarks/benchmark.rake b/test/benchmarks/benchmark.rake
index 62efb0e3..ca8f887f 100755
--- a/test/benchmarks/benchmark.rake
+++ b/test/benchmarks/benchmark.rake
@@ -370,6 +370,17 @@ namespace :benchmarks do
     run_work(true)
   end
 
+  task run_big: %i[setup setup_redis] do
+    require 'memory_profiler'
+    require './test/unique_files'
+    1000.times { require_unique_file }
+    # warmup
+    3.times { Coverband.report_coverage }
+    MemoryProfiler.report do
+      10.times { Coverband.report_coverage }
+    end.pretty_print
+  end
+
   # desc 'runs benchmarks file store'
   task run_file: %i[setup setup_file] do
     puts 'Coverband configured with file store'
diff --git a/test/unique_files.rb b/test/unique_files.rb
index 872803a8..d16b049e 100644
--- a/test/unique_files.rb
+++ b/test/unique_files.rb
@@ -19,6 +19,8 @@ def remove_unique_files
   FileUtils.rm_r(UNIQUE_FILES_DIR) if File.exist?(UNIQUE_FILES_DIR)
 end
 
-Minitest.after_run do
-  remove_unique_files
+if defined?(Minitest)
+  Minitest.after_run do
+    remove_unique_files
+  end
 end