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

Add stub when using expect().to receive()

Browse source 
This appears to match RSpec's behavior.
This commit is contained in:
Michael Miller 2021-07-10 20:28:51 -06:00
parent 384fb3093f
commit 23fd075d51
No known key found for this signature in database
GPG key ID: FB9F12F7C646A4AD
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/spectator/mocks/registry.cr
View file

@ -79,11 +79,15 @@ module Spectator::Mocks
end
def expect(object, stub : MethodStub) : Nil
fetch_instance(object).expected.add(stub)
entry = fetch_instance(object)
entry.expected.add(stub)
entry.stubs.unshift(stub)
end
def expect(type : T.class, stub : MethodStub) : Nil forall T
fetch_type(type).expected.add(stub)
entry = fetch_type(type)
entry.expected.add(stub)
entry.stubs.unshift(stub)
end
private def fetch_instance(object)