update to parse database.yml through erb

ZhangHanDong · Jun 30, 2008 · ea6a859 · ea6a859
1 parent 2ecc092
commit ea6a859
Show file tree

Hide file tree

Showing 2 changed files 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(File.read "#{RAILS_ROOT}/config/database.yml")
+      connections = YAML.load(ERB.new(IO.read("#{RAILS_ROOT}/config/database.yml")).result)
       raise "Cannot find database specification.  Configuration '#{str}' expected in config/database.yml" if (connections[str].nil?)      
       return connections[str]
     end

diff --git a/test/use_db_test.rb b/test/use_db_test.rb
@@ -2,7 +2,7 @@
 
 class UseDbTest < Test::Unit::TestCase
   # Replace this with your real tests.
-  def test_this_plugin
+  def test_it_will_get_the_right_connection
     flunk
   end
 end