From 346934089b24d4eaedfd1450ee278ab715ef150e Mon Sep 17 00:00:00 2001
From: Yasser Isa <yasserisa@users.noreply.github.com>
Date: Thu, 14 Dec 2023 09:19:47 -0700
Subject: [PATCH] CI: Add sharding to GoogleTest job

Co-authored-by: Jack Grigg <jack@electriccoin.co>
---
 .github/workflows/ci.yml | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 3d39ab2fc42..8c0992533dc 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -58,6 +58,7 @@ jobs:
     outputs:
       build_matrix: ${{ steps.set-matrices.outputs.build_matrix }}
       test_matrix: ${{ steps.set-matrices.outputs.test_matrix }}
+      test_names: ${{ steps.set-matrices.outputs.test_names }}
     steps:
       - id: set-matrices
         env:
@@ -65,6 +66,7 @@ jobs:
         run: |
           jq -r -n 'env.CFG | fromjson | @json "build_matrix=\(.data)"' >> $GITHUB_OUTPUT
           jq -r -n 'env.CFG | fromjson | [.data[] | select(.test_os)] | @json "test_matrix=\(.)"' >> $GITHUB_OUTPUT
+          jq -r -n 'env.CFG | fromjson | [.data[] | select(.test_os) | .name] | @json "test_names=\(.)"' >> $GITHUB_OUTPUT
 
   build:
     name: Build tier ${{ matrix.tier }} platform ${{ matrix.platform }}
@@ -240,12 +242,14 @@ jobs:
         run: ${{ format('./test_bitcoin{0}', matrix.file_ext) }} -p
 
   test-gtest:
-    name: GoogleTest tier ${{ matrix.tier }} platform ${{ matrix.platform }}
+    name: GoogleTest tier ${{ matrix.tier }} platform ${{ matrix.platform }} - shard ${{ matrix.gtest_shards }}
     needs: [matrices, build]
     runs-on: ${{ matrix.test_os }}
     continue-on-error: ${{ matrix.tier != 1 }}
     strategy:
       matrix:
+        name: ${{ fromJson(needs.matrices.outputs.test_names) }}
+        shard_index: [0, 1]
         include: ${{ fromJson(needs.matrices.outputs.test_matrix) }}
     steps:
       - name: Download zcash-gtest artifact
@@ -274,4 +278,7 @@ jobs:
               runner.os == 'Linux' && '.zcash-params' || 'ZcashParams'
             }}
       - name: Run GoogleTests
+        env:
+          GTEST_TOTAL_SHARDS: 2
+          GTEST_SHARD_INDEX: ${{ matrix.shard_index }}
         run: ${{ format('./zcash-gtest{0}', matrix.file_ext) }}