fix failing test case

sinisterchipmunk · Sep 7, 2016 · 5ef999a · 5ef999a
1 parent 8239c5a
commit 5ef999a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lib/rspec-prof.rb b/lib/rspec-prof.rb
@@ -51,19 +51,21 @@ def save_to filename
 end
 
 RSpec.configure do |config|
+  profiler = nil
+
   config.before(:suite) do
     unless ['all', 'each', ''].include?(ENV['RSPEC_PROFILE'].to_s)
       raise "ENV['RSPEC_PROFILE'] should be blank, 'all' or 'each', but was '#{ENV['RSPEC_PROFILE']}'"
     end
 
     if ENV['RSPEC_PROFILE'] == 'all'
-      @profiler = RSpecProf.new.start
+      profiler = RSpecProf.new.start
     end
   end
 
   config.after(:suite) do
     if ENV['RSPEC_PROFILE'] == 'all'
-      @profiler.save_to "#{RSpecProf.output_dir}/all.#{RSpecProf.file_extension}"
+      profiler.save_to "#{RSpecProf.output_dir}/all.#{RSpecProf.file_extension}"
     end
   end