extern/hishtory
1
0
Fork 0
mirror of https://github.com/ddworken/hishtory.git synced 2024-11-28 19:23:53 +01:00
Code Issues Packages Projects Releases Wiki Activity

Shard the fuzz test

Browse Source
This commit is contained in:
David Dworken 2024-02-10 19:59:26 -08:00
parent 7fb25c22c1
commit f2c3aebe3a
No known key found for this signature in database
2 changed files with 35 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
client/client_test.go
View File

@ -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 != "" {
numTestShards, err := strconv.Atoi(numTestShardsStr)
require.NoError(t, err)
currentShardNumber, err := strconv.Atoi(currentShardNumberStr)
require.NoError(t, err)
if testShardNumber%numTestShards != currentShardNumber {
if currentShardNumberStr == "" {
return 0
}
currentShardNumber, err := strconv.Atoi(currentShardNumberStr)
if err != nil {
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")
}
}

6
client/fuzz_test.go
View File

@ -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()