108 lines
2.7 KiB
Ruby
108 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe StubMethodCalls do
|
|
include described_class
|
|
|
|
let(:object) do
|
|
Class.new do
|
|
def self.test_method
|
|
'test'
|
|
end
|
|
|
|
def self.test_method_two(response: nil)
|
|
response || 'test_two'
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#stub_method' do
|
|
let(:method_to_stub) { :test_method }
|
|
|
|
it 'stubs the method response' do
|
|
stub_method(object, method_to_stub) { true }
|
|
|
|
expect(object.send(method_to_stub)).to eq(true)
|
|
end
|
|
|
|
context 'when calling it on an already stubbed method' do
|
|
before do
|
|
stub_method(object, method_to_stub) { false }
|
|
end
|
|
|
|
it 'stubs correctly' do
|
|
stub_method(object, method_to_stub) { true }
|
|
|
|
expect(object.send(method_to_stub)).to eq(true)
|
|
end
|
|
end
|
|
|
|
context 'methods that accept arguments' do
|
|
it 'stubs correctly' do
|
|
stub_method(object, method_to_stub) { |a, b| a + b }
|
|
|
|
expect(object.send(method_to_stub, 1, 2)).to eq(3)
|
|
end
|
|
|
|
context 'methods that use named arguments' do
|
|
let(:method_to_stub) { :test_method_two }
|
|
|
|
it 'stubs correctly' do
|
|
stub_method(object, method_to_stub) { |a: 'test'| a }
|
|
|
|
expect(object.send(method_to_stub, a: 'testing')).to eq('testing')
|
|
expect(object.send(method_to_stub)).to eq('test')
|
|
end
|
|
|
|
context 'stubbing non-existent method' do
|
|
let(:method_to_stub) { :another_method }
|
|
|
|
it 'stubs correctly' do
|
|
stub_method(object, method_to_stub) { |a: 'test'| a }
|
|
|
|
expect(object.send(method_to_stub, a: 'testing')).to eq('testing')
|
|
expect(object.send(method_to_stub)).to eq('test')
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#restore_original_method' do
|
|
before do
|
|
stub_method(object, :test_method) { true }
|
|
end
|
|
|
|
it 'restores original behaviour' do
|
|
expect(object.test_method).to eq(true)
|
|
|
|
restore_original_method(object, :test_method)
|
|
|
|
expect(object.test_method).to eq('test')
|
|
end
|
|
|
|
context 'method is not stubbed' do
|
|
specify do
|
|
expect do
|
|
restore_original_method(object, 'some_other_method')
|
|
end.to raise_error(NotImplementedError, "some_other_method has not been stubbed on #{object}")
|
|
end
|
|
end
|
|
end
|
|
|
|
describe '#restore_original_methods' do
|
|
before do
|
|
stub_method(object, :test_method) { true }
|
|
stub_method(object, :test_method_two) { true }
|
|
end
|
|
|
|
it 'restores original behaviour' do
|
|
restore_original_methods(object)
|
|
|
|
expect(object.test_method).to eq('test')
|
|
expect(object.test_method_two).to eq('test_two')
|
|
end
|
|
end
|
|
end
|