From 087e134ed0885ec11325ffabe8ab2bebef77a33a Mon Sep 17 00:00:00 2001
From: Steve Klabnik <steve@steveklabnik.com>
Date: Thu, 29 Nov 2012 17:01:25 -0800
Subject: [PATCH] Adding TestUnit integration.

This should make sure everything works well with T::U.

Fixes #354.
---
 lib/draper.rb                            | 5 +++--
 lib/draper/railtie.rb                    | 3 +++
 lib/draper/tasks/tu.rake                 | 3 +++
 lib/draper/test/test_unit_integration.rb | 9 +++++++++
 4 files changed, 18 insertions(+), 2 deletions(-)
 create mode 100644 lib/draper/tasks/tu.rake
 create mode 100644 lib/draper/test/test_unit_integration.rb

diff --git a/lib/draper.rb b/lib/draper.rb
index 27a50e2..d10cf03 100644
--- a/lib/draper.rb
+++ b/lib/draper.rb
@@ -15,8 +15,9 @@ require 'draper/collection_decorator'
 require 'draper/railtie' if defined?(Rails)
 
 # Test Support
-require 'draper/test/rspec_integration'    if defined?(RSpec) and RSpec.respond_to?(:configure)
-require 'draper/test/minitest_integration' if defined?(MiniTest::Rails)
+require 'draper/test/rspec_integration'     if defined?(RSpec) and RSpec.respond_to?(:configure)
+require 'draper/test/minitest_integration'  if defined?(MiniTest::Rails)
+require 'draper/test/test_unit_integration'
 
 module Draper
   def self.setup_action_controller(base)
diff --git a/lib/draper/railtie.rb b/lib/draper/railtie.rb
index 40bd689..f0e158c 100755
--- a/lib/draper/railtie.rb
+++ b/lib/draper/railtie.rb
@@ -40,5 +40,8 @@ module Draper
       Draper::ViewContext.build_view_context
     end
 
+    rake_tasks do
+      Dir[File.join(File.dirname(__FILE__),'tasks/*.rake')].each { |f| load f }
+    end
   end
 end
diff --git a/lib/draper/tasks/tu.rake b/lib/draper/tasks/tu.rake
new file mode 100644
index 0000000..b200f85
--- /dev/null
+++ b/lib/draper/tasks/tu.rake
@@ -0,0 +1,3 @@
+Rake::Task["test:run"].enhance do
+  Rake::Task["test:decorators"].invoke
+end
diff --git a/lib/draper/test/test_unit_integration.rb b/lib/draper/test/test_unit_integration.rb
new file mode 100644
index 0000000..907bd1a
--- /dev/null
+++ b/lib/draper/test/test_unit_integration.rb
@@ -0,0 +1,9 @@
+require "rake/testtask"
+require "rails/test_unit/sub_test_task"
+
+namespace :test do
+  Rails::SubTestTask.new(:decorators => "test:prepare") do |t|
+    t.libs << "test"
+    t.pattern = "test/decorators/**/*_test.rb"
+  end
+end