From 8efed4ef62ff5327bff3ec222f4d7d079bc7f336 Mon Sep 17 00:00:00 2001 From: Amol Gautam Date: Thu, 24 Mar 2022 17:03:16 -0400 Subject: [PATCH] Adding awx_task_env as wrapper while invoking backend --- awx/api/views/__init__.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/awx/api/views/__init__.py b/awx/api/views/__init__.py index 37bf8cfab7..7374947952 100644 --- a/awx/api/views/__init__.py +++ b/awx/api/views/__init__.py @@ -172,6 +172,7 @@ from awx.api.views.root import ( # noqa ) from awx.api.views.webhooks import WebhookKeyView, GithubWebhookReceiver, GitlabWebhookReceiver # noqa from awx.api.pagination import UnifiedJobEventPagination +from awx.main.utils import set_environ logger = logging.getLogger('awx.api.views') @@ -1555,8 +1556,9 @@ class CredentialExternalTest(SubDetailAPIView): backend_kwargs[field_name] = value backend_kwargs.update(request.data.get('metadata', {})) try: - obj.credential_type.plugin.backend(**backend_kwargs) - return Response({}, status=status.HTTP_202_ACCEPTED) + with set_environ(**settings.AWX_TASK_ENV): + obj.credential_type.plugin.backend(**backend_kwargs) + return Response({}, status=status.HTTP_202_ACCEPTED) except requests.exceptions.HTTPError as exc: message = 'HTTP {}'.format(exc.response.status_code) return Response({'inputs': message}, status=status.HTTP_400_BAD_REQUEST)