63 lines
1.4 KiB
Ruby
63 lines
1.4 KiB
Ruby
class Hbc::FakeSystemCommand
|
|
def self.responses
|
|
@responses ||= {}
|
|
end
|
|
|
|
def self.expectations
|
|
@expectations ||= {}
|
|
end
|
|
|
|
def self.system_calls
|
|
@system_calls ||= Hash.new(0)
|
|
end
|
|
|
|
def self.clear
|
|
@responses = nil
|
|
@expectations = nil
|
|
@system_calls = nil
|
|
end
|
|
|
|
def self.stubs_command(command, response='')
|
|
responses[command] = response
|
|
end
|
|
|
|
def self.expects_command(command, response='', times=1)
|
|
stubs_command(command, response)
|
|
expectations[command] = times
|
|
end
|
|
|
|
def self.verify_expectations!
|
|
expectations.each do |command, times|
|
|
unless system_calls[command] == times
|
|
fail("expected #{command.inspect} to be run #{times} times, but got #{system_calls[command]}")
|
|
end
|
|
end
|
|
end
|
|
|
|
def self.run(command, options={})
|
|
command = Hbc::SystemCommand._process_options(command, options)
|
|
unless responses.key?(command)
|
|
fail("no response faked for #{command.inspect}, faked responses are: #{responses.inspect}")
|
|
end
|
|
system_calls[command] += 1
|
|
Hbc::SystemCommand::Result.new(command, responses[command], '', 0)
|
|
end
|
|
|
|
def self.run!(command, options={})
|
|
run(command, options.merge(:must_succeed => true))
|
|
end
|
|
end
|
|
|
|
module FakeSystemCommandHooks
|
|
def after_teardown
|
|
super
|
|
Hbc::FakeSystemCommand.verify_expectations!
|
|
ensure
|
|
Hbc::FakeSystemCommand.clear
|
|
end
|
|
end
|
|
|
|
class MiniTest::Spec
|
|
include FakeSystemCommandHooks
|
|
end
|