update to work properly if models are loaded inside of ERB, eg. in fi…

…xtures
peppyheppy · Oct 10, 2008 · 4311da1 · 4311da1
1 parent ea6a859
commit 4311da1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/use_db.rb b/lib/use_db.rb
@@ -55,7 +55,7 @@ def get_use_db_conn_spec(options)
       return options
     else
       str = "#{prefix}#{rails_env}#{suffix}"
-      connections = YAML.load(ERB.new(IO.read("#{RAILS_ROOT}/config/database.yml")).result)
+      connections = YAML.load(ERB.new(IO.read("#{RAILS_ROOT}/config/database.yml"), nil, nil, '_use_db_erbout').result)
       raise "Cannot find database specification.  Configuration '#{str}' expected in config/database.yml" if (connections[str].nil?)      
       return connections[str]
     end
@@ -64,4 +64,4 @@ def get_use_db_conn_spec(options)
 
 class UseDbPluginClass
   extend UseDbPlugin
-end
+end