Merge branch 'master' of github.com:couchrest/couchrest_model

couchrest · Jun 25, 2011 · 05ed7b1 · 05ed7b1
2 parents e40b965 + 4ba85b4
commit 05ed7b1
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|