[Fix rubocop#515] Fix an error for Rails/BulkChangeTable

Fixes rubocop#515 Follow up ruby/psych#487 and rails/rails@255b5ff This PR fixes an error for `Rails/BulkChangeTable` when using Psych 4.0.
koic · Jul 1, 2021 · 32d7443 · 32d7443
1 parent 7485ea7
commit 32d7443
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 ## master (unreleased)
 
+### Bug fixes
+
+* [#515](https://github.com/rubocop/rubocop-rails/issues/515): Fix an error for `Rails/BulkChangeTable` when using Psych 4.0. ([@koic][])
+
 ## 2.11.1 (2021-06-25)
 
 ### Bug fixes

diff --git a/lib/rubocop/cop/rails/bulk_change_table.rb b/lib/rubocop/cop/rails/bulk_change_table.rb
@@ -194,7 +194,11 @@ def database_from_yaml
         def database_yaml
           return nil unless File.exist?('config/database.yml')
 
-          yaml = YAML.load_file('config/database.yml')
+          yaml = if YAML.respond_to?(:unsafe_load_file)
+                   YAML.unsafe_load_file('config/database.yml')
+                 else
+                   YAML.load_file('config/database.yml')
+                 end
           return nil unless yaml.is_a? Hash
 
           config = yaml['development']