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

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

* run udp exporter 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:
@@ -9,12 +9,22 @@ on:
 env:
   RUBY_VERSION: "3.2"
 
+permissions:
+  id-token: write
+
 jobs:
-  build:
-    environment: 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
 
       # Install ruby and bundle dependencies and cache!
@@ -33,46 +43,4 @@ jobs:
           bundle install
           bundle exec rake test
 
-      - name: Download and run X-Ray Daemon
-        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: Setup Sample App
-        run: |
-          cd .github/test-sample-apps/integ-test-http-server/ruby-on-rails
-          bundle install
-
-      - name: Run Sample App in Background
-        run: |
-          cd .github/test-sample-apps/integ-test-http-server/ruby-on-rails
-          LISTEN_ADDRESS=127.0.0.1:8080 rails server &
-          # Wait for test server to initialize
-          sleep 5
-
-      - 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
-
       # TODO: Publish OTLP UDP Exporter to RubyGems

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

@@ -0,0 +1,72 @@
+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
+
+      # Install ruby and bundle dependencies and cache!
+      - name: Install Ruby ${{ env.RUBY_VERSION }} with dependencies
+        uses: ruby/[email protected]
+        with:
+          ruby-version: "${{ env.RUBY_VERSION }}"
+          working-directory: "exporter/xray-udp"
+          bundler: "latest"
+          bundler-cache: true
+          cache-version: "v1-ruby-xray-udp"
+
+      - 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: Setup Sample App
+        run: |
+          cd .github/test-sample-apps/integ-test-http-server/ruby-on-rails
+          bundle install
+
+      - name: Run Sample App in Background
+        run: |
+          cd .github/test-sample-apps/integ-test-http-server/ruby-on-rails
+          LISTEN_ADDRESS=127.0.0.1:8080 rails server &
+          # Wait for test server to initialize
+          sleep 5
+
+      - 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