Adjusting the logic to install the CI::Reporter::TestUnit mediator so…

… that it will work for newer ruby-test versions (>= 2.1) as well.

lib/ci/reporter/rake/test_unit_loader.rb

@@ -5,6 +5,7 @@
 $: << File.dirname(__FILE__) + "/../../.."
 require 'ci/reporter/test_unit'
 
+# Intercepts mediator creation in ruby-test < 2.1
 module Test #:nodoc:all
   module Unit
     module UI
@@ -19,3 +20,17 @@ def create_mediator(suite)
     end
   end
 end
+
+# Intercepts mediator creation in ruby-test >= 2.1
+module Test #:nodoc:all
+  module Unit
+    module UI
+      class TestRunner
+        def setup_mediator
+          # swap in our custom mediator
+          @mediator = CI::Reporter::TestUnit.new(@suite)
+        end
+      end
+    end
+  end
+end