diff options
Diffstat (limited to 'test/yaml/test_omap.rb')
-rw-r--r-- | test/yaml/test_omap.rb | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/test/yaml/test_omap.rb b/test/yaml/test_omap.rb new file mode 100644 index 0000000000..d1d30e9e93 --- /dev/null +++ b/test/yaml/test_omap.rb @@ -0,0 +1,56 @@ +require 'test/unit' +require 'yaml' + +module YAML + class TestOmap < Test::Unit::TestCase + def test_keys + map = YAML::Omap.new + map['foo'] = 'bar' + assert_equal 'bar', map['foo'] + end + + def test_order + map = YAML::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 = YAML::Omap[*list.flatten] + assert_equal list, map.to_a + assert_equal 'b', map['a'] + assert_equal 'c', map['b'] + end + + def test_to_yaml + map = YAML::Omap['a', 'b', 'c', 'd'] + yaml = map.to_yaml + 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 = YAML::Omap[*list.flatten] + loaded = YAML.load(YAML.dump(map)) + + assert_equal map, loaded + assert_equal list, loaded.to_a + end + + ### + # FIXME: Syck should also support !!omap as shorthand + def test_load + list = [["a", "b"], ["c", "d"]] + map = YAML.load(<<-eoyml) +--- !omap +- a: b +- c: d + eoyml + assert_equal list, map.to_a + end + end +end |