itests: Wait for both workers in TestWindowPostWorkerDisconnected

filecoin-project · Mar 6, 2023 · 9c2f8ee · 9c2f8ee
1 parent b0ebdb6
commit 9c2f8ee
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/itests/worker_test.go b/itests/worker_test.go
@@ -408,6 +408,13 @@ func TestWindowPostWorkerDisconnected(t *testing.T) {
 	var goodWorker kit.TestWorker
 	ens.Worker(miner, &goodWorker, kit.WithTaskTypes([]sealtasks.TaskType{sealtasks.TTGenerateWindowPoSt}), kit.ThroughRPC()).Start()
 
+	// wait for all workers
+	require.Eventually(t, func() bool {
+		w, err := miner.WorkerStats(ctx)
+		require.NoError(t, err)
+		return len(w) == 3 // 2 post + 1 miner-builtin
+	}, 10*time.Second, 100*time.Millisecond)
+
 	tryDl := func(dl uint64) {
 		p, err := miner.ComputeWindowPoSt(ctx, dl, types.EmptyTSK)
 		require.NoError(t, err)