lib/draper/test_case.rb in draper-2.1.0 vs lib/draper/test_case.rb in draper-3.0.0.pre1

@@ -27,16 +27,12 @@
     include Behavior
     include ViewContextTeardown
   end
 end
 
-if defined?(ActionController::TestCase)
-  class ActionController::TestCase
-    include Draper::TestCase::ViewContextTeardown
-  end
+if defined? ActionController::TestCase
+  ActionController::TestCase.include Draper::TestCase::ViewContextTeardown
 end
 
-if defined?(ActionMailer::TestCase)
-  class ActionMailer::TestCase
-    include Draper::TestCase::ViewContextTeardown
-  end
+if defined? ActionMailer::TestCase
+  ActionMailer::TestCase.include Draper::TestCase::ViewContextTeardown
 end