From 73c4d33e1b70f9d55bc1d0ed004eaefc8c56de47 Mon Sep 17 00:00:00 2001 From: Yevhenii Semendiak Date: Thu, 20 Apr 2023 00:32:54 +0300 Subject: [PATCH] add job's project_name in pass_config data (#2081) --- platform_api/orchestrator/jobs_service.py | 4 +++- tests/unit/test_job_service.py | 1 + 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/platform_api/orchestrator/jobs_service.py b/platform_api/orchestrator/jobs_service.py index bc71d0661..7242505db 100644 --- a/platform_api/orchestrator/jobs_service.py +++ b/platform_api/orchestrator/jobs_service.py @@ -191,6 +191,7 @@ async def _setup_pass_config( cluster_name: str, org_name: Optional[str], job_request: JobRequest, + project_name: Optional[str] = None, ) -> JobRequest: if NEURO_PASSED_CONFIG in job_request.container.env: raise JobsServiceException( @@ -206,6 +207,7 @@ async def _setup_pass_config( "cluster": cluster_name, "org_name": org_name, "url": str(self._api_base_url), + "project_name": project_name, } ).encode() ).decode() @@ -291,7 +293,7 @@ async def create_job( if pass_config: job_request = await self._setup_pass_config( - user, cluster_name, org_name, job_request + user, cluster_name, org_name, job_request, project_name ) record = JobRecord.create( diff --git a/tests/unit/test_job_service.py b/tests/unit/test_job_service.py index 3ed48812e..15ba79d56 100644 --- a/tests/unit/test_job_service.py +++ b/tests/unit/test_job_service.py @@ -337,6 +337,7 @@ async def test_create_job_pass_config( assert passed_data["token"] == f"token-{test_user.name}" assert passed_data["cluster"] == original_job.cluster_name assert passed_data["org_name"] == original_job.org_name + assert passed_data["project_name"] == original_job.project_name token_uri = f"token://{original_job.cluster_name}/job/{original_job.id}" assert mock_auth_client.grants[0] == ( test_user.name,