1 files changed, 4 insertions, 56 deletions
diff --git a/spec/mspec/lib/mspec/runner/formatters/profile.rb b/spec/mspec/lib/mspec/runner/formatters/profile.rb
index 6d15a56349..38ef5b12ed 100644
--- a/spec/mspec/lib/mspec/runner/formatters/profile.rb
+++ b/spec/mspec/lib/mspec/runner/formatters/profile.rb
@@ -1,9 +1,9 @@
-require 'mspec/expectations/expectations'
 require 'mspec/runner/formatters/dotted'
+require 'mspec/runner/actions/profile'
 
 class ProfileFormatter < DottedFormatter
-  def initialize(out=nil)
-    super
+  def initialize(out = nil)
+    super(out)
 
     @describe_name = nil
     @describe_time = nil
@@ -12,59 +12,7 @@ class ProfileFormatter < DottedFormatter
   end
 
   def register
-    super
-    MSpec.register :enter, self
-  end
-
-  # Callback for the MSpec :enter event. Prints the
-  # +describe+ block string.
-  def enter(describe)
-    if @describe_time
-      @describes << [@describe_name, Time.now.to_f - @describe_time]
-    end
-
-    @describe_name = describe
-    @describe_time = Time.now.to_f
-  end
-
-  # Callback for the MSpec :before event. Prints the
-  # +it+ block string.
-  def before(state)
-    super
-
-    @it_name = state.it
-    @it_time = Time.now.to_f
-  end
-
-  # Callback for the MSpec :after event. Prints a
-  # newline to finish the description string output.
-  def after(state = nil)
-    @its << [@describe_name, @it_name, Time.now.to_f - @it_time]
-    super
-  end
-
-  def finish
-    puts "\nProfiling info:"
-
-    desc = @describes.sort { |a,b| b.last <=> a.last }
-    desc.delete_if { |a| a.last <= 0.001 }
-    show = desc[0, 100]
-
-    puts "Top #{show.size} describes:"
-
-    show.each do |des, time|
-      printf "%3.3f - %s\n", time, des
-    end
-
-    its = @its.sort { |a,b| b.last <=> a.last }
-    its.delete_if { |a| a.last <= 0.001 }
-    show = its[0, 100]
-
-    puts "\nTop #{show.size} its:"
-    show.each do |des, it, time|
-      printf "%3.3f - %s %s\n", time, des, it
-    end
-
+    (@profile = ProfileAction.new).register
     super
   end
 end