Sha256: fc3f8cc95e91467824d0e594379a2e3876daabff4986edded3eb840710e7c794

Contents?: true

Size: 979 Bytes

Versions: 11

Compression:

Stored size: 979 Bytes

Contents

module Database
  def setup_database(db_name, db_config)
    setup_pg(db_name, db_config) if RUBY_PLATFORM == 'ruby'
    setup_jdbc(db_name, db_config) if RUBY_PLATFORM == 'java'
  end

  def setup_pg(db_name, db_config)
    begin
      ActiveRecord::Base.establish_connection(db_name)
      ActiveRecord::Base.connection
    rescue PG::ConnectionBad
      ActiveRecord::Base.establish_connection db_config.merge('database' => nil)
      ActiveRecord::Base.connection.create_database db_config['database']
      ActiveRecord::Base.establish_connection db_config
    end
  end

  def setup_jdbc(db_name, db_config)
    begin
      ActiveRecord::Base.establish_connection(db_name)
      ActiveRecord::Base.connection
    rescue ActiveRecord::JDBCError
      ActiveRecord::Base.establish_connection db_config.merge('database' => nil)
      ActiveRecord::Base.connection.create_database db_config['database']
      ActiveRecord::Base.establish_connection db_config
    end
  end
end

Version data entries

11 entries across 11 versions & 1 rubygems

Version Path
usda-nutrient-database-2.1.2 spec/support/database.rb
usda-nutrient-database-1.5.0 spec/support/database.rb
usda-nutrient-database-1.4.1 spec/support/database.rb
usda-nutrient-database-2.1.1 spec/support/database.rb
usda-nutrient-database-2.1.0 spec/support/database.rb
usda-nutrient-database-2.0.0 spec/support/database.rb
usda-nutrient-database-1.4.0 spec/support/database.rb
usda-nutrient-database-1.3.0 spec/support/database.rb
usda-nutrient-database-1.2.0 spec/support/database.rb
usda-nutrient-database-1.1.0 spec/support/database.rb
usda-nutrient-database-1.0.1 spec/support/database.rb