From 7fbf57aff8bdeacfec085307568bb96044ba4d7c Mon Sep 17 00:00:00 2001
From: Siddharth Kshetrapal <siddharthkp@github.com>
Date: Tue, 5 Nov 2024 18:12:52 +0100
Subject: [PATCH] Update check-for-integration-result.yml

---
 .../check-for-integration-result.yml          | 26 ++++++++++++++-----
 1 file changed, 19 insertions(+), 7 deletions(-)

diff --git a/.github/workflows/check-for-integration-result.yml b/.github/workflows/check-for-integration-result.yml
index f2c5d6f9d9c..a5ce1bc3acb 100644
--- a/.github/workflows/check-for-integration-result.yml
+++ b/.github/workflows/check-for-integration-result.yml
@@ -16,15 +16,27 @@ jobs:
         with:
           github-token: ${{ secrets.GITHUB_TOKEN }}
           script: |
-            const result = await github.rest.issues.listComments({
+            const issue = {
               issue_number: context.issue.number,
               owner: context.repo.owner,
               repo: context.repo.repo
-            });
-            const integrationComments = result.data.filter(c => c.user.login == 'primer-integration[bot]') // && c.body.includes('test-result'))
+            }
+          
+            const result = await github.rest.issues.listComments(issue);
+            const integrationComments = result.data.filter(c => c.user.login == 'primer-integration[bot]' && c.body.includes('<!-- test-result'))
             console.log(integrationComments)
+            if (integrationComments.length === 0) return
 
-            const latestComment = integrationComments.sort(c => c.createdAt)[0]
-            if (latestComment) {
-              // add label
-            }
+            const latestComment = integrationComments.pop()
+            const pass = latestComment.includes(':green-circle:')
+            
+            await github.rest.issues.addLabels({
+              ...issue,
+              labels: [pass ? 'integration-tests: passing' : 'integration-tests: failing'],
+            })
+
+            await github.rest.issues.removeLabel({
+              ...issue,
+              name: ['integration-tests: recommended'],
+            })
+