spec/ruby/library/yaml: Test YAML.unsafe_load instead of YAML.load in…

… 3.1
ruby · Jun 2, 2021 · 14bfab6 · 14bfab6
1 parent 665b4d3
commit 14bfab6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/mspec/matchers/match_yaml.rb b/lib/mspec/matchers/match_yaml.rb
@@ -30,7 +30,11 @@ def clean_yaml(yaml)
   def valid_yaml?(obj)
     require 'yaml'
     begin
-      YAML.load(obj)
+      if defined?(YAML.unsafe_load)
+        YAML.unsafe_load(obj)
+      else
+        YAML.load(obj)
+      end
     rescue
       false
     else