lib/rake/funnel/tasks/nunit.rb in rake-funnel-0.0.6.pre vs lib/rake/funnel/tasks/nunit.rb in rake-funnel-0.1.0.pre

@@ -1,9 +1,11 @@
 require 'rake/tasklib'
 
 module Rake::Funnel::Tasks
   class NUnit < Rake::TaskLib
+    include Rake::Funnel::Support
+
     attr_accessor :name, :files, :args
 
     def initialize(name = :test)
       @name = name
       @args = {}
@@ -26,21 +28,21 @@
       desc "Test #{test_assemblies.all_or_default.join(', ')}"
       task name do
         Rake::Funnel::Integration::TeamCity::NUnitPlugin.setup(nunit)
 
         cmd = [
-          *Rake::Funnel::Support::Mono.invocation(nunit),
+          *Mono.invocation(nunit),
           *test_assemblies.all,
-          *Rake::Funnel::Support::Mapper.new(:NUnit).map(args)
+          *Mapper.new(:NUnit).map(args)
         ]
 
         sh(*cmd)
       end
 
       self
     end
 
     def test_assemblies
-      Rake::Funnel::Support::Finder.new(files, self, 'No test assemblies found.')
+      Finder.new(files, self, 'No test assemblies found.')
     end
   end
 end