コード例 #1
0
ファイル: monitor_agent.py プロジェクト: kowalski/feat
 def handle_agent_death(self, state, recp):
     recp = IRecipient(recp)
     task = state.handler_tasks.get(recp.key, None)
     if task:
         return task.notify_finish()
     else:
         task = self.initiate_protocol(HandleDeath, recp,
                                       state.notification_sender)
         self._register_task(recp.key, task)
         return task.notify_finish()
コード例 #2
0
ファイル: export_agent.py プロジェクト: pepribas/F3AT
 def _start_global(self, state, blackbox):
     kwargs = dict()
     if blackbox:
         kwargs['blackbox'] = blackbox
     task = state.agent.initiate_protocol(start_agent.GloballyStartAgent,
                                          state.descriptor, **kwargs)
     return task.notify_finish()
コード例 #3
0
ファイル: test_simulation_task.py プロジェクト: pepribas/F3AT
 def testTask(self):
     agent = self.get_local('agent')
     yield agent.run_tasks()
     self.assertEqual(agent.get_result("task_result1"), 18 + 18*2)
     self.assertEqual(agent.get_result("task_result2"), 42 + 42*2)
     task = yield agent.start_task(66)
     result = yield task.notify_finish()
     self.assertEqual(result, 66 + 66*2)
コード例 #4
0
ファイル: migration_agent.py プロジェクト: kowalski/feat
 def apply_migration(self, state, migration, export=None):
     '''
     apply_migration(migration) -> Apply the migration object.
     '''
     tunel = state.exports.get_by_name(export)
     task = self.initiate_protocol(ApplyMigration, tunel.recipient,
                                   migration)
     return task.notify_finish()
コード例 #5
0
 def testRetryingProtocol(self):
     d = self.cb_after(None, self.agent, 'initiate_protocol')
     task = self.agent.retrying_protocol(ErrorTask, max_retries=3)
     self.finished = task.notify_finish()
     yield d
     self.assertEqual(task.attempt, 1)
     yield self.cb_after(None, self.agent, 'initiate_protocol')
     yield self.cb_after(None, self.agent, 'initiate_protocol')
     yield self.cb_after(None, self.agent, 'initiate_protocol')
     self.assertEqual(task.attempt, task.max_retries+1)
     self.assertFailure(self.finished, DummyException)
コード例 #6
0
ファイル: export_agent.py プロジェクト: pepribas/F3AT
    def apply_migration_step(self, state, migration_or_id, step):
        migration = self._get_migration(migration_or_id)

        if isinstance(step, int):
            step = migration.get_steps()[step]
        if not isinstance(step, MigrationStep):
            raise AttributeError("Second parameter expected to be a "
                                 "MigrationStep or an index in step of"
                                 "migration, got %r instead." % step)
        task = self.initiate_protocol(ApplyMigrationStep, migration, step,
                                      state.notification_sender)
        return task.notify_finish()
コード例 #7
0
ファイル: host_agent.py プロジェクト: f3at/feat
 def start_agent(self, state, doc_id, allocation_id=None,
                 static_name=None, **kwargs):
     task = self.initiate_protocol(StartAgent, doc_id, allocation_id,
                                   kwargs=kwargs, static_name=static_name)
     return task.notify_finish()
コード例 #8
0
ファイル: host_agent.py プロジェクト: f3at/feat
 def resolve_missing_shard_agent_problem(self, state, host_recipients):
     task = state.medium.initiate_protocol(
         problem.CollectiveSolver, MissingShard(self), host_recipients)
     return task.notify_finish()