spy.rb in warnings_logger-0.1.1

- old
+ new
@@ -1,85 +1,136 @@
 require "forwardable"
 
 module WarningsLogger
   # Adapted from <http://myronmars.to/n/dev-blog/2011/08/making-your-gem-warning-free>
   class Spy
-    def self.call(args)
-      configuration = Configuration.new(args)
-      new(configuration).call
+    # DEPRECATED
+    def self.call(**options)
+      WarningsLogger.configure do |config|
+        options.each do |key, value|
+          config.public_send("#{key}=", value)
+        end
+      end
+
+      WarningsLogger.enable
     end
 
+    def self.enable(configuration)
+      new(configuration).enable
+    end
+
     extend Forwardable
 
     def initialize(configuration)
       @filesystem = Filesystem.new(configuration)
       @reader = Reader.new(filesystem)
       @partitioner = Partitioner.new(
         configuration: configuration,
-        reader: reader,
+        reader: reader
       )
       @reporter = Reporter.new(
         configuration: configuration,
         filesystem: filesystem,
-        partitioner: partitioner,
+        partitioner: partitioner
       )
+
+      @original_stderr = nil
     end
 
-    def call
+    def enable
       filesystem.prepare
       capture_warnings
-      report_warnings_at_exit
+      report_warnings_after_tests_run
     end
 
     private
 
     attr_reader :filesystem, :reader, :partitioner, :reporter
 
     def_delegators :filesystem, :warnings_file
 
-    def_delegators :partitioner, :relevant_warning_groups,
+    def_delegators(
+      :partitioner,
+      :relevant_warning_groups,
       :irrelevant_warning_groups
+    )
 
     def capture_warnings
+      @original_stderr = $stderr.dup
       $stderr.reopen(warnings_file.path)
     end
 
-    def report_warnings_at_exit
-      at_exit do
-        printing_exceptions do
-          report_and_exit
+    def release_warnings
+      $stderr.reopen(@original_stderr)
+    end
+
+    def report_warnings_after_tests_run
+      spy = self
+
+      if should_integrate_with_rspec?
+        RSpec.configure do |config|
+          config.after(:suite) do
+            spy.instance_eval do
+              release_warnings
+              printing_exceptions do
+                report_and_possibly_fail
+              end
+            end
+          end
         end
+      elsif should_integrate_with_minitest?
+        Minitest.after_run do
+          release_warnings
+          printing_exceptions do
+            report_and_possibly_fail
+          end
+        end
+      else
+        at_exit do
+          release_warnings
+          printing_exceptions do
+            report_and_possibly_fail
+          end
+        end
       end
     end
 
     def printing_exceptions
       yield
     rescue StandardError => error
-      puts "\n--- ERROR IN AT_EXIT --------------------------------"
-      puts "#{error.class}: #{error.message}"
-      puts error.backtrace.join("\n")
-      puts "-----------------------------------------------------"
+      warn "\n--- ERROR IN AT_EXIT --------------------------------"
+      warn "#{error.class}: #{error.message}"
+      warn error.backtrace.join("\n")
+      warn "-----------------------------------------------------"
       raise error
     end
 
-    def report_and_exit
+    def report_and_possibly_fail
       reader.read
       partitioner.partition
 
       if relevant_warning_groups.any?
         report_warnings_and_fail
       else
-        print_warnings_and_fail
+        print_warnings
       end
     end
 
     def report_warnings_and_fail
       reporter.report
       exit(1)
     end
 
-    def print_warnings_and_fail
+    def print_warnings
       filesystem.warnings_file.rewind
       puts filesystem.warnings_file.read
+    end
+
+    def should_integrate_with_rspec?
+      defined?(RSpec)
+    end
+
+    def should_integrate_with_minitest?
+      defined?(Minitest) && Minitest.class_variable_get("@@installed_at_exit")
     end
   end
 end