Sha256: 5193d2a5c7d97b5d03a99e53548eb231899292b22bf24bad12c8a03dc2f15d00
Contents?: true
Size: 1.56 KB
Versions: 20
Compression:
Stored size: 1.56 KB
Contents
# frozen_string_literal: true
module BoltSpec
module Plans
class TaskStub < ActionStub
def matches(targets, _task, arguments, options)
if @invocation[:targets] && Set.new(@invocation[:targets]) != Set.new(targets.map(&:name))
return false
end
if @invocation[:arguments] && arguments != @invocation[:arguments]
return false
end
if @invocation[:options] && options != @invocation[:options]
return false
end
true
end
def call(targets, task, arguments, options)
@calls += 1
if @return_block
# Merge arguments and options into params to match puppet function signature.
check_resultset(@return_block.call(targets: targets, task: task, params: arguments.merge(options)), task)
else
Bolt::ResultSet.new(targets.map { |target| @data[target.name] || default_for(target) })
end
end
def parameters
@invocation[:arguments] + @invocation[:options] if @invocation.include?(:arguments)
end
# Allow any data.
def result_for(target, data)
Bolt::Result.new(target, value: Bolt::Util.walk_keys(data, &:to_s))
end
# Public methods
# Restricts the stub to only match invocations with certain parameters.
# All parameters must match exactly.
def with_params(params)
@invocation[:arguments] = params.reject { |k, _v| k.start_with?('_') }
@invocation[:options] = params.select { |k, _v| k.start_with?('_') }
self
end
end
end
end
Version data entries
20 entries across 20 versions & 1 rubygems