Merge pull request #42285 from cthornton/include-replicas

Support using replicas when using `rails dbconsole`

Author: eugeneius

railties/lib/rails/commands/dbconsole/dbconsole_command.rb

@@ -102,7 +102,7 @@ def db_config
       # first time around to show a consistent error message to people
       # relying on 2-level database configuration.
 
-      @db_config = configurations.configs_for(env_name: environment, name: database)
+      @db_config = configurations.configs_for(env_name: environment, name: database, include_replicas: true)
 
       unless @db_config
         raise ActiveRecord::AdapterNotSpecified,

railties/test/commands/dbconsole_test.rb

@@ -226,6 +226,25 @@ def test_specifying_a_custom_database_and_environment
     end
   end
 
+  def test_specifying_a_replica_database
+    options = {
+      database: "primary_replica",
+    }
+
+    sample_config = {
+      "test" => {
+        "primary" => {},
+        "primary_replica" => {
+          "replica" => true
+        }
+      }
+    }
+
+    app_db_config(sample_config) do
+      assert_equal "primary_replica", Rails::DBConsole.new(options).db_config.name
+    end
+  end
+
   def test_specifying_a_missing_database
     app_db_config({}) do
       e = assert_raises(ActiveRecord::AdapterNotSpecified) do