def context(tmpdir_factory): temp_dir = tmpdir_factory.mktemp("context", numbered=True) context = Context() context.config = { 'log_dir': os.path.join(str(temp_dir), "log"), 'artifact_dir': os.path.join(str(temp_dir), "artifact"), 'work_dir': os.path.join(str(temp_dir), "work"), 'artifact_upload_timeout': 200, 'artifact_expiration_hours': 1, 'reclaim_interval': 0.001, 'task_script': ('bash', '-c', '>&2 echo bar && echo foo && exit 1'), 'task_max_timeout': .1, } context.claim_task = { 'credentials': { 'a': 'b' }, 'status': { 'taskId': 'taskId' }, 'task': { 'task_defn': True }, 'runId': 'runId', } return context
def context(tmpdir): context = Context() context.config = { 'log_dir': os.path.join(tmpdir, "log"), 'artifact_dir': os.path.join(tmpdir, "artifact"), 'task_log_dir': os.path.join(tmpdir, "artifact", "public", "logs"), 'work_dir': os.path.join(tmpdir, "work"), 'artifact_upload_timeout': 200, 'artifact_expiration_hours': 1, 'reclaim_interval': 0.001, 'task_script': ('bash', '-c', '>&2 echo bar && echo foo && exit 1'), 'task_max_timeout': .1, } context.claim_task = { 'credentials': {'a': 'b'}, 'status': {'taskId': 'taskId'}, 'task': {'dependencies': ['dependency1', 'dependency2'], 'taskGroupId': 'dependency0'}, 'runId': 'runId', } return context
def context(tmpdir): GPG_HOME = os.path.join(os.path.dirname(__file__), "data", "gpg") context_ = Context() context_.config = { "artifact_dir": ARTIFACT_DIR, "work_dir": os.path.join(tmpdir, "work"), "log_dir": os.path.join(tmpdir, "log"), "chain_of_trust_hash_algorithm": "sha256", "cot_schema_path": DEFAULT_CONFIG['cot_schema_path'], "gpg_home": GPG_HOME, "gpg_encoding": 'utf-8', "gpg_options": None, "gpg_path": os.environ.get("GPG_PATH", None), "gpg_public_keyring": "%(gpg_home)s/pubring.gpg", "gpg_secret_keyring": "%(gpg_home)s/secring.gpg", "gpg_use_agent": None, "sign_chain_of_trust": True, "worker_id": "worker_id", "worker_type": "worker_type", } context_.claim_task = { "runId": 2, "status": { "taskId": "taskId", }, "task": { 'dependencies': [], "payload": {}, "scopes": ["foo"], "taskGroupId": "taskGroupId", "workerType": "workerType", }, "workerGroup": "worker_group", "credentials": { 'c': 'd' }, } yield context_