lib/nc.rb in rspec-nc-0.0.6 vs lib/nc.rb in rspec-nc-0.1.0

@@ -1,13 +1,17 @@
 require 'rspec/core/formatters/base_text_formatter'
 require 'terminal-notifier'
 
 class Nc < RSpec::Core::Formatters::BaseTextFormatter
+  if RSpec::Core::Formatters.respond_to? :register
+    RSpec::Core::Formatters.register self.class, :example_failed
+  end
+
   def dump_summary(duration, example_count, failure_count, pending_count)
     body = []
-    body << "Finished in #{format_duration duration}"
-    body << summary_line(example_count, failure_count, pending_count)
+    body << "Finished in #{_format_duration duration}"
+    body << _summary_line(example_count, failure_count, pending_count)
 
     name = File.basename(File.expand_path '.')
 
     title = if failure_count > 0
       "\u26D4 #{name}: #{failure_count} failed example#{failure_count == 1 ? nil : 's'}"
@@ -25,6 +29,31 @@
   private
 
   def say(title, body)
     TerminalNotifier.notify body, :title => title
   end
+
+  def _format_duration(duration)
+    if respond_to? :format_duration
+      format_duration duration
+    else
+      require 'rspec/core/formatters/helpers'
+      RSpec::Core::Formatters::Helpers.format_duration duration
+    end
+  end
+
+  def _summary_line(example_count, failure_count, pending_count)
+    if respond_to? :summary_line
+      summary_line(example_count, failure_count, pending_count)
+    else
+      RSpec::Core::Notifications::SummaryNotification.new(
+        nil,
+        SizeResponder.new(example_count),
+        SizeResponder.new(failure_count),
+        SizeResponder.new(pending_count),
+        nil
+      ).totals_line
+    end
+  end
+
+  SizeResponder = Struct.new(:size)
 end