Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
tree: 972deb535f
Fetching contributors…

Cannot retrieve contributors at this time

155 lines (124 sloc) 5.416 kb
module RSpec
module Mocks
class Proxy
class << self
def warn_about_expectations_on_nil
defined?(@warn_about_expectations_on_nil) ? @warn_about_expectations_on_nil : true
end
def warn_about_expectations_on_nil=(new_value)
@warn_about_expectations_on_nil = new_value
end
def allow_message_expectations_on_nil
@warn_about_expectations_on_nil = false
# ensure nil.rspec_verify is called even if an expectation is not set in the example
# otherwise the allowance would effect subsequent examples
RSpec::Mocks::space.add(nil) unless RSpec::Mocks::space.nil?
end
def allow_message_expectations_on_nil?
!warn_about_expectations_on_nil
end
end
def initialize(object, name=nil, options={})
@object = object
@name = name
@error_generator = ErrorGenerator.new object, name, options
@expectation_ordering = OrderGroup.new @error_generator
@messages_received = []
@options = options
@already_proxied_respond_to = false
@null_object = false
end
def null_object?
@null_object
end
def respond_to?(sym, incl_private=false)
null_object? ? true : super
end
# Tells the object to ignore any messages that aren't explicitly set as
# stubs or message expectations.
def as_null_object
@null_object = true
@object
end
def already_proxied_respond_to # :nodoc:
@already_proxied_respond_to = true
end
def already_proxied_respond_to? # :nodoc:
@already_proxied_respond_to
end
def add_message_expectation(location, method_name, opts={}, &block)
method_double[method_name].add_expectation @error_generator, @expectation_ordering, location, opts, &block
end
def add_negative_message_expectation(location, method_name, &implementation)
method_double[method_name].add_negative_expectation @error_generator, @expectation_ordering, location, &implementation
end
def add_stub(location, method_name, opts={}, &implementation)
method_double[method_name].add_stub @error_generator, @expectation_ordering, location, opts, &implementation
end
def remove_stub(method_name)
method_double[method_name].remove_stub
end
def verify #:nodoc:
method_doubles.each {|d| d.verify}
ensure
reset
end
def reset
method_doubles.each {|d| d.reset}
end
def received_message?(method_name, *args, &block)
@messages_received.any? {|array| array == [method_name, args, block]}
end
def has_negative_expectation?(method_name)
method_double[method_name].expectations.detect {|expectation| expectation.negative_expectation_for?(method_name)}
end
def record_message_received(method_name, *args, &block)
@messages_received << [method_name, args, block]
end
def message_received(method_name, *args, &block)
expectation = find_matching_expectation(method_name, *args)
stub = find_matching_method_stub(method_name, *args)
if (stub && expectation && expectation.called_max_times?) || (stub && !expectation)
if expectation = find_almost_matching_expectation(method_name, *args)
expectation.advise(*args) unless expectation.expected_messages_received?
end
stub.invoke(*args, &block)
elsif expectation
expectation.invoke(*args, &block)
elsif expectation = find_almost_matching_expectation(method_name, *args)
expectation.advise(*args) if null_object? unless expectation.expected_messages_received?
raise_unexpected_message_args_error(expectation, *args) unless (has_negative_expectation?(method_name) or null_object?)
elsif @object.is_a?(Class)
@object.superclass.send(method_name, *args, &block)
else
@object.__send__(:method_missing, method_name, *args, &block)
end
end
def raise_unexpected_message_args_error(expectation, *args)
@error_generator.raise_unexpected_message_args_error(expectation, *args)
end
def raise_unexpected_message_error(method_name, *args)
@error_generator.raise_unexpected_message_error method_name, *args
end
private
def method_double
@method_double ||= Hash.new {|h,k|
h[k] = MethodDouble.new(@object, k, self)
}
end
def method_doubles
method_double.values
end
def find_matching_expectation(method_name, *args)
method_double[method_name].expectations.find {|expectation| expectation.matches?(method_name, *args) && !expectation.called_max_times?} ||
method_double[method_name].expectations.find {|expectation| expectation.matches?(method_name, *args)}
end
def find_almost_matching_expectation(method_name, *args)
method_double[method_name].expectations.find {|expectation| expectation.matches_name_but_not_args(method_name, *args)}
end
def find_matching_method_stub(method_name, *args)
method_double[method_name].stubs.find {|stub| stub.matches?(method_name, *args)}
end
end
end
end
Jump to Line
Something went wrong with that request. Please try again.