Index: testing/legion/task_controller.py |
diff --git a/testing/legion/task_controller.py b/testing/legion/task_controller.py |
index e0812b40f2c365d9d254952d89b7ba238db50936..df9ddbdeb56370b26b6e08e974fc59050a17edea 100644 |
--- a/testing/legion/task_controller.py |
+++ b/testing/legion/task_controller.py |
@@ -49,21 +49,18 @@ class TaskController(object): |
_task_count = 0 |
_tasks = [] |
- def __init__(self, isolate_file, config_vars, dimensions, priority=100, |
+ def __init__(self, isolated_hash, dimensions, priority=100, |
idle_timeout_secs=common_lib.DEFAULT_TIMEOUT_SECS, |
connection_timeout_secs=common_lib.DEFAULT_TIMEOUT_SECS, |
- verbosity='ERROR', name=None): |
- assert isinstance(config_vars, dict) |
+ verbosity='ERROR', name=None, run_id=None): |
assert isinstance(dimensions, dict) |
type(self)._tasks.append(self) |
type(self)._task_count += 1 |
self.verbosity = verbosity |
self._name = name or 'Task%d' % type(self)._task_count |
self._priority = priority |
- self._isolate_file = isolate_file |
- self._isolated_file = isolate_file + 'd' |
+ self._isolated_hash = isolated_hash |
self._idle_timeout_secs = idle_timeout_secs |
- self._config_vars = config_vars |
self._dimensions = dimensions |
self._connect_event = threading.Event() |
self._connected = False |
@@ -71,6 +68,10 @@ class TaskController(object): |
self._otp = self._CreateOTP() |
self._rpc = None |
+ run_id = run_id or datetime.datetime.now().strftime('%Y-%m-%d-%H-%M-%S') |
+ self._task_name = '%s/%s/%s' % ( |
+ os.path.splitext(sys.argv[0])[0], self._name, run_id) |
+ |
parser = argparse.ArgumentParser() |
parser.add_argument('--isolate-server') |
parser.add_argument('--swarming-server') |
@@ -137,7 +138,6 @@ class TaskController(object): |
"""Creates the task machine.""" |
logging.info('Creating %s', self.name) |
self._connect_event.clear() |
- self._ExecuteIsolate() |
self._ExecuteSwarming() |
def WaitForConnection(self): |
@@ -163,31 +163,15 @@ class TaskController(object): |
self._rpc = None |
self._connected = False |
- def _ExecuteIsolate(self): |
- """Executes isolate.py.""" |
- cmd = [ |
- 'python', |
- ISOLATE_PY, |
- 'archive', |
- '--isolate', self._isolate_file, |
- '--isolated', self._isolated_file, |
- ] |
- |
- if self._isolate_server: |
- cmd.extend(['--isolate-server', self._isolate_server]) |
- for key, value in self._config_vars.iteritems(): |
- cmd.extend(['--config-var', key, value]) |
- |
- self._ExecuteProcess(cmd) |
- |
def _ExecuteSwarming(self): |
"""Executes swarming.py.""" |
cmd = [ |
'python', |
SWARMING_PY, |
'trigger', |
- self._isolated_file, |
+ self._isolated_hash, |
'--priority', str(self._priority), |
+ '--task-name', self._task_name, |
] |
if self._isolate_server: |