Merge pull request #99 from crx/orm_fix

Set default ORM in Rails 3.1
couchrest · Jun 25, 2011 · 4ba85b4 · 4ba85b4
2 parents ca1d1cd + f6d8853
commit 4ba85b4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/lib/couchrest/railtie.rb b/lib/couchrest/railtie.rb
@@ -2,10 +2,13 @@
 require "active_model/railtie"
 
 module CouchRest
-  # = Active Record Railtie
   class ModelRailtie < Rails::Railtie
-    config.generators.orm :couchrest_model
-    config.generators.test_framework  :test_unit, :fixture => false
+    def self.generator
+      config.respond_to?(:app_generators) ? :app_generators : :generators
+    end
+
+    config.send(generator).orm :couchrest_model
+    config.send(generator).test_framework  :test_unit, :fixture => false
 
     initializer "couchrest_model.configure_default_connection" do
       CouchRest::Model::Base.configure do |conf|