def test_switch_command_non_existent(self):
try:
master.switch_command('non_existent')
self.fail(msg="An unknown command should throw a KeyError")
except KeyError:
pass
def test_switch_command_help(self):
try:
master.switch_command('help')
except KeyError:
self.fail(msg="The master help command does not exist")
def test_switch_command_worker_status(self):
try:
master.switch_command('worker_status')
except KeyError:
self.fail(msg="The master worker_status command does not exist")
def test_switch_command_start_swarm(self):
try:
master.switch_command('start_swarm')
except KeyError:
self.fail(msg="The master start_swarm command does not exist")
def test_switch_command_init(self):
try:
master.switch_command('init')
except KeyError:
self.fail(msg="The master init command does not exist")