Symbolize and freeze keys when loading from YAML

ruby-i18n · Dec 13, 2021 · 5e5a092 · 5e5a092
1 parent cb4a4be
commit 5e5a092
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/lib/i18n/backend/base.rb b/lib/i18n/backend/base.rb
@@ -241,7 +241,7 @@ def load_rb(filename)
         def load_yml(filename)
           begin
             if YAML.respond_to?(:unsafe_load_file) # Psych 4.0 way
-              YAML.unsafe_load_file(filename)
+              YAML.unsafe_load_file(filename, symbolize_names: true, freeze: true)
             else
               YAML.load_file(filename)
             end

diff --git a/test/backend/simple_test.rb b/test/backend/simple_test.rb
@@ -72,7 +72,12 @@ def setup
 
   test "simple load_yml: loads data from a YAML file" do
     data = I18n.backend.send(:load_yml, "#{locales_dir}/en.yml")
-    assert_equal({ 'en' => { 'foo' => { 'bar' => 'baz' } } }, data)
+    if ::YAML.respond_to?(:unsafe_load_file)
+      assert_equal({ :en => { :foo => { :bar => 'baz' } } }, data)
+      assert_predicate data.dig(:en, :foo, :bar), :frozen?
+    else
+      assert_equal({ 'en' => { 'foo' => { 'bar' => 'baz' } } }, data)
+    end
   end
 
   test "simple load_json: loads data from a JSON file" do