зеркало из https://github.com/github/ruby.git
200 строки
6.3 KiB
Ruby
200 строки
6.3 KiB
Ruby
#frozen_string_literal: false
|
|
require_relative 'test_helper'
|
|
require 'json/add/core'
|
|
require 'json/add/complex'
|
|
require 'json/add/rational'
|
|
require 'json/add/bigdecimal'
|
|
require 'json/add/ostruct'
|
|
require 'json/add/set'
|
|
require 'date'
|
|
|
|
class JSONAdditionTest < Test::Unit::TestCase
|
|
include JSON
|
|
|
|
class A
|
|
def initialize(a)
|
|
@a = a
|
|
end
|
|
|
|
attr_reader :a
|
|
|
|
def ==(other)
|
|
a == other.a
|
|
end
|
|
|
|
def self.json_create(object)
|
|
new(*object['args'])
|
|
end
|
|
|
|
def to_json(*args)
|
|
{
|
|
'json_class' => self.class.name,
|
|
'args' => [ @a ],
|
|
}.to_json(*args)
|
|
end
|
|
end
|
|
|
|
class A2 < A
|
|
def to_json(*args)
|
|
{
|
|
'json_class' => self.class.name,
|
|
'args' => [ @a ],
|
|
}.to_json(*args)
|
|
end
|
|
end
|
|
|
|
class B
|
|
def self.json_creatable?
|
|
false
|
|
end
|
|
|
|
def to_json(*args)
|
|
{
|
|
'json_class' => self.class.name,
|
|
}.to_json(*args)
|
|
end
|
|
end
|
|
|
|
class C
|
|
def self.json_creatable?
|
|
false
|
|
end
|
|
|
|
def to_json(*args)
|
|
{
|
|
'json_class' => 'JSONAdditionTest::Nix',
|
|
}.to_json(*args)
|
|
end
|
|
end
|
|
|
|
def test_extended_json
|
|
a = A.new(666)
|
|
assert A.json_creatable?
|
|
json = generate(a)
|
|
a_again = parse(json, :create_additions => true)
|
|
assert_kind_of a.class, a_again
|
|
assert_equal a, a_again
|
|
end
|
|
|
|
def test_extended_json_default
|
|
a = A.new(666)
|
|
assert A.json_creatable?
|
|
json = generate(a)
|
|
a_hash = parse(json)
|
|
assert_kind_of Hash, a_hash
|
|
end
|
|
|
|
def test_extended_json_disabled
|
|
a = A.new(666)
|
|
assert A.json_creatable?
|
|
json = generate(a)
|
|
a_again = parse(json, :create_additions => true)
|
|
assert_kind_of a.class, a_again
|
|
assert_equal a, a_again
|
|
a_hash = parse(json, :create_additions => false)
|
|
assert_kind_of Hash, a_hash
|
|
assert_equal(
|
|
{"args"=>[666], "json_class"=>"JSONAdditionTest::A"}.sort_by { |k,| k },
|
|
a_hash.sort_by { |k,| k }
|
|
)
|
|
end
|
|
|
|
def test_extended_json_fail1
|
|
b = B.new
|
|
assert !B.json_creatable?
|
|
json = generate(b)
|
|
assert_equal({ "json_class"=>"JSONAdditionTest::B" }, parse(json))
|
|
end
|
|
|
|
def test_extended_json_fail2
|
|
c = C.new
|
|
assert !C.json_creatable?
|
|
json = generate(c)
|
|
assert_raise(ArgumentError, NameError) { parse(json, :create_additions => true) }
|
|
end
|
|
|
|
def test_raw_strings
|
|
raw = ''
|
|
raw.respond_to?(:encode!) and raw.encode!(Encoding::ASCII_8BIT)
|
|
raw_array = []
|
|
for i in 0..255
|
|
raw << i
|
|
raw_array << i
|
|
end
|
|
json = raw.to_json_raw
|
|
json_raw_object = raw.to_json_raw_object
|
|
hash = { 'json_class' => 'String', 'raw'=> raw_array }
|
|
assert_equal hash, json_raw_object
|
|
assert_match(/\A\{.*\}\z/, json)
|
|
assert_match(/"json_class":"String"/, json)
|
|
assert_match(/"raw":\[0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27,28,29,30,31,32,33,34,35,36,37,38,39,40,41,42,43,44,45,46,47,48,49,50,51,52,53,54,55,56,57,58,59,60,61,62,63,64,65,66,67,68,69,70,71,72,73,74,75,76,77,78,79,80,81,82,83,84,85,86,87,88,89,90,91,92,93,94,95,96,97,98,99,100,101,102,103,104,105,106,107,108,109,110,111,112,113,114,115,116,117,118,119,120,121,122,123,124,125,126,127,128,129,130,131,132,133,134,135,136,137,138,139,140,141,142,143,144,145,146,147,148,149,150,151,152,153,154,155,156,157,158,159,160,161,162,163,164,165,166,167,168,169,170,171,172,173,174,175,176,177,178,179,180,181,182,183,184,185,186,187,188,189,190,191,192,193,194,195,196,197,198,199,200,201,202,203,204,205,206,207,208,209,210,211,212,213,214,215,216,217,218,219,220,221,222,223,224,225,226,227,228,229,230,231,232,233,234,235,236,237,238,239,240,241,242,243,244,245,246,247,248,249,250,251,252,253,254,255\]/, json)
|
|
raw_again = parse(json, :create_additions => true)
|
|
assert_equal raw, raw_again
|
|
end
|
|
|
|
MyJsonStruct = Struct.new 'MyJsonStruct', :foo, :bar
|
|
|
|
def test_core
|
|
t = Time.now
|
|
assert_equal t, JSON(JSON(t), :create_additions => true)
|
|
d = Date.today
|
|
assert_equal d, JSON(JSON(d), :create_additions => true)
|
|
d = DateTime.civil(2007, 6, 14, 14, 57, 10, Rational(1, 12), 2299161)
|
|
assert_equal d, JSON(JSON(d), :create_additions => true)
|
|
assert_equal 1..10, JSON(JSON(1..10), :create_additions => true)
|
|
assert_equal 1...10, JSON(JSON(1...10), :create_additions => true)
|
|
assert_equal "a".."c", JSON(JSON("a".."c"), :create_additions => true)
|
|
assert_equal "a"..."c", JSON(JSON("a"..."c"), :create_additions => true)
|
|
s = MyJsonStruct.new 4711, 'foot'
|
|
assert_equal s, JSON(JSON(s), :create_additions => true)
|
|
struct = Struct.new :foo, :bar
|
|
s = struct.new 4711, 'foot'
|
|
assert_raise(JSONError) { JSON(s) }
|
|
begin
|
|
raise TypeError, "test me"
|
|
rescue TypeError => e
|
|
e_json = JSON.generate e
|
|
e_again = JSON e_json, :create_additions => true
|
|
assert_kind_of TypeError, e_again
|
|
assert_equal e.message, e_again.message
|
|
assert_equal e.backtrace, e_again.backtrace
|
|
end
|
|
assert_equal(/foo/, JSON(JSON(/foo/), :create_additions => true))
|
|
assert_equal(/foo/i, JSON(JSON(/foo/i), :create_additions => true))
|
|
end
|
|
|
|
def test_utc_datetime
|
|
now = Time.now
|
|
d = DateTime.parse(now.to_s) # usual case
|
|
assert_equal d, parse(d.to_json, :create_additions => true)
|
|
d = DateTime.parse(now.utc.to_s) # of = 0
|
|
assert_equal d, parse(d.to_json, :create_additions => true)
|
|
d = DateTime.civil(2008, 6, 17, 11, 48, 32, Rational(1,24))
|
|
assert_equal d, parse(d.to_json, :create_additions => true)
|
|
d = DateTime.civil(2008, 6, 17, 11, 48, 32, Rational(12,24))
|
|
assert_equal d, parse(d.to_json, :create_additions => true)
|
|
end
|
|
|
|
def test_rational_complex
|
|
assert_equal Rational(2, 9), parse(JSON(Rational(2, 9)), :create_additions => true)
|
|
assert_equal Complex(2, 9), parse(JSON(Complex(2, 9)), :create_additions => true)
|
|
end
|
|
|
|
def test_bigdecimal
|
|
assert_equal BigDecimal('3.141', 23), JSON(JSON(BigDecimal('3.141', 23)), :create_additions => true)
|
|
assert_equal BigDecimal('3.141', 666), JSON(JSON(BigDecimal('3.141', 666)), :create_additions => true)
|
|
end if defined?(::BigDecimal)
|
|
|
|
def test_ostruct
|
|
o = OpenStruct.new
|
|
# XXX this won't work; o.foo = { :bar => true }
|
|
o.foo = { 'bar' => true }
|
|
assert_equal o, parse(JSON(o), :create_additions => true)
|
|
end
|
|
|
|
def test_set
|
|
s = Set.new([:a, :b, :c, :a])
|
|
assert_equal s, JSON.parse(JSON(s), :create_additions => true)
|
|
end
|
|
end
|