def test_slavedir_good(self):
"""Test checking valid buildslave directory."""
# patch open() to return file with valid buildslave tac contents
self.setUpOpen("Application('buildslave')")
# check that isBuildslaveDir() flags directory as good
self.assertTrue(base.isBuildslaveDir("testdir"))
# check that open() was called with correct path
self.open.assert_called_once_with(self.tac_file_path)
def test_read_error(self):
"""Test that read() errors on buildbot.tac file are handled."""
# patch open() to return file object that raises IOError on read()
self.setUpReadError(1, "read-error", "dummy")
# check that isBuildslaveDir() flags directory as invalid
self.assertFalse(base.isBuildslaveDir("testdir"))
# check that correct error message was printed to stdout
self.assertReadErrorMessage("read-error")
# check that open() was called with correct path
self.open.assert_called_once_with(self.tac_file_path)
def test_open_error(self):
"""Test that open() errors are handled."""
# patch open() to raise IOError
self.setUpOpenError(1, "open-error", "dummy")
# check that isBuildslaveDir() flags directory as invalid
self.assertFalse(base.isBuildslaveDir("testdir"))
# check that correct error message was printed to stdout
self.assertReadErrorMessage("open-error")
# check that open() was called with correct path
self.open.assert_called_once_with(self.tac_file_path)
def stop(config, signame="TERM"):
quiet = config['quiet']
basedir = config['basedir']
if not base.isBuildslaveDir(basedir):
return 1
try:
stopSlave(basedir, quiet, signame)
except SlaveNotRunning:
if not quiet:
log.msg("buildslave not running")
return 0
def test_unexpected_tac_contents(self):
"""Test that unexpected contents in buildbot.tac is handled."""
# patch open() to return file with unexpected contents
self.setUpOpen("dummy-contents")
# check that isBuildslaveDir() flags directory as invalid
self.assertFalse(base.isBuildslaveDir("testdir"))
# check that correct error message was printed to the log
self.assertLogged(
re.escape("unexpected content in '%s'" % self.tac_file_path),
"invalid buildslave directory 'testdir'",
"unexpected error message on stdout")
# check that open() was called with correct path
self.open.assert_called_once_with(self.tac_file_path)
def restart(config):
quiet = config['quiet']
basedir = config['basedir']
if not base.isBuildslaveDir(basedir):
return 1
try:
stop.stopSlave(basedir, quiet)
except stop.SlaveNotRunning:
if not quiet:
log.msg("no old buildslave process found to stop")
if not quiet:
log.msg("now restarting buildslave process..")
return start.startSlave(basedir, quiet, config['nodaemon'])
def upgradeSlave(config):
basedir = os.path.expanduser(config['basedir'])
if not base.isBuildslaveDir(basedir):
return 1
buildbot_tac = open(os.path.join(basedir, "buildbot.tac")).read()
new_buildbot_tac = buildbot_tac.replace(
"from buildbot.slave.bot import BuildSlave",
"from buildslave.bot import BuildSlave")
if new_buildbot_tac != buildbot_tac:
open(os.path.join(basedir, "buildbot.tac"), "w").write(new_buildbot_tac)
log.msg("buildbot.tac updated")
else:
log.msg("No changes made")
return 0
def startCommand(config):
basedir = config["basedir"]
if not base.isBuildslaveDir(basedir):
return 1
return startSlave(basedir, config["quiet"], config["nodaemon"])
