File: and_call_original_spec.rb

package info (click to toggle)
ruby-spy 1.0.1-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 360 kB
  • sloc: ruby: 3,101; makefile: 2
file content (152 lines) | stat: -rw-r--r-- 5,004 bytes parent folder | download | duplicates (2)
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
require 'spec_helper'

describe "and_call_through" do
  context "on a partial mock object" do
    let(:klass) do
      Class.new do
        def meth_1
          :original
        end

        def meth_2(x)
          yield x, :additional_yielded_arg
        end

        def self.new_instance
          new
        end
      end
    end

    let(:instance) { klass.new }

    it 'passes the received message through to the original method' do
      spy = Spy.on(instance, :meth_1).and_call_through
      expect(instance.meth_1).to eq(:original)
      expect(spy).to have_been_called
    end

    it 'passes args and blocks through to the original method' do
      spy = Spy.on(instance, :meth_2).and_call_through
      value = instance.meth_2(:submitted_arg) { |a, b| [a, b] }
      expect(value).to eq([:submitted_arg, :additional_yielded_arg])
      expect(spy).to have_been_called
    end

    it 'works for singleton methods' do
      def instance.foo; :bar; end
      spy = Spy.on(instance, :foo).and_call_through
      expect(instance.foo).to eq(:bar)
      expect(spy).to have_been_called
    end

    it 'works for methods added through an extended module' do
      instance.extend Module.new { def foo; :bar; end }
      spy = Spy.on(instance, :foo).and_call_through
      expect(instance.foo).to eq(:bar)
      expect(spy).to have_been_called
    end

    it "works for method added through an extended module onto a class's ancestor" do
      sub_sub_klass = Class.new(Class.new(klass))
      klass.extend Module.new { def foo; :bar; end }
      spy = Spy.on(sub_sub_klass, :foo).and_call_through
      expect(sub_sub_klass.foo).to eq(:bar)
      expect(spy).to have_been_called
    end

    it "finds the method on the most direct ancestor even if the method " +
       "is available on more distant ancestors" do
      klass.extend Module.new { def foo; :klass_bar; end }
      sub_klass = Class.new(klass)
      sub_klass.extend Module.new { def foo; :sub_klass_bar; end }
      spy = Spy.on(sub_klass, :foo).and_call_through
      expect(sub_klass.foo).to eq(:sub_klass_bar)
      expect(spy).to have_been_called
    end

    it 'works for class methods defined on a superclass' do
      subclass = Class.new(klass)
      spy = Spy.on(subclass, :new_instance).and_call_through
      expect(subclass.new_instance).to be_a(subclass)
      expect(spy).to have_been_called
    end

    it 'works for class methods defined on a grandparent class' do
      sub_subclass = Class.new(Class.new(klass))
      spy = Spy.on(sub_subclass, :new_instance).and_call_through
      expect(sub_subclass.new_instance).to be_a(sub_subclass)
      expect(spy).to have_been_called
    end

    it 'works for class methods defined on the Class class' do
      spy = Spy.on(klass, :new).and_call_through
      expect(klass.new).to be_an_instance_of(klass)
      expect(spy).to have_been_called
    end

    it "works for instance methods defined on the object's class's superclass" do
      subclass = Class.new(klass)
      inst = subclass.new
      spy = Spy.on(inst, :meth_1).and_call_through
      expect(inst.meth_1).to eq(:original)
      expect(spy).to have_been_called
    end

    it 'works for aliased methods' do
      klass = Class.new do
        class << self
          alias alternate_new new
        end
      end

      spy = Spy.on(klass, :alternate_new).and_call_through
      expect(klass.alternate_new).to be_an_instance_of(klass)
      expect(spy).to have_been_called
    end

    context 'on an object that defines method_missing' do
      before do
        klass.class_eval do
          def respond_to_missing?(name, _)
            if name.to_s == "greet_jack"
              true
            else
              super
            end
          end

          def method_missing(name, *args)
            if name.to_s == "greet_jack"
              "Hello, jack"
            else
              super
            end
          end
        end
      end

      it 'works when the method_missing definition handles the message' do
        spy = Spy.on(instance, :greet_jack).and_call_through
        expect(instance.greet_jack).to eq("Hello, jack")
        expect(spy).to have_been_called
      end

      it 'raises an error on invocation if method_missing does not handle the message' do
        Spy::Subroutine.new(instance, :not_a_handled_message).hook(force: true).and_call_through

        # Note: it should raise a NoMethodError (and usually does), but
        # due to a weird rspec-expectations issue (see #183) it sometimes
        # raises a `NameError` when a `be_xxx` predicate matcher has been
        # recently used. `NameError` is the superclass of `NoMethodError`
        # so this example will pass regardless.
        # If/when we solve the rspec-expectations issue, this can (and should)
        # be changed to `NoMethodError`.
        expect {
          instance.not_a_handled_message
        }.to raise_error(NameError, /not_a_handled_message/)
      end
    end
  end
end