lib/test_dummy/railtie.rb in test_dummy-0.2.10 vs lib/test_dummy/railtie.rb in test_dummy-0.3.0

@@ -4,17 +4,21 @@
     ActiveRecord::Base.send(:include, TestDummy)
   end
   if (defined?(ActiveSupport) and defined?(ActiveSupport::TestCase))
     ActiveSupport::TestCase.send(:include, TestDummy::TestHelper)
   end
+  if (defined?(Test) and defined?(Test::Unit))
+    Test::Unit::TestCase.send(:include, TestDummy::TestHelper)
+  end
 else
   class TestDummy::Railtie < Rails::Railtie
     def self.apply!
       if (defined?(ActiveRecord))
         ActiveRecord::Base.send(:include, TestDummy)
       end
 
       ActiveSupport::TestCase.send(:include, TestDummy::TestHelper)
+      Test::Unit::TestCase.send(:include, TestDummy::TestHelper)
     end
     
     config.before_configuration do
       apply!
     end