add job's project_name in pass_config data (#2081)

neuro-inc · Apr 19, 2023 · 73c4d33 · 73c4d33
1 parent ba0b1dd
commit 73c4d33
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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,