test/psych/visitors/test_depth_first.rb


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

# frozen_string_literal: false
require 'psych/helper'

module Psych
  module Visitors
    class TestDepthFirst < TestCase
      class Collector < Struct.new(:calls)
        def initialize(calls = [])
          super
        end

        def call obj
          calls << obj
        end
      end

      def test_scalar
        collector = Collector.new
        visitor   = Visitors::DepthFirst.new collector
        visitor.accept Psych.parse_stream '--- hello'

        assert_equal 3, collector.calls.length
      end

      def test_sequence
        collector = Collector.new
        visitor   = Visitors::DepthFirst.new collector
        visitor.accept Psych.parse_stream "---\n- hello"

        assert_equal 4, collector.calls.length
      end

      def test_mapping
        collector = Collector.new
        visitor   = Visitors::DepthFirst.new collector
        visitor.accept Psych.parse_stream "---\nhello: world"

        assert_equal 5, collector.calls.length
      end

      def test_alias
        collector = Collector.new
        visitor   = Visitors::DepthFirst.new collector
        visitor.accept Psych.parse_stream "--- &yay\n- foo\n- *yay\n"

        assert_equal 5, collector.calls.length
      end
    end
  end
end