module Dry::Core::Container::Stub

def stub(key, value, &block)
  unless key?(key)
    raise ::ArgumentError, "cannot stub #{key.to_s.inspect} - no such key in container"
  end
  _stubs[key.to_s] = value
  if block
    yield
    unstub(key)
  end
  self
end