lib/test_dummy/railtie.rb in test_dummy-0.1.4 vs lib/test_dummy/railtie.rb in test_dummy-0.2.0

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