databricks
diff --git a/‎.github/workflows/check-mark.yml‎
Lines changed: 0 additions & 67 deletions b/‎.github/workflows/check-mark.yml‎
Lines changed: 0 additions & 67 deletions
diff --git a/‎.github/workflows/integration-tests.yml‎
Lines changed: 26 additions & 5 deletions b/‎.github/workflows/integration-tests.yml‎
Lines changed: 26 additions & 5 deletions
@@ -3,11 +3,15 @@ name: Integration Tests
 on:
 
   pull_request:
-    branches: [ main ]
+    types: [opened, synchronize]
+
+  merge_group:
+  
 
 jobs:
-  check:
-    name: Run Check
+  write-message:
+    if: github.event_name == 'pull_request'
+    name: Write Message
     runs-on: ubuntu-latest
 
     steps:
@@ -38,8 +42,25 @@ jobs:
           Run integration tests using the corresponding workflow:
                     
           Inputs:
-          pull_request_number=${{github.event.pull_request.number}}
-          commit_sha=${{ github.event.pull_request.head.sha }}
+          PR Number:${{github.event.pull_request.number}}
+          Commit SHA:${{ github.event.pull_request.head.sha }}
           
           This check will be approved automatically on success.
           "
+
+  # The hash for the merge queue may not be the same as the hash for the PR.
+  # Auto approve the check for the merge queue to avoid running integration tests twice.
+  auto-approve:
+    if: github.event_name == 'merge_group'
+    runs-on: ubuntu-latest
+    steps:
+      - name: Mark Check
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        shell: bash
+        run: |
+            gh api -X POST -H "Accept: application/vnd.github+json" \
+              -H "X-GitHub-Api-Version: 2022-11-28" \
+              /repos/${{ github.repository }}/statuses/${{ github.sha }} \
+              -f 'state=success' \
+              -f 'context=Integration Tests Check'