test/prism/attribute_write_test.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
51
52
53
54
55
56
57
58
59
60

# frozen_string_literal: true

require_relative "test_helper"

module Prism
  class AttributeWriteTest < TestCase
    module Target
      def self.value
        2
      end

      def self.value=(value)
        2
      end

      def self.[]=(index, value)
        2
      end
    end

    def test_named_call_with_operator
      assert_attribute_write("Target.value = 1")
    end

    def test_named_call_without_operator
      assert_attribute_write("Target.value=(1)")
    end

    def test_indexed_call_with_operator
      assert_attribute_write("Target[0] = 1")
    end

    def test_indexed_call_without_operator
      refute_attribute_write("Target.[]=(0, 1)")
    end

    def test_comparison_operators
      refute_attribute_write("Target.value == 1")
      refute_attribute_write("Target.value === 1")
    end

    private

    def parse(source)
      Prism.parse(source).value.statements.body.first
    end

    def assert_attribute_write(source)
      call = parse(source)
      assert(call.attribute_write?)
      assert_equal(1, eval(source))
    end

    def refute_attribute_write(source)
      call = parse(source)
      refute(call.attribute_write?)
      refute_equal(1, eval(source))
    end
  end
end