tests/e2e: use WaitLeader in corrupt test

Signed-off-by: Ivan Valdes <ivan@vald.es>
ivanvc · Feb 8, 2024 · 94e7f71 · 94e7f71
1 parent 8b24932
commit 94e7f71
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/e2e/corrupt_test.go b/tests/e2e/corrupt_test.go
@@ -320,8 +320,10 @@ func TestCompactHashCheckDetectCorruptionInterrupt(t *testing.T) {
 	_, err = epc.Procs[slowCompactionNodeIndex].Logs().Expect("finished scheduled compaction")
 	require.NoError(t, err, "can't get log indicating finished scheduled compaction")
 
-	// Wait for compaction hash check
-	time.Sleep(checkTime * 5)
+	// Wait until the leader finished compaction hash check.
+	leaderIndex := epc.WaitLeader(t)
+	_, err = epc.Procs[leaderIndex].Logs().ExpectWithContext(ctx, expect.ExpectedResponse{Value: "finished compaction hash check"})
+	require.NoError(t, err, "can't get log indicating finished compaction hash check")
 
 	alarmResponse, err := cc.AlarmList()
 	require.NoError(t, err, "error on alarm list")