lib/draper/test/minitest_integration.rb in draper-1.0.0.beta6 vs lib/draper/test/minitest_integration.rb in draper-1.0.0

@@ -1,7 +1,6 @@
-class MiniTest::Rails::ActiveSupport::TestCase
-  # Use AS::TestCase for the base class when describing a decorator
+class Draper::TestCase
   register_spec_type(self) do |desc|
-    desc < Draper::Decorator if desc.is_a?(Class)
+    desc < Draper::Decorator || desc < Draper::CollectionDecorator if desc.is_a?(Class)
   end
   register_spec_type(/Decorator( ?Test)?\z/i, self)
 end