rails--rails/railties/test/application/dbconsole_test.rb

require "isolation/abstract_unit"
require "console_helpers"

module ApplicationTests
  class DBConsoleTest < ActiveSupport::TestCase
    include ActiveSupport::Testing::Isolation
    include ConsoleHelpers

    def setup
      skip "PTY unavailable" unless available_pty?

      build_app
    end

    def teardown
      teardown_app
    end

    def test_use_value_defined_in_environment_file_in_database_yml
      Dir.chdir(app_path) do
        app_file "config/database.yml", <<-YAML
          development:
             database: <%= Rails.application.config.database %>
             adapter: sqlite3
             pool: <%= ENV.fetch("RAILS_MAX_THREADS") { 5 } %>
             timeout: 5000
        YAML

        app_file "config/environments/development.rb", <<-RUBY
          Rails.application.configure do
            config.database = "db/development.sqlite3"
          end
        RUBY
      end

      master, slave = PTY.open
      spawn_dbconsole(slave)
      assert_output("sqlite>", master)
    ensure
      master.puts ".exit"
    end

    def test_respect_environment_option
      Dir.chdir(app_path) do
        app_file "config/database.yml", <<-YAML
          default: &default
            adapter: sqlite3
            pool: <%= ENV.fetch("RAILS_MAX_THREADS") { 5 } %>
            timeout: 5000

          development:
            <<: *default
            database: db/development.sqlite3

          production:
            <<: *default
            database: db/production.sqlite3
        YAML
      end

      master, slave = PTY.open
      spawn_dbconsole(slave, "-e production")
      assert_output("sqlite>", master)

      master.puts "pragma database_list;"
      assert_output("production.sqlite3", master)
    ensure
      master.puts ".exit"
    end

    private
      def spawn_dbconsole(fd, options = nil)
        Process.spawn("#{app_path}/bin/rails dbconsole #{options}", in: fd, out: fd, err: fd)
      end
  end
end