File: dispatcher_test.rb

package info (click to toggle)
ruby3.4 3.4.5-1~exp1
  • links: PTS, VCS
  • area: main
  • in suites: experimental
  • size: 154,784 kB
  • sloc: ruby: 1,259,653; ansic: 829,955; yacc: 28,233; pascal: 7,359; sh: 3,864; python: 1,799; cpp: 1,158; asm: 808; makefile: 801; javascript: 414; lisp: 109; perl: 62; awk: 36; sed: 4; xml: 4
file content (46 lines) | stat: -rw-r--r-- 1,252 bytes parent folder | download
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
# frozen_string_literal: true

require_relative "../test_helper"

module Prism
  class DispatcherTest < TestCase
    class TestListener
      attr_reader :events_received

      def initialize
        @events_received = []
      end

      def on_call_node_enter(node)
        events_received << :on_call_node_enter
      end

      def on_call_node_leave(node)
        events_received << :on_call_node_leave
      end

      def on_integer_node_enter(node)
        events_received << :on_integer_node_enter
      end
    end

    def test_dispatching_events
      listener = TestListener.new
      dispatcher = Dispatcher.new
      dispatcher.register(listener, :on_call_node_enter, :on_call_node_leave, :on_integer_node_enter)

      root = Prism.parse(<<~RUBY).value
        def foo
          something(1, 2, 3)
        end
      RUBY

      dispatcher.dispatch(root)
      assert_equal([:on_call_node_enter, :on_integer_node_enter, :on_integer_node_enter, :on_integer_node_enter, :on_call_node_leave], listener.events_received)

      listener.events_received.clear
      dispatcher.dispatch_once(root.statements.body.first.body.body.first)
      assert_equal([:on_call_node_enter, :on_call_node_leave], listener.events_received)
    end
  end
end