iv-org-mirror/shard-spectator
1
0
Fork 0
mirror of https://gitea.invidious.io/iv-org/shard-spectator.git synced 2024-08-15 00:53:35 +00:00
Code Issues Projects Releases Packages Wiki Activity

Allow stubbing/overriding anonymous double methods

Browse source
This commit is contained in:
Michael Miller 2019-11-14 17:12:27 -07:00
parent b9fe1b6a60
commit e44f0229d6
2 changed files with 17 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/spectator/mocks/anonymous_double.cr
View file

@ -8,8 +8,15 @@ module Spectator::Mocks
end
macro method_missing(call)
@values.fetch({{call.name.symbolize}}) do
raise ::Spectator::Mocks::UnexpectedMessageError.new("#{self} received unexpected message {{call.name}}")
args = ::Spectator::Mocks::GenericArguments.create({{call.args.splat}})
call = ::Spectator::Mocks::GenericMethodCall.new({{call.name.symbolize}}, args)
::Spectator::Harness.current.mocks.record_call(self, call)
if (stub = ::Spectator::Harness.current.mocks.find_stub(self, call))
stub.call!(args, typeof(@values.fetch({{call.name.symbolize}}) { raise }))
else
@values.fetch({{call.name.symbolize}}) do
raise ::Spectator::Mocks::UnexpectedMessageError.new("#{self} received unexpected message {{call.name}}")
end
end
end
end

9
src/spectator/mocks/anonymous_null_double.cr
View file

@ -4,7 +4,14 @@ module Spectator::Mocks
end
macro method_missing(call)
@values.fetch({{call.name.symbolize}}) { self }
args = ::Spectator::Mocks::GenericArguments.create({{call.args.splat}})
call = ::Spectator::Mocks::GenericMethodCall.new({{call.name.symbolize}}, args)
::Spectator::Harness.current.mocks.record_call(self, call)
if (stub = ::Spectator::Harness.current.mocks.find_stub(self, call))
stub.call!(args, typeof(@values.fetch({{call.name.symbolize}}) { self }))
else
@values.fetch({{call.name.symbolize}}) { self }
end
end
end
end