2
0
Fork 0
mirror of https://github.com/inspec/inspec synced 2024-12-29 06:23:15 +00:00
inspec/lib/vulcano/backend/mock.rb
Dominik Richter 0125bcace8 add call tracing to mock backend
Signed-off-by: Dominik Richter <dominik.richter@gmail.com>
2015-09-22 01:42:26 +02:00

101 lines
2.4 KiB
Ruby

# encoding: utf-8
module Vulcano::Backends
Struct.new("MockCommand", :stdout, :stderr, :exit_status)
class Mock < Vulcano.backend(1)
name 'mock'
attr_accessor :files, :commands
def initialize(conf)
@conf = conf
@files = {}
@commands = {}
trace_calls unless @conf[:quiet]
end
def file(path)
@files[path] ||= File.new(self, path)
end
def run_command(cmd)
@commands[cmd] || Command.new(self, cmd)
end
def mock_command(stdout, stderr, exit_status)
Struct::MockCommand.new(stdout, stderr, exit_status)
end
def to_s
'Mock Backend Runner'
end
private
def trace_calls
interface_methods = {
'Vulcano::Backends::Mock' => Vulcano::Backends::Mock.instance_methods(false),
'Vulcano::Backends::Mock::File' => FileCommon.instance_methods(false),
}
set_trace_func proc{|event, file, line, id, binding, classname|
next unless
classname.to_s.start_with?('Vulcano::Backends::Mock') and
event == 'call' and
interface_methods[classname.to_s].include?(id)
# kindly borrowed from the wonderful simple-tracer by matugm
arg_names = eval("method(__method__).parameters.map { |arg| arg[1].to_s }",binding)
args = eval("#{arg_names}.map { |arg| eval(arg) }",binding).join(', ')
prefix = '-' * (classname.to_s.count(':') - 2) + '> '
puts "#{prefix}#{id} #{args}"
}
end
end
class Mock
class File < FileCommon
def initialize(runtime, path)
@path = path
# mock dataset
@exists = (rand < 0.8) ? true : false
@is_file = (@exists && rand < 0.7) ? true : false
@size = 0
@content = ''
if @exists && @is_file
@size = (rand**3 * 1000).to_i
@size = 0 if rand < 0.2
end
if @size > 0
@content = (0...50).map { ('a'..'z').to_a[rand(26)] }.join
end
end
def size
@size
end
def content
@content
end
def file?
@is_file
end
def exists?
@exists
end
end
class Command
attr_accessor :stdout, :stderr, :exit_status
def initialize(runtime, cmd)
@exit_code = (rand < 0.7) ? 0 : (100 * rand).to_i
@stdout = (0...50).map { ('a'..'z').to_a[rand(26)] }.join
@stderr = (0...50).map { ('a'..'z').to_a[rand(26)] }.join
end
end
end
end