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.")