From 334bf17d452b6c6a08c24ca496958a2aed03d281 Mon Sep 17 00:00:00 2001
From: Markus Schirp <mbj@seonic.net>
Date: Thu, 22 Nov 2012 02:51:16 +0100
Subject: [PATCH] Add basic statistics

---
 lib/mutant/reporter/cli.rb | 79 +++++++++++++++++++++++++++++++++++---
 lib/mutant/runner.rb       |  2 +
 2 files changed, 75 insertions(+), 6 deletions(-)

diff --git a/lib/mutant/reporter/cli.rb b/lib/mutant/reporter/cli.rb
index cb8897ef..9472c826 100644
--- a/lib/mutant/reporter/cli.rb
+++ b/lib/mutant/reporter/cli.rb
@@ -4,6 +4,36 @@ module Mutant
     class CLI < self
       include Equalizer.new(:io)
 
+      class Stats
+        attr_reader :subject
+        attr_reader :mutation
+        attr_reader :kill
+        attr_reader :time
+
+        def initialize
+          @start = Time.now
+          @subject = @mutation = @kill = @time = 0
+        end
+
+        def runtime
+          Time.now - @start
+        end
+
+        def subject
+          @subject +=1
+        end
+
+        def alive
+          @mutation - @kill
+        end
+
+        def killer(killer)
+          @mutation +=1
+          @kill +=1 unless killer.fail?
+          @time += killer.runtime
+        end
+      end
+
       # Reporter subject
       #
       # @param [Subject] subject
@@ -13,6 +43,7 @@ module Mutant
       # @api private
       #
       def subject(subject)
+        stats.subject
         puts("Subject: #{subject.identification}")
       end
 
@@ -51,15 +82,42 @@ module Mutant
       # @api private
       #
       def killer(killer)
-        if killer.fail?
-          failure(killer)
-        else
-          puts("Killed: #{killer.identification} (%02.2fs)" % killer.runtime)
-        end
+        stats.killer(killer)
+
+        color, word =
+          if killer.fail?
+            [Color::RED,   'Alive']
+          else
+            [Color::GREEN, 'Killed']
+          end
+
+        puts(colorize(color, "#{word}: #{killer.identification} (%02.2fs)" % killer.runtime))
 
         self
       end
 
+      # Report errors
+      #
+      # @param [Enumerable<Killer>]
+      #
+      # @api private
+      #
+      # @return [self]
+      #
+      def errors(errors)
+        errors.each do |error|
+          failure(error)
+        end
+
+        puts
+        puts "subjects:  #{stats.subject}"
+        puts "mutations: #{stats.mutation}"
+        puts "kills:     #{stats.kill}"
+        puts "alive:     #{stats.alive}"
+        puts "mtime:     %02.2fs" % stats.time
+        puts "rtime:     %02.2fs" % stats.runtime
+      end
+
       # Return IO stream
       #
       # @return [IO]
@@ -68,6 +126,14 @@ module Mutant
       # 
       attr_reader :io
 
+      # Retun stats
+      #
+      # @return [Stats]
+      #
+      # @api private
+      #
+      attr_reader :stats
+
     private 
 
       # Initialize reporter
@@ -80,6 +146,7 @@ module Mutant
       #
       def initialize(io)
         @io = io
+        @stats = Stats.new
       end
 
       # Report failure on killer
@@ -101,7 +168,7 @@ module Mutant
           raise "Unable to create a diff"
         end
         puts(diff)
-        puts
+        puts("Took: (%02.2fs)" % killer.runtime)
       end
 
       # Test for colored output
diff --git a/lib/mutant/runner.rb b/lib/mutant/runner.rb
index b1966204..c2c9d40b 100644
--- a/lib/mutant/runner.rb
+++ b/lib/mutant/runner.rb
@@ -50,6 +50,8 @@ module Mutant
       reporter.config(config)
 
       run
+
+      reporter.errors(@errors)
     end
 
     # Return reporter