improve flakiness of a few tests

api/test/util.go

@@ -8,6 +8,7 @@ import (
 
 	"github.com/filecoin-project/go-address"
 	"github.com/filecoin-project/lotus/chain/types"
+	"github.com/filecoin-project/lotus/miner"
 )
 
 func SendFunds(ctx context.Context, t *testing.T, sender TestNode, addr address.Address, amount abi.TokenAmount) {
@@ -34,3 +35,30 @@ func SendFunds(ctx context.Context, t *testing.T, sender TestNode, addr address.
 		t.Fatal("did not successfully send money")
 	}
 }
+
+func MineUntilBlock(ctx context.Context, t *testing.T, sn TestStorageNode, cb func()) {
+	for i := 0; i < 1000; i++ {
+		var success bool
+		var err error
+		wait := make(chan struct{})
+		sn.MineOne(ctx, miner.MineReq{
+			Done: func(win bool, e error) {
+				success = win
+				err = e
+				wait <- struct{}{}
+			},
+		})
+		<-wait
+		if err != nil {
+			t.Fatal(err)
+		}
+		if success {
+			if cb != nil {
+				cb()
+			}
+			return
+		}
+		t.Log("did not mine block, trying again", i)
+	}
+	t.Fatal("failed to mine 1000 times in a row...")
+}