lib/mcollective/vendor/json/tests/test_json_addition.rb

   1 #!/usr/bin/env ruby
   2 # -*- coding:utf-8 -*-
   3
   4 require 'test/unit'
   5 require File.join(File.dirname(__FILE__), 'setup_variant')
   6 require 'json/add/core'
   7 require 'json/add/complex'
   8 require 'json/add/rational'
   9 require 'date'
  10
  11 class TC_JSONAddition < Test::Unit::TestCase
  12   include JSON
  13
  14   class A
  15     def initialize(a)
  16       @a = a
  17     end
  18
  19     attr_reader :a
  20
  21     def ==(other)
  22       a == other.a
  23     end
  24
  25     def self.json_create(object)
  26       new(*object['args'])
  27     end
  28
  29     def to_json(*args)
  30       {
  31         'json_class'  => self.class.name,
  32         'args'        => [ @a ],
  33       }.to_json(*args)
  34     end
  35   end
  36
  37   class A2 < A
  38     def to_json(*args)
  39       {
  40         'json_class'  => self.class.name,
  41         'args'        => [ @a ],
  42       }.to_json(*args)
  43     end
  44   end
  45
  46   class B
  47     def self.json_creatable?
  48       false
  49     end
  50
  51     def to_json(*args)
  52       {
  53         'json_class'  => self.class.name,
  54       }.to_json(*args)
  55     end
  56   end
  57
  58   class C
  59     def self.json_creatable?
  60       false
  61     end
  62
  63     def to_json(*args)
  64       {
  65         'json_class'  => 'TC_JSONAddition::Nix',
  66       }.to_json(*args)
  67     end
  68   end
  69
  70   def test_extended_json
  71     a = A.new(666)
  72     assert A.json_creatable?
  73     json = generate(a)
  74     a_again = JSON.parse(json, :create_additions => true)
  75     assert_kind_of a.class, a_again
  76     assert_equal a, a_again
  77   end
  78
  79   def test_extended_json_default
  80     a = A.new(666)
  81     assert A.json_creatable?
  82     json = generate(a)
  83     a_hash = JSON.parse(json)
  84     assert_kind_of Hash, a_hash
  85   end
  86
  87   def test_extended_json_disabled
  88     a = A.new(666)
  89     assert A.json_creatable?
  90     json = generate(a)
  91     a_again = JSON.parse(json, :create_additions => true)
  92     assert_kind_of a.class, a_again
  93     assert_equal a, a_again
  94     a_hash = JSON.parse(json, :create_additions => false)
  95     assert_kind_of Hash, a_hash
  96     assert_equal(
  97       {"args"=>[666], "json_class"=>"TC_JSONAddition::A"}.sort_by { |k,| k },
  98       a_hash.sort_by { |k,| k }
  99     )
 100   end
 101
 102   def test_extended_json_fail1
 103     b = B.new
 104     assert !B.json_creatable?
 105     json = generate(b)
 106     assert_equal({ "json_class"=>"TC_JSONAddition::B" }, JSON.parse(json))
 107   end
 108
 109   def test_extended_json_fail2
 110     c = C.new
 111     assert !C.json_creatable?
 112     json = generate(c)
 113     assert_raises(ArgumentError, NameError) { JSON.parse(json, :create_additions => true) }
 114   end
 115
 116   def test_raw_strings
 117     raw = ''
 118     raw.respond_to?(:encode!) and raw.encode!(Encoding::ASCII_8BIT)
 119     raw_array = []
 120     for i in 0..255
 121       raw << i
 122       raw_array << i
 123     end
 124     json = raw.to_json_raw
 125     json_raw_object = raw.to_json_raw_object
 126     hash = { 'json_class' => 'String', 'raw'=> raw_array }
 127     assert_equal hash, json_raw_object
 128     assert_match(/\A\{.*\}\Z/, json)
 129     assert_match(/"json_class":"String"/, json)
 130     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)
 131     raw_again = JSON.parse(json, :create_additions => true)
 132     assert_equal raw, raw_again
 133   end
 134
 135   MyJsonStruct = Struct.new 'MyJsonStruct', :foo, :bar
 136
 137   def test_core
 138     t = Time.now
 139     assert_equal t, JSON(JSON(t), :create_additions => true)
 140     d = Date.today
 141     assert_equal d, JSON(JSON(d), :create_additions => true)
 142     d = DateTime.civil(2007, 6, 14, 14, 57, 10, Rational(1, 12), 2299161)
 143     assert_equal d, JSON(JSON(d), :create_additions => true)
 144     assert_equal 1..10, JSON(JSON(1..10), :create_additions => true)
 145     assert_equal 1...10, JSON(JSON(1...10), :create_additions => true)
 146     assert_equal "a".."c", JSON(JSON("a".."c"), :create_additions => true)
 147     assert_equal "a"..."c", JSON(JSON("a"..."c"), :create_additions => true)
 148     s = MyJsonStruct.new 4711, 'foot'
 149     assert_equal s, JSON(JSON(s), :create_additions => true)
 150     struct = Struct.new :foo, :bar
 151     s = struct.new 4711, 'foot'
 152     assert_raises(JSONError) { JSON(s) }
 153     begin
 154       raise TypeError, "test me"
 155     rescue TypeError => e
 156       e_json = JSON.generate e
 157       e_again = JSON e_json, :create_additions => true
 158       assert_kind_of TypeError, e_again
 159       assert_equal e.message, e_again.message
 160       assert_equal e.backtrace, e_again.backtrace
 161     end
 162     assert_equal(/foo/, JSON(JSON(/foo/), :create_additions => true))
 163     assert_equal(/foo/i, JSON(JSON(/foo/i), :create_additions => true))
 164   end
 165
 166   def test_utc_datetime
 167     now = Time.now
 168     d = DateTime.parse(now.to_s, :create_additions => true)                    # usual case
 169     assert_equal d, JSON.parse(d.to_json, :create_additions => true)
 170     d = DateTime.parse(now.utc.to_s)                # of = 0
 171     assert_equal d, JSON.parse(d.to_json, :create_additions => true)
 172     d = DateTime.civil(2008, 6, 17, 11, 48, 32, Rational(1,24))
 173     assert_equal d, JSON.parse(d.to_json, :create_additions => true)
 174     d = DateTime.civil(2008, 6, 17, 11, 48, 32, Rational(12,24))
 175     assert_equal d, JSON.parse(d.to_json, :create_additions => true)
 176   end
 177
 178   def test_rational_complex
 179     assert_equal Rational(2, 9), JSON.parse(JSON(Rational(2, 9)), :create_additions => true)
 180     assert_equal Complex(2, 9), JSON.parse(JSON(Complex(2, 9)), :create_additions => true)
 181   end
 182 end