From 0f686637a0e0523366b2481f32c4d93ddda7903a Mon Sep 17 00:00:00 2001
From: Min Xia <xiami@amazon.com>
Date: Tue, 15 Oct 2024 17:13:43 -0700
Subject: [PATCH] fix lambda canary workflow

---
 .github/workflows/node-lambda-test.yml   | 4 ++--
 .github/workflows/python-lambda-test.yml | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/.github/workflows/node-lambda-test.yml b/.github/workflows/node-lambda-test.yml
index acd885b15..cb75d3fdd 100644
--- a/.github/workflows/node-lambda-test.yml
+++ b/.github/workflows/node-lambda-test.yml
@@ -97,13 +97,13 @@ jobs:
         run: echo ARTIFACTS_DIR="${{ github.workspace }}/lambda_artifacts" >> $GITHUB_ENV
 
       - name: Download Lambda Layer and Function artifacts for E2E Test
-        if: ${{ github.event.inputs.CALLER_WORKFLOW_NAME != 'appsignals-node-e2e-lambda-canary-test' }}
+        if: ${{ env.CALLER_WORKFLOW_NAME != 'appsignals-node-e2e-lambda-canary-test' }}
         run: |
           aws s3 cp s3://${{ env.STAGING_S3_BUCKET }}/layer-${{ github.run_id }}.zip ${{ env.ARTIFACTS_DIR }}/layer.zip |
           aws s3 cp s3://${{ env.STAGING_S3_BUCKET }}/function-${{ github.run_id }}.zip ${{ env.ARTIFACTS_DIR }}/jsfunction.zip
 
       - name: Set Canary Environment Variable
-        if: ${{ github.event.inputs.CALLER_WORKFLOW_NAME == 'appsignals-node-e2e-lambda-canary-test' }}
+        if: ${{ env.CALLER_WORKFLOW_NAME == 'appsignals-node-e2e-lambda-canary-test' }}
         run: | 
           echo IS_CANARY=true >> $GITHUB_ENV |
           aws s3 cp {{ env.SAMPLE_APP_ZIP }} ${{ env.ARTIFACTS_DIR }}/jsfunction.zip
diff --git a/.github/workflows/python-lambda-test.yml b/.github/workflows/python-lambda-test.yml
index e0a86ab05..4a9574f7e 100644
--- a/.github/workflows/python-lambda-test.yml
+++ b/.github/workflows/python-lambda-test.yml
@@ -71,7 +71,7 @@ jobs:
       - uses: actions/checkout@v4
         with:
           repository: 'aws-observability/aws-application-signals-test-framework'
-          ref: ${{ inputs.caller-workflow-name == 'main-build' && 'main' || github.ref }}
+          ref: ${{ env.CALLER_WORKFLOW_NAME == 'main-build' && 'main' || github.ref }}
           fetch-depth: 0
 
         # We initialize Gradlew Daemon early on during the workflow because sometimes initialization
@@ -112,13 +112,13 @@ jobs:
         run: echo ARTIFACTS_DIR="${{ github.workspace }}/lambda_artifacts" >> $GITHUB_ENV
 
       - name: Download Lambda Layer and Function artifacts for E2E Test
-        if: ${{ github.event.inputs.CALLER_WORKFLOW_NAME != 'appsignals-python-e2e-lambda-canary-test' }}
+        if: ${{ env.CALLER_WORKFLOW_NAME != 'appsignals-python-e2e-lambda-canary-test' }}
         run: |
           aws s3 cp s3://${{ env.STAGING_S3_BUCKET }}/layer-${{ github.run_id }}.zip ${{ env.ARTIFACTS_DIR }}/layer.zip |
           aws s3 cp s3://${{ env.STAGING_S3_BUCKET }}/function-${{ github.run_id }}.zip ${{ env.ARTIFACTS_DIR }}/pyfunction.zip
 
       - name: Set Canary Environment Variable
-        if: ${{ github.event.inputs.CALLER_WORKFLOW_NAME == 'appsignals-python-e2e-lambda-canary-test' }}
+        if: ${{ env.CALLER_WORKFLOW_NAME == 'appsignals-python-e2e-lambda-canary-test' }}
         run: |
           echo IS_CANARY=true >> $GITHUB_ENV |
           aws s3 cp {{ env.SAMPLE_APP_ZIP }} ${{ env.ARTIFACTS_DIR }}/pyfunction.zip