mirror of
https://gitea.invidious.io/iv-org/shard-spectator.git
synced 2024-08-15 00:53:35 +00:00
Add default stubs from parent contexts
This commit is contained in:
parent
d2e27d1e6e
commit
6c0693f2a4
2 changed files with 20 additions and 9 deletions
|
@ -11,11 +11,20 @@ module Spectator::Mocks
|
|||
@entries = {} of Key => Entry
|
||||
|
||||
def initialize(context : TestContext)
|
||||
@all_instances = context.stubs.map do |k, v|
|
||||
current_context = context
|
||||
while current_context
|
||||
current_context.stubs.each do |k, v|
|
||||
stubs = if @all_instances.has_key?(k)
|
||||
@all_instances[k].stubs
|
||||
else
|
||||
entry = Entry.new
|
||||
entry.stubs.concat(v)
|
||||
{k, entry}
|
||||
end.to_h
|
||||
@all_instances[k] = entry
|
||||
entry.stubs
|
||||
end
|
||||
stubs.concat(v)
|
||||
end
|
||||
current_context = current_context.parent?
|
||||
end
|
||||
end
|
||||
|
||||
def reset : Nil
|
||||
|
|
|
@ -3,6 +3,8 @@ require "./test_values"
|
|||
|
||||
module Spectator
|
||||
class TestContext
|
||||
getter! parent
|
||||
|
||||
getter values
|
||||
|
||||
getter stubs : Hash(String, Deque(Mocks::MethodStub))
|
||||
|
|
Loading…
Reference in a new issue