From 9ea6844b571b3530921fe1a9cd66bcd5bd399d47 Mon Sep 17 00:00:00 2001
From: musa-asad <musaasad@amazon.com>
Date: Fri, 13 Dec 2024 12:29:54 -0500
Subject: [PATCH] Add TA.

---
 .github/workflows/e2e-test.yml     | 3 +++
 .github/workflows/eks-e2e-test.yml | 6 ++++++
 2 files changed, 9 insertions(+)

diff --git a/.github/workflows/e2e-test.yml b/.github/workflows/e2e-test.yml
index ff945689c2..214eca29dd 100644
--- a/.github/workflows/e2e-test.yml
+++ b/.github/workflows/e2e-test.yml
@@ -92,6 +92,7 @@ jobs:
       CWA_GITHUB_TEST_REPO_BRANCH: ${{ steps.set-outputs.outputs.CWA_GITHUB_TEST_REPO_BRANCH }}
       ECR_INTEGRATION_TEST_REPO: ${{ steps.set-outputs.outputs.ECR_INTEGRATION_TEST_REPO }}
       ECR_OPERATOR_REPO: ${{ steps.set-outputs.outputs.ECR_OPERATOR_REPO }}
+      ECR_TARGET_ALLOCATOR_REPO: ${{ steps.set-outputs.outputs.ECR_TARGET_ALLOCATOR_REPO }}
     steps:
       - uses: actions/checkout@v3
         with:
@@ -111,6 +112,7 @@ jobs:
           echo "::set-output name=CWA_GITHUB_TEST_REPO_BRANCH::${{ env.CWA_GITHUB_TEST_REPO_BRANCH }}"
           echo "::set-output name=ECR_INTEGRATION_TEST_REPO::cwagent-integration-test"
           echo "::set-output name=ECR_OPERATOR_REPO::$(echo "${{ vars.ECR_OPERATOR_STAGING_REPO }}" | awk -F'/' '{print $NF}')"
+          echo "::set-output name=ECR_TARGET_ALLOCATOR_REPO::$(echo "${{ vars.ECR_TARGET_ALLOCATOR_STAGING_REPO }}" | awk -F'/' '{print $NF}')"
 
       - name: Echo test variables
         run: |
@@ -119,6 +121,7 @@ jobs:
           echo "CWA_GITHUB_TEST_REPO_BRANCH: ${{ steps.set-outputs.outputs.CWA_GITHUB_TEST_REPO_BRANCH }}"
           echo "ECR_INTEGRATION_TEST_REPO: ${{ steps.set-outputs.outputs.ECR_INTEGRATION_TEST_REPO }}"
           echo "ECR_OPERATOR_REPO: ${{ steps.set-outputs.outputs.ECR_OPERATOR_REPO }}"
+          echo "ECR_TARGET_ALLOCATOR_REPO: ${{ steps.set-outputs.outputs.ECR_TARGET_ALLOCATOR_REPO }}"
 
   GenerateTestMatrix:
     needs: [BuildAgent, BuildOperator]
diff --git a/.github/workflows/eks-e2e-test.yml b/.github/workflows/eks-e2e-test.yml
index 0d537cbe8c..31edb07b0a 100644
--- a/.github/workflows/eks-e2e-test.yml
+++ b/.github/workflows/eks-e2e-test.yml
@@ -36,6 +36,9 @@ on:
       cloudwatch_agent_operator_repository:
         required: true
         type: string
+      cloudwatch_agent_target_allocator_repository:
+        required: false
+        type: string
       cloudwatch_agent_operator_tag:
         required: true
         type: string
@@ -128,6 +131,9 @@ jobs:
               -var="cloudwatch_agent_operator_repository_url=${{ steps.login-ecr.outputs.registry }}" \
               -var="cloudwatch_agent_operator_repository=${{ inputs.cloudwatch_agent_operator_repository }}" \
               -var="cloudwatch_agent_operator_tag=${{ inputs.cloudwatch_agent_operator_tag }}" \
+              -var="cloudwatch_agent_target_allocator_repository_url=${{ steps.login-ecr.outputs.registry }}" \
+              -var="cloudwatch_agent_target_allocator_repository=${{ inputs.cloudwatch_agent_target_allocator_repository }}" \
+              -var="cloudwatch_agent_target_allocator_tag=${{ inputs.cloudwatch_agent_operator_tag }}" \
               -var="test_dir=${{ matrix.arrays.test_dir }}" \
               -var="agent_config=${{ inputs.agent_config }}" \
               -var="prometheus_config=${{ inputs.prometheus_config }}" \