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

Fix clearing stubs for mocks

Browse source 
Prevent reverting initial defaults set by mock macro.
This commit is contained in:
Michael Miller 2022-05-15 16:21:05 -06:00
parent a39b27387c
commit 4f0e2f6e46
No known key found for this signature in database
GPG key ID: 32B47AE8F388A1FF
1 changed files with 15 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/spectator/mocks/mock.cr
View file

@ -14,13 +14,19 @@ module Spectator
include ::Spectator::Mocked include ::Spectator::Mocked
{% begin %} {% begin %}
private getter _spectator_stubs = [ private getter(_spectator_stubs) do
{% for key, value in value_methods %} [
::Spectator::ValueStub.new({{key.id.symbolize}}, {{value}}), {% for key, value in value_methods %}
{% end %} ::Spectator::ValueStub.new({{key.id.symbolize}}, {{value}}),
] of ::Spectator::Stub {% end %}
] of ::Spectator::Stub
end
{% end %} {% end %}
def _spectator_clear_stubs : Nil
@_spectator_stubs = nil
end
# Returns the mock's name formatted for user output. # Returns the mock's name formatted for user output.
private def _spectator_stubbed_name : String private def _spectator_stubbed_name : String
\{% if anno = @type.annotation(::Spectator::StubbedName) %} \{% if anno = @type.annotation(::Spectator::StubbedName) %}
@ -73,6 +79,10 @@ module Spectator
end end
end end
def _spectator_clear_stubs : Nil
@@_spectator_mock_registry.delete(self)
end
# Returns the mock's name formatted for user output. # Returns the mock's name formatted for user output.
private def _spectator_stubbed_name : String private def _spectator_stubbed_name : String
\{% if anno = @type.annotation(::Spectator::StubbedName) %} \{% if anno = @type.annotation(::Spectator::StubbedName) %}