summaryrefslogtreecommitdiff
path: root/test/psych/json/test_stream.rb
blob: 45d0219aad5649fb8a4ff97752ff2bab58b71414 (plain)
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
require_relative '../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.000000000Z\"}\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