Sha256: 7410052706a379147c2bd21f3212efc119cb5911d953a7d7c8e14cef5085296c
Contents?: true
Size: 1.7 KB
Versions: 38
Compression:
Stored size: 1.7 KB
Contents
# frozen_string_literal: true
module BoltSpec
module Plans
class ScriptStub < ActionStub
def matches(targets, _script, 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, script, arguments, options)
@calls += 1
if @return_block
# Merge arguments and options into params to match puppet function signature.
params = options.merge('arguments' => arguments)
check_resultset(@return_block.call(targets: targets, script: script, params: params), script)
else
Bolt::ResultSet.new(targets.map { |target| @data[target.name] || default_for(target) })
end
end
def parameters
@invocation[:params]
end
def result_for(target, stdout: '', stderr: '')
value = {
'stdout' => stdout,
'stderr' => stderr,
'merged_output' => "#{stdout}\n#{stderr}".strip,
'exit_code' => 0
}
Bolt::Result.for_command(target, value, 'script', '', [])
end
# Public methods
def with_params(params)
@invocation[:params] = params
@invocation[:arguments] = params['arguments']
@invocation[:options] = params.select { |k, _v| k.start_with?('_') }
.transform_keys { |k| k.sub(/^_/, '').to_sym }
self
end
end
end
end
Version data entries
38 entries across 38 versions & 1 rubygems