def test_env_var(): expected = {"name": "ENV_CONFIG", "value": "ABC"} config = TaskConfig() config.set_container_env(env_vars={"ENV_CONFIG": "ABC"}) assert ( config.as_dict()["overrides"]["containerOverrides"][0]["environment"][0] == expected )
def start(): env = os.environ input = Input(dict(env)) # pylint: disable=redefined-builtin task_params_file = env.get("INPUT_TASK_PARAMS_FILE") config = TaskConfig(task_params_file) config.set_repository(env["GITHUB_REPOSITORY"]) config.set( **input.as_dict(), group=config.repository, startedBy=env.get("GITHUB_ACTOR", "UNKNOWN"), ) config.task_count = int(env.get("INPUT_TASK_COUNT", 1)) logger.info( f"Start task execution with definition '{config.task_definition}' on cluster '{config.cluster}'" ) config.set_container_env({ "GITHUB_PERSONAL_TOKEN": input["githubPat"], "GITHUB_OWNER": env["GITHUB_REPOSITORY_OWNER"], "GITHUB_REPOSITORY": config.repository, "RUNNER_NAME": env["GITHUB_JOB"], }) config.set_capacity_provider(input.get("capacityProvider")) logger.info("Ready to execute with config:") logger.info(config.as_dict()) task = Task(config) task.run() if input.should_wait: task.wait() newline = "\n" logger.info( f"The following tasks were launched: \n{newline.join(task.task_arns)}" ) logger.info("Task(s) successfuly created.") logger.info(f"You can follow it on: {task.url}")