mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
c7c2ad5749
In future versions of Psych, the `load` method will be mostly the same as the `safe_load` method. In other words, the `load` method won't allow arbitrary object deserialization (which can be used to escalate to an RCE). People that need to load *trusted* documents can use the `unsafe_load` method. This commit introduces the `unsafe_load` method so that people can incrementally upgrade. For example, if they try to upgrade to 4.0.0 and something breaks, they can downgrade, audit callsites, change to `safe_load` or `unsafe_load` as required, and then upgrade to 4.0.0 smoothly. https://github.com/ruby/psych/commit/cb50aa8d3f
76 lines
1.6 KiB
Ruby
76 lines
1.6 KiB
Ruby
# frozen_string_literal: true
|
|
require_relative 'helper'
|
|
|
|
module Psych
|
|
class TestOmap < TestCase
|
|
def test_parse_as_map
|
|
o = Psych.unsafe_load "--- !!omap\na: 1\nb: 2"
|
|
assert_kind_of Psych::Omap, o
|
|
assert_equal 1, o['a']
|
|
assert_equal 2, o['b']
|
|
end
|
|
|
|
def test_self_referential
|
|
map = Psych::Omap.new
|
|
map['foo'] = 'bar'
|
|
map['self'] = map
|
|
assert_equal(map, Psych.unsafe_load(Psych.dump(map)))
|
|
end
|
|
|
|
def test_keys
|
|
map = Psych::Omap.new
|
|
map['foo'] = 'bar'
|
|
assert_equal 'bar', map['foo']
|
|
end
|
|
|
|
def test_order
|
|
map = Psych::Omap.new
|
|
map['a'] = 'b'
|
|
map['b'] = 'c'
|
|
assert_equal [%w{a b}, %w{b c}], map.to_a
|
|
end
|
|
|
|
def test_square
|
|
list = [["a", "b"], ["b", "c"]]
|
|
map = Psych::Omap[*list.flatten]
|
|
assert_equal list, map.to_a
|
|
assert_equal 'b', map['a']
|
|
assert_equal 'c', map['b']
|
|
end
|
|
|
|
def test_dump
|
|
map = Psych::Omap['a', 'b', 'c', 'd']
|
|
yaml = Psych.dump(map)
|
|
assert_match('!omap', yaml)
|
|
assert_match('- a: b', yaml)
|
|
assert_match('- c: d', yaml)
|
|
end
|
|
|
|
def test_round_trip
|
|
list = [["a", "b"], ["b", "c"]]
|
|
map = Psych::Omap[*list.flatten]
|
|
assert_cycle(map)
|
|
end
|
|
|
|
def test_load
|
|
list = [["a", "b"], ["c", "d"]]
|
|
map = Psych.load(<<-eoyml)
|
|
--- !omap
|
|
- a: b
|
|
- c: d
|
|
eoyml
|
|
assert_equal list, map.to_a
|
|
end
|
|
|
|
# NOTE: This test will not work with Syck
|
|
def test_load_shorthand
|
|
list = [["a", "b"], ["c", "d"]]
|
|
map = Psych.load(<<-eoyml)
|
|
--- !!omap
|
|
- a: b
|
|
- c: d
|
|
eoyml
|
|
assert_equal list, map.to_a
|
|
end
|
|
end
|
|
end
|