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

Ensure stubs defined with allow syntax are cleared

Browse source
This commit is contained in:
Michael Miller 2022-10-09 15:48:00 -06:00
parent 2516803b0d
commit 090c95b162
No known key found for this signature in database
GPG key ID: 32B47AE8F388A1FF
3 changed files with 29 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/spectator/mocks/allow.cr
View file

@ -1,3 +1,4 @@
require "../harness"
require "./stub"
require "./stubbable"
require "./stubbed_type"
@ -21,6 +22,7 @@ module Spectator
# Applies a stub to the targeted stubbable object.
def to(stub : Stub) : Nil
@target._spectator_define_stub(stub)
Harness.current?.try &.cleanup { @target._spectator_remove_stub(stub) }
end
end
end