minitest_integration.rb in draper-0.18.0

- old
+ new

@@ -1,26 +1,7 @@
-require 'draper/test/view_context'
-
-module MiniTest
-  class Spec
-    class Decorator < Spec
-      before { Draper::ViewContext.infect!(self) }
-    end
+class MiniTest::Rails::ActiveSupport::TestCase
+  # Use AS::TestCase for the base class when describing a decorator
+  register_spec_type(self) do |desc|
+    desc < Draper::Base if desc.is_a?(Class)
   end
-end
-
-class MiniTest::Unit::DecoratorTestCase < MiniTest::Unit::TestCase
-  if method_defined?(:before_setup)
-    # for minitext >= 2.11
-    def before_setup
-      super
-      Draper::ViewContext.infect!(self)
-    end
-  else
-    # for older minitest, like what ships w/Ruby 1.9
-    add_setup_hook { Draper::ViewContext.infect!(self) }
-  end
-end
-
-MiniTest::Spec.register_spec_type(MiniTest::Spec::Decorator) do |desc|
-  desc.superclass == Draper::Base
+  register_spec_type(/Decorator( ?Test)?\z/i, self)
 end