1
0
Fork 0
mirror of https://github.com/ruby/ruby.git synced 2022-11-09 12:17:21 -05:00
ruby--ruby/test/psych/test_psych.rb

241 lines
5.8 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
require_relative 'helper'
require 'stringio'
require 'tempfile'
class TestPsych < Psych::TestCase
def teardown
Psych.domain_types.clear
end
def test_line_width_invalid
assert_raises(ArgumentError) { Psych.dump('x', { :line_width => -2 }) }
end
def test_line_width_no_limit
data = { 'a' => 'a b' * 50}
expected = "---\na: #{'a b' * 50}\n"
assert_equal(expected, Psych.dump(data, { :line_width => -1 }))
end
def test_line_width_limit
yml = Psych.dump('123456 7', { :line_width => 5 })
assert_match(/^\s*7/, yml)
end
def test_indent
yml = Psych.dump({:a => {'b' => 'c'}}, {:indentation => 5})
assert_match(/^[ ]{5}b/, yml)
end
def test_canonical
yml = Psych.dump({:a => {'b' => 'c'}}, {:canonical => true})
assert_match(/\? "b/, yml)
end
def test_header
yml = Psych.dump({:a => {'b' => 'c'}}, {:header => true})
assert_match(/YAML/, yml)
end
def test_version_array
yml = Psych.dump({:a => {'b' => 'c'}}, {:version => [1,1]})
assert_match(/1.1/, yml)
end
def test_version_string
yml = Psych.dump({:a => {'b' => 'c'}}, {:version => '1.1'})
assert_match(/1.1/, yml)
end
def test_version_bool
yml = Psych.dump({:a => {'b' => 'c'}}, {:version => true})
assert_match(/1.1/, yml)
end
def test_load_argument_error
assert_raises(TypeError) do
Psych.load nil
end
end
def test_non_existing_class_on_deserialize
e = assert_raises(ArgumentError) do
Psych.load("--- !ruby/object:NonExistent\nfoo: 1")
end
assert_equal 'undefined class/module NonExistent', e.message
end
def test_dump_stream
things = [22, "foo \n", {}]
stream = Psych.dump_stream(*things)
assert_equal things, Psych.load_stream(stream)
end
def test_dump_file
hash = {'hello' => 'TGIF!'}
Tempfile.create('fun.yml') do |io|
assert_equal io, Psych.dump(hash, io)
io.rewind
assert_equal Psych.dump(hash), io.read
end
end
def test_dump_io
hash = {'hello' => 'TGIF!'}
stringio = StringIO.new ''.dup
assert_equal stringio, Psych.dump(hash, stringio)
assert_equal Psych.dump(hash), stringio.string
end
def test_simple
assert_equal 'foo', Psych.load("--- foo\n")
end
def test_libyaml_version
assert Psych.libyaml_version
assert_equal Psych.libyaml_version.join('.'), Psych::LIBYAML_VERSION
end
def test_load_stream
docs = Psych.load_stream("--- foo\n...\n--- bar\n...")
assert_equal %w{ foo bar }, docs
end
def test_parse_stream
docs = Psych.parse_stream("--- foo\n...\n--- bar\n...")
assert_equal %w{ foo bar }, docs.children.map { |x| x.transform }
end
def test_add_builtin_type
got = nil
Psych.add_builtin_type 'omap' do |type, val|
got = val
end
Psych.load('--- !!omap hello')
assert_equal 'hello', got
ensure
Psych.remove_type 'omap'
end
def test_domain_types
got = nil
Psych.add_domain_type 'foo.bar,2002', 'foo' do |type, val|
got = val
end
Psych.load('--- !foo.bar,2002/foo hello')
assert_equal 'hello', got
Psych.load("--- !foo.bar,2002/foo\n- hello\n- world")
assert_equal %w{ hello world }, got
Psych.load("--- !foo.bar,2002/foo\nhello: world")
assert_equal({ 'hello' => 'world' }, got)
end
def test_load_file
Tempfile.create(['yikes', 'yml']) {|t|
t.binmode
t.write('--- hello world')
t.close
assert_equal 'hello world', Psych.load_file(t.path)
}
end
def test_load_file_default_return_value
Tempfile.create(['empty', 'yml']) {|t|
assert_equal false, Psych.load_file(t.path)
}
end
def test_load_file_with_fallback
Tempfile.create(['empty', 'yml']) {|t|
assert_equal 42, Psych.load_file(t.path, fallback: 42)
}
end
def test_load_file_with_fallback_nil_or_false
Tempfile.create(['empty', 'yml']) {|t|
assert_nil Psych.load_file(t.path, fallback: nil)
assert_equal false, Psych.load_file(t.path, fallback: false)
}
end
def test_load_file_with_fallback_hash
Tempfile.create(['empty', 'yml']) {|t|
assert_equal Hash.new, Psych.load_file(t.path, fallback: Hash.new)
}
end
def test_load_file_with_fallback_for_nil
Tempfile.create(['nil', 'yml']) {|t|
t.binmode
t.write('--- null')
t.close
assert_nil Psych.load_file(t.path, fallback: 42)
}
end
def test_load_file_with_fallback_for_false
Tempfile.create(['false', 'yml']) {|t|
t.binmode
t.write('--- false')
t.close
assert_equal false, Psych.load_file(t.path, fallback: 42)
}
end
def test_parse_file
Tempfile.create(['yikes', 'yml']) {|t|
t.binmode
t.write('--- hello world')
t.close
assert_equal 'hello world', Psych.parse_file(t.path).transform
}
end
def test_degenerate_strings
assert_equal false, Psych.load(' ')
assert_equal false, Psych.parse(' ')
assert_equal false, Psych.load('')
assert_equal false, Psych.parse('')
end
def test_callbacks
types = []
appender = lambda { |*args| types << args }
Psych.add_builtin_type('foo', &appender)
Psych.add_domain_type('example.com,2002', 'foo', &appender)
Psych.load <<-eoyml
- !tag:yaml.org,2002:foo bar
- !tag:example.com,2002:foo bar
eoyml
assert_equal [
["tag:yaml.org,2002:foo", "bar"],
["tag:example.com,2002:foo", "bar"]
], types
end
def test_symbolize_names
yaml = <<-eoyml
foo:
bar: baz
hoge:
- fuga: piyo
eoyml
result = Psych.load(yaml)
assert_equal result, { "foo" => { "bar" => "baz"}, "hoge" => [{ "fuga" => "piyo" }] }
result = Psych.load(yaml, symbolize_names: true)
assert_equal result, { foo: { bar: "baz" }, hoge: [{ fuga: "piyo" }] }
result = Psych.safe_load(yaml, symbolize_names: true)
assert_equal result, { foo: { bar: "baz" }, hoge: [{ fuga: "piyo" }] }
end
end