lib/draper/test/minitest_integration.rb in draper-0.15.0 vs lib/draper/test/minitest_integration.rb in draper-0.16.0

@@ -7,10 +7,19 @@
     end
   end
 end
 
 class MiniTest::Unit::DecoratorTestCase < MiniTest::Unit::TestCase
-  add_setup_hook { Draper::ViewContext.infect!(self) }
+  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
 end