Improve test worker check creation (#22063)

Author: AAraKKe

.github/workflows/zz-test-worker-poc.yaml

@@ -13,46 +13,14 @@ on:
         type: string
 
 jobs:
-  create-check-run:
+  setup:
     runs-on: ubuntu-latest
     permissions:
       id-token: write
       contents: read
-    outputs:
-      check_run_id: ${{ steps.create_check.outputs.result }}
-    steps:
-      - name: Get GitHub token via dd-octo-sts
-        uses: DataDog/dd-octo-sts-action@08f2144903ced3254a3dafec2592563409ba2aa0
-        id: octo-sts
-        with:
-          scope: DataDog/integrations-core
-          policy: self.test-worker-poc
-
-      - name: Create Check Run
-        id: create_check
-        uses: actions/github-script@60a0d83039c74a4aee543508d2ffcb1c3799cdea # v7.0.1
-        with:
-          github-token: ${{ steps.octo-sts.outputs.token }}
-          script: |
-            const response = await github.rest.checks.create({
-              owner: context.repo.owner,
-              repo: context.repo.repo,
-              name: 'Test Worker POC',
-              head_sha: '${{ inputs.ref }}',
-              status: 'in_progress',
-              details_url: '${{ github.server_url }}/${{ github.repository }}/actions/runs/${{ github.run_id }}',
-              output: {
-                title: 'Test Worker POC Running',
-                summary: 'The test worker POC workflow is currently running.'
-              }
-            });
-            return response.data.id;
-
-  setup:
-    needs: create-check-run
-    runs-on: ubuntu-latest
     outputs:
       work_matrix: ${{ steps.set_matrix.outputs.work_matrix }}
+      token: ${{ steps.octo-sts.outputs.token }}
 
     steps:
       - name: Check out repository
@@ -65,6 +33,13 @@ jobs:
         with:
           python-version: '3.13'
 
+      - name: Get GitHub token via dd-octo-sts
+        uses: DataDog/dd-octo-sts-action@08f2144903ced3254a3dafec2592563409ba2aa0
+        id: octo-sts
+        with:
+          scope: DataDog/integrations-core
+          policy: self.test-worker-poc
+
       - name: Parse matrix and set matrix output
         id: set_matrix
         run: |
@@ -94,36 +69,43 @@ jobs:
         with:
           ref: ${{ inputs.ref }}
 
+      - name: Create Check Run
+        id: create_check
+        uses: actions/github-script@60a0d83039c74a4aee543508d2ffcb1c3799cdea # v7.0.1
+        with:
+          github-token: ${{ needs.setup.outputs.token }}
+          script: |
+            const response = await github.rest.checks.create({
+              owner: context.repo.owner,
+              repo: context.repo.repo,
+              name: 'Test Worker POC Test - ${{ matrix.name }}',
+              head_sha: '${{ inputs.ref }}',
+              status: 'in_progress',
+              details_url: '${{ github.server_url }}/${{ github.repository }}/actions/runs/${{ github.run_id }}',
+              output: {
+                title: 'Test Worker POC Test - ${{ matrix.name }}',
+                summary: 'Test Worker POC Test - ${{ matrix.name }} is currently running.'
+              }
+            });
+            return response.data.id;
+
       - name: Run Generic Task
+        id: run_task
         run: |
           echo "Running generic logic for job: ${{ matrix.name }}"
           echo "Platform: ${{ matrix.platform }}"
           echo "Runner: ${{ matrix.runner }}"
           echo "Target: ${{ matrix.target }}"
           echo "Git SHA: $(git rev-parse HEAD)"
 
-  complete-check-run:
-    runs-on: ubuntu-latest
-    if: always()
-    needs: [create-check-run, run_work]
-    permissions:
-      id-token: write
-      contents: read
-    steps:
-      - name: Get GitHub token via dd-octo-sts
-        uses: DataDog/dd-octo-sts-action@08f2144903ced3254a3dafec2592563409ba2aa0
-        id: octo-sts
-        with:
-          scope: DataDog/integrations-core
-          policy: self.test-worker-poc
-
       - name: Complete Check Run
+        if: always()
         uses: actions/github-script@60a0d83039c74a4aee543508d2ffcb1c3799cdea # v7.0.1
         with:
-          github-token: ${{ steps.octo-sts.outputs.token }}
+          github-token: ${{ needs.setup.outputs.token }}
           script: |
-            const check_run_id = ${{ needs.create-check-run.outputs.check_run_id }};
-            const conclusion = '${{ needs.run_work.result }}';
+            const check_run_id = ${{ steps.create_check.outputs.result }};
+            const conclusion = '${{ steps.run_task.outcome }}'; // Outcome of the specific task
 
             // Map actions conclusion to check run conclusion
             // https://docs.github.com/en/rest/checks/runs?apiVersion=2022-11-28#update-a-check-run
@@ -144,8 +126,9 @@ jobs:
               check_run_id: check_run_id,
               status: 'completed',
               conclusion: checkConclusion,
+              details_url: '${{ github.server_url }}/${{ github.repository }}/actions/runs/${{ github.run_id }}',
               output: {
-                title: 'Test Worker POC Completed',
+                title: 'Test Worker POC Test - ${{ matrix.name }} Completed',
                 summary: `The workflow completed with status: ${conclusion}`
               }
             });