Merge pull request rails#50089 from fatkodima/fix-console-with-database-url

Fix rails console to work with `DATABASE_URL` for PostgreSQL

Author: byroot

activerecord/lib/active_record/railtie.rb

@@ -88,7 +88,9 @@ class Railtie < Rails::Railtie # :nodoc:
 
     initializer "active_record.postgresql_time_zone_aware_types" do
       ActiveSupport.on_load(:active_record_postgresqladapter) do
-        ActiveRecord::Base.time_zone_aware_types << :timestamptz
+        ActiveSupport.on_load(:active_record) do
+          ActiveRecord::Base.time_zone_aware_types << :timestamptz
+        end
       end
     end
 

railties/test/application/runner_test.rb

@@ -148,5 +148,14 @@ def self.count
 
       assert_match "42", rails("runner", "puts Task.count")
     end
+
+    def test_works_with_database_url
+      db_name = use_postgresql
+      previous_url = ENV["DATABASE_URL"]
+      ENV["DATABASE_URL"] = "postgres://localhost/#{db_name}"
+      assert_equal "1", rails("runner", "print 1")
+    ensure
+      ENV["DATABASE_URL"] = previous_url
+    end
   end
 end