Update UDP release workflow to test via batch-get-traces (#272)

* update udp release workflow to test via bath-get-traces

* run e2e test on push to main

Author: jj22ee

.github/workflows/release-udp-exporter.yml

@@ -1,4 +1,4 @@
-name: Release ADOT OTLP UDP Exporter
+name: Release ADOT X-Ray UDP Exporter
 on:
   workflow_dispatch:
     inputs:
@@ -8,14 +8,21 @@ on:
         type: string
 
 permissions:
-  id-token: write
   contents: write
 
 jobs:
-  test-and-release:
+  validate-udp-exporter-e2e-test:
+    name: "Validate X-Ray UDP Exporter E2E Test Succeeds"
+    uses: ./.github/workflows/udp-exporter-e2e-test.yml
+    secrets: inherit
+    permissions:
+      id-token: write
+
+  release-udp-exporter:
     runs-on: ubuntu-latest
+    needs: validate-udp-exporter-e2e-test
     steps:
-      - name: Checkout Contrib Repo @ SHA - ${{ github.sha }}
+      - name: Checkout Repo @ SHA - ${{ github.sha }}
         uses: actions/checkout@v4
       - name: Install Go
         uses: actions/setup-go@v5
@@ -28,51 +35,10 @@ jobs:
           cd exporters/xrayudp
           go test ./...
 
-      - name: Download and run X-Ray Daemon
+      - name: Tag UDP Exporter module
         run: |
-          mkdir xray-daemon
-          cd xray-daemon
-          wget https://s3.us-west-2.amazonaws.com/aws-xray-assets.us-west-2/xray-daemon/aws-xray-daemon-linux-3.x.zip
-          unzip aws-xray-daemon-linux-3.x.zip
-          ./xray -o -n us-west-2 -f ./daemon-logs.log --log-level debug &
-
-      - name: Build Sample App
-        run: |
-          cd .github/test-sample-apps/integ-test-http-server
-          go build -o integ-test-http-server .
-
-      - name: Run Sample App in Background
-        run: |
-          cd .github/test-sample-apps/integ-test-http-server
-          ./integ-test-http-server &
-
-      - name: Call Sample App Endpoint
-        id: call-endpoint
-        run: |
-          echo "traceId=$(curl localhost:8080/test)" >> $GITHUB_OUTPUT
-
-      - name: Verify X-Ray daemon received traces
-        run: |
-          sleep 10
-          echo "X-Ray daemon logs:"
-          cat xray-daemon/daemon-logs.log
-
-          # Check if the daemon received and processed some data
-          if grep -q "sending.*batch" xray-daemon/daemon-logs.log; then
-            echo "✅ X-Ray daemon processed trace data (AWS upload errors are expected)"
-            exit 0
-          elif grep -q "processor:.*segment" xray-daemon/daemon-logs.log; then
-            echo "✅ X-Ray daemon processed segment data (AWS upload errors are expected)"
-            exit 0
-          else
-            echo "❌ No evidence of traces being received by X-Ray daemon"
-            exit 1
-          fi
+          git tag exporters/xrayudp/v${{ inputs.udp-exporter-version }}
 
       # TODO: Uncomment when we make the first release
-      # - name: Tag UDP Exporter module
-      #   run: |
-      #     git tag exporters/xrayudp/v${{ inputs.udp-exporter-version }}
-
       # - name: Publish tag for UDP Exporter module
       #   run: git push --tags

.github/workflows/udp-exporter-e2e-test.yml

@@ -0,0 +1,65 @@
+name: Test ADOT X-Ray UDP Exporter
+on:
+  workflow_call:
+  push:
+    branches:
+      - main
+
+permissions:
+  id-token: write
+
+jobs:
+  udp-exporter-e2e-test:
+    runs-on: ubuntu-latest
+    steps:
+      - name: Checkout Repo @ SHA - ${{ github.sha }}
+        uses: actions/checkout@v4
+      - name: Install Go
+        uses: actions/setup-go@v5
+        with:
+            go-version: "1.24.0"
+            check-latest: true
+
+      - name: Configure AWS credentials for Testing Tracing
+        uses: aws-actions/configure-aws-credentials@v4
+        with:
+          role-to-assume: ${{ secrets.XRAY_UDP_EXPORTER_TEST_ROLE }}
+          aws-region: 'us-east-1'
+
+      - name: Download and run X-Ray Daemon
+        run: |
+          mkdir xray-daemon
+          cd xray-daemon
+          wget https://s3.us-east-2.amazonaws.com/aws-xray-assets.us-east-2/xray-daemon/aws-xray-daemon-linux-3.x.zip
+          unzip aws-xray-daemon-linux-3.x.zip
+          ./xray -o -n us-east-2 -f ./daemon-logs.log --log-level debug &
+
+      - name: Build Sample App
+        run: |
+          cd .github/test-sample-apps/integ-test-http-server
+          go build -o integ-test-http-server .
+
+      - name: Run Sample App in Background
+        run: |
+          cd .github/test-sample-apps/integ-test-http-server
+          ./integ-test-http-server &
+
+      - name: Call Sample App Endpoint
+        id: call-endpoint
+        run: |
+          echo "traceId=$(curl localhost:8080/test)" >> $GITHUB_OUTPUT
+
+      - name: Check if traces are successfully sent to AWS X-Ray
+        run: |
+          sleep 20
+          # # Print Daemon Logs for debugging
+          # cat xray-daemon/daemon-logs.log
+
+          traceId=${{ steps.call-endpoint.outputs.traceId }}
+          numTracesFound=$(aws xray batch-get-traces --trace-ids $traceId --region us-east-2 | jq '.Traces' | jq length)
+          if [[ numTracesFound -ne "1" ]]; then
+            echo "TraceId $traceId not found in X-Ray."
+            exit 1
+          else
+            echo "TraceId $traceId found in X-Ray."
+          fi