Shard the fuzz test

client/client_test.go

@@ -60,15 +60,37 @@ var shellTesters []shellTester = []shellTester{bashTester{}, zshTester{}}
 
 var shardNumberAllocator int = 0
 
-func markTestForSharding(t *testing.T, testShardNumber int) {
+func numTestShards() int {
 	numTestShardsStr := os.Getenv("NUM_TEST_SHARDS")
+	if numTestShardsStr == "" {
+		return 0
+	}
+	numTestShards, err := strconv.Atoi(numTestShardsStr)
+	if err != nil {
+		panic(fmt.Errorf("failed to parse NUM_TEST_SHARDS: %v", err))
+	}
+	return numTestShards
+}
+
+func currentShardNumber() int {
 	currentShardNumberStr := os.Getenv("CURRENT_SHARD_NUM")
-	if numTestShardsStr != "" && currentShardNumberStr != "" {
+	if currentShardNumberStr == "" {
-		numTestShards, err := strconv.Atoi(numTestShardsStr)
+		return 0
-		require.NoError(t, err)
+	}
-		currentShardNumber, err := strconv.Atoi(currentShardNumberStr)
+	currentShardNumber, err := strconv.Atoi(currentShardNumberStr)
-		require.NoError(t, err)
+	if err != nil {
-		if testShardNumber%numTestShards != currentShardNumber {
+		panic(fmt.Errorf("failed to parse CURRENT_SHARD_NUM: %v", err))
+	}
+	return currentShardNumber
+}
+
+func isShardedTestRun() bool {
+	return numTestShards() != 0 && currentShardNumber() != 0
+}
+
+func markTestForSharding(t *testing.T, testShardNumber int) {
+	if isShardedTestRun() {
+		if testShardNumber%numTestShards() == currentShardNumber() {
 			t.Skip("Skipping sharded test")
 		}
 	}

client/fuzz_test.go

@@ -142,6 +142,12 @@ func FuzzTestMultipleUsers(f *testing.F) {
 	if skipSlowTests() {
 		f.Skip("skipping slow tests")
 	}
+	if isShardedTestRun() {
+		if currentShardNumber() == 0 {
+			f.Skip("Skipping sharded test")
+		}
+	}
+
 	s := os.Getenv("SPLIT_TESTS")
 	if s != "" && s != "BASIC" {
 		f.Skip()