def test_process_host_action_unblock(self):
self._remove_site_process_host_action()
host = models.Host.objects.create(plan=self._plan, host=self.hosts[0],
blocked=True)
assert host.blocked
rpc_utils._process_host_action(host, failure_actions.HostAction.UNBLOCK)
host = models.Host.objects.get(id=host.id)
self.assertFalse(host.blocked)
self.god.check_playback()
def test_process_host_action_site(self):
self._remove_site_process_host_action
action = object()
failure_actions.HostAction.values.append(action)
host = models.Host.objects.create(plan=self._plan, host=self.hosts[0])
self.assertRaises(AssertionError, rpc_utils._process_host_action,
host, action)
self.god.check_playback()
self._called = False
def _site_process_host_action(host, action):
self._called = True
return True
self._replace_site_process_host_action(_site_process_host_action)
rpc_utils._process_host_action(host, action)
self.assertTrue(self._called)
self.god.check_playback()
