Pull mock initialization code into initialize

This commit is contained in:
Jack Nagel 2014-06-10 20:03:57 -05:00
parent 65673d60c1
commit df7756aeb9

View File

@ -4,9 +4,14 @@ require 'yaml'
class UpdaterTests < Test::Unit::TestCase
class UpdaterMock < ::Updater
def initialize(*args)
super
@outputs = Hash.new { |h, k| h[k] = [] }
@expected = []
@called = []
end
def in_repo_expect(cmd, output = '')
@outputs ||= Hash.new { |h,k| h[k] = [] }
@expected ||= []
@expected << cmd
@outputs[cmd] << output
end
@ -14,7 +19,6 @@ class UpdaterTests < Test::Unit::TestCase
def `(cmd, *args)
cmd = "#{cmd} #{args*' '}".strip
if @expected.include?(cmd) and !@outputs[cmd].empty?
@called ||= []
@called << cmd
@outputs[cmd].shift
else