Skip to content

Commit

Permalink
or you could just make __it private
Browse files Browse the repository at this point in the history
  • Loading branch information
dchelimsky committed Mar 18, 2010
1 parent 650cd24 commit a9f8a85
Showing 1 changed file with 13 additions and 11 deletions.
24 changes: 13 additions & 11 deletions lib/rspec/mocks/spec_methods.rb
Original file line number Diff line number Diff line change
Expand Up @@ -21,23 +21,17 @@ module ExampleMethods
# stub_person.name => "Joe"
# stub_person.email => "joe@domain.com"
def double(*args)
__declare_double('Double', *args)
declare_double('Double', *args)
end

# Alias for double
# Just like double, but use double
def mock(*args)
__declare_double('Mock', *args)
declare_double('Mock', *args)
end

# Alias for double
# Just like double, but use double
def stub(*args)
__declare_double('Stub', *args)
end

def __declare_double(declared_as, *args) # :nodoc:
args << {} unless Hash === args.last
args.last[:__declared_as] = declared_as
Rspec::Mocks::Mock.new(*args)
declare_double('Stub', *args)
end

# Disables warning messages about expectations being set on nil.
Expand All @@ -48,6 +42,14 @@ def allow_message_expectations_on_nil
Proxy.allow_message_expectations_on_nil
end

private

def declare_double(declared_as, *args) # :nodoc:
args << {} unless Hash === args.last
args.last[:__declared_as] = declared_as
Rspec::Mocks::Mock.new(*args)
end

end
end
end

0 comments on commit a9f8a85

Please sign in to comment.