111 lines
2.5 KiB
Ruby
111 lines
2.5 KiB
Ruby
# frozen_string_literal: true
|
|
require 'psych/helper'
|
|
|
|
module Psych
|
|
module JSON
|
|
class TestStream < TestCase
|
|
def setup
|
|
@io = StringIO.new
|
|
@stream = Psych::JSON::Stream.new(@io)
|
|
@stream.start
|
|
end
|
|
|
|
def test_explicit_documents
|
|
@io = StringIO.new
|
|
@stream = Psych::JSON::Stream.new(@io)
|
|
@stream.start
|
|
|
|
@stream.push({ 'foo' => 'bar' })
|
|
|
|
assert !@stream.finished?, 'stream not finished'
|
|
@stream.finish
|
|
assert @stream.finished?, 'stream finished'
|
|
|
|
assert_match(/^---/, @io.string)
|
|
assert_match(/\.\.\.$/, @io.string)
|
|
end
|
|
|
|
def test_null
|
|
@stream.push(nil)
|
|
assert_match(/^--- null/, @io.string)
|
|
end
|
|
|
|
def test_string
|
|
@stream.push "foo"
|
|
assert_match(/(["])foo\1/, @io.string)
|
|
end
|
|
|
|
def test_symbol
|
|
@stream.push :foo
|
|
assert_match(/(["])foo\1/, @io.string)
|
|
end
|
|
|
|
def test_int
|
|
@stream.push 10
|
|
assert_match(/^--- 10/, @io.string)
|
|
end
|
|
|
|
def test_float
|
|
@stream.push 1.2
|
|
assert_match(/^--- 1.2/, @io.string)
|
|
end
|
|
|
|
def test_hash
|
|
hash = { 'one' => 'two' }
|
|
@stream.push hash
|
|
|
|
json = @io.string
|
|
assert_match(/}$/, json)
|
|
assert_match(/^--- \{/, json)
|
|
assert_match(/["]one['"]/, json)
|
|
assert_match(/["]two['"]/, json)
|
|
end
|
|
|
|
def test_list_to_json
|
|
list = %w{ one two }
|
|
@stream.push list
|
|
|
|
json = @io.string
|
|
assert_match(/\]$/, json)
|
|
assert_match(/^--- \[/, json)
|
|
assert_match(/["]one["]/, json)
|
|
assert_match(/["]two["]/, json)
|
|
end
|
|
|
|
class Foo; end
|
|
|
|
def test_json_dump_exclude_tag
|
|
@stream << Foo.new
|
|
json = @io.string
|
|
refute_match('Foo', json)
|
|
end
|
|
|
|
class Bar
|
|
def encode_with coder
|
|
coder.represent_seq 'omg', %w{ a b c }
|
|
end
|
|
end
|
|
|
|
def test_json_list_dump_exclude_tag
|
|
@stream << Bar.new
|
|
json = @io.string
|
|
refute_match('omg', json)
|
|
end
|
|
|
|
def test_time
|
|
time = Time.utc(2010, 10, 10)
|
|
@stream.push({'a' => time })
|
|
json = @io.string
|
|
assert_match "{\"a\": \"2010-10-10 00:00:00.000000000 Z\"}\n", json
|
|
end
|
|
|
|
def test_datetime
|
|
time = Time.new(2010, 10, 10).to_datetime
|
|
@stream.push({'a' => time })
|
|
json = @io.string
|
|
assert_match "{\"a\": \"#{time.strftime("%Y-%m-%d %H:%M:%S.%9N %:z")}\"}\n", json
|
|
end
|
|
end
|
|
end
|
|
end
|