Add delete method to clear stubs for a mock

This commit is contained in:
Michael Miller 2022-05-15 16:19:55 -06:00
parent 847fd38534
commit a39b27387c
No known key found for this signature in database
GPG key ID: 32B47AE8F388A1FF
4 changed files with 58 additions and 6 deletions

View file

@ -34,5 +34,11 @@ module Spectator
@object_stubs[key] = yield
end
end
# Clears all stubs defined for a mocked object.
def delete(object : Reference) : Nil
key = Box.box(object)
@object_stubs.delete(key)
end
end
end

View file

@ -39,6 +39,12 @@ module Spectator
end
end
# Clears all stubs defined for a mocked object.
def delete(object : T) : Nil
key = value_bytes(object)
@object_stubs.delete(key)
end
# Extracts heap-managed bytes for a value.
#
# Strings are used because a string pool is used.