def test_set_pending_executor_removal(self):
agent = Agent(paw='123', sleep_min=2, sleep_max=8, watchdog=0, executors=['cmd', 'test'], platform='windows')
executor_to_remove = 'test'
want = dict(executor=executor_to_remove, action='remove')
agent.set_pending_executor_removal(executor_to_remove)
assert agent.executor_change_to_assign == want
assert agent.executors == ['cmd']
def test_removing_nonexistent_executor(self):
original_executors = ['cmd', 'test']
agent = Agent(paw='123',
sleep_min=2,
sleep_max=8,
watchdog=0,
executors=original_executors,
platform='windows')
agent.set_pending_executor_removal('idontexist')
assert agent.executor_change_to_assign is None
assert agent.executors == original_executors
def test_heartbeat_modification_during_pending_executor_removal(
self, loop):
original_executors = ['cmd', 'test']
agent = Agent(paw='123',
sleep_min=2,
sleep_max=8,
watchdog=0,
executors=original_executors,
platform='windows')
agent.set_pending_executor_removal('test')
loop.run_until_complete(
agent.heartbeat_modification(executors=original_executors))
assert agent.executors == ['cmd']
