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()