def tick_daemon(self, *args):
"""
Call all the functions that the daemon would call if running once and
return.
"""
from aiida.daemon.tasks import manual_tick_all
manual_tick_all()
def test_workflow_fast_kill(self):
from aiida.cmdline.commands.workflow import Workflow as WfCmd
params = dict()
params['nmachine'] = 2
# Create a workflow with 2 subworkflows and start it
head_wf = WFTestSimpleWithSubWF()
head_wf.start()
# Get the user
dbuser = User.search_for_users(email=self.user_email)[0]
wfl = get_workflow_list(user=dbuser)
running_no = 0
for w in get_workflow_list(user=dbuser, all_states=True):
if w.get_aiida_class().get_state() == wf_states.RUNNING:
running_no += 1
self.assertEquals(running_no, 3,
"Only 3 running workflows should be found")
# Killing the head workflow
wf_cmd = WfCmd()
wf_cmd.workflow_kill(*[str(head_wf.pk), '-f', '-q'])
# At this point no running workflow should be found
running_no = 0
for w in get_workflow_list(user=dbuser, all_states=True):
if w.get_aiida_class().get_state() == wf_states.RUNNING:
running_no += 1
self.assertEquals(running_no, 0,
"No running workflows should be found")
self.assertNotEquals(get_all_running_steps(), 0,
"At this point there will be running steps.")
# Making the daemon to advance. This will automatically set
# to FINISHED all the running steps that are (directly) under
# a finished workflow
manual_tick_all()
self.assertEquals(len(list(get_all_running_steps())), 0,
"At this point there should be no running steps.")
running_no = 0
for w in get_workflow_list(user=dbuser, all_states=True):
if w.get_aiida_class().get_state() == wf_states.RUNNING:
running_no += 1
self.assertEquals(
running_no, 0, "At this point there should be "
"no running workflows.")
# Make the daemon to advance a bit more and make sure that no
# workflows resurrect.
for _ in range(5):
manual_tick_all()
self.assertEquals(len(list(get_all_running_steps())), 0,
"At this point there should be no running steps.")
running_no = 0
for w in get_workflow_list(user=dbuser, all_states=True):
if w.get_aiida_class().get_state() == wf_states.RUNNING:
running_no += 1
self.assertEquals(
running_no, 0, "At this point there should be "
"no running workflows.")