def test_get_task_bucket(scopes, expected, raises): task = {'scopes': scopes} config = {'bucket_config': {'dep': ''}, 'taskcluster_scope_prefix': 'project:releng:beetmover:'} if raises: with pytest.raises(ScriptWorkerTaskException): get_task_bucket(task, config) else: assert expected == get_task_bucket(task, config)
def test_get_task_bucket(scopes, expected, raises): task = {"scopes": scopes} config = { "bucket_config": { "dep": "" }, "taskcluster_scope_prefix": "project:releng:beetmover:" } if raises: with pytest.raises(ScriptWorkerTaskException): get_task_bucket(task, config) else: assert expected == get_task_bucket(task, config)
async def async_main(context): for module in ("botocore", "boto3", "chardet"): logging.getLogger(module).setLevel(logging.INFO) setup_mimetypes() validate_task_schema(context) connector = aiohttp.TCPConnector( limit=context.config["aiohttp_max_connections"]) async with aiohttp.ClientSession(connector=connector) as session: context.session = session # determine the task bucket and action context.bucket = get_task_bucket(context.task, context.config) context.action = get_task_action(context.task, context.config, valid_actions=action_map.keys()) if action_map.get(context.action): await action_map[context.action](context) else: log.critical("Unknown action {}!".format(context.action)) sys.exit(3) log.info("Success!")
async def async_main(context): # determine the task and make a quick validation check against its schema context.task = get_task(context.config) # e.g. $cfg['work_dir']/task.json validate_task_schema(context) # determine the task bucket and action context.bucket = get_task_bucket(context.task, context.config) context.action = get_task_action(context.task, context.config) if action_map.get(context.action): await action_map[context.action](context) else: log.critical("Unknown action {}!".format(context.action)) sys.exit(3) log.info('Success!')