From f6d88530b711ac374bdef57c8b1f07c3d22bff41 Mon Sep 17 00:00:00 2001
From: Chase DuBois <carchase@gmail.com>
Date: Sun, 12 Jun 2011 14:19:59 -0400
Subject: [PATCH] set default ORM in Rails 3.1

---
 lib/couchrest/railtie.rb | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/lib/couchrest/railtie.rb b/lib/couchrest/railtie.rb
index b1266c4..b08d00f 100644
--- a/lib/couchrest/railtie.rb
+++ b/lib/couchrest/railtie.rb
@@ -2,10 +2,13 @@ require "rails"
 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|