lnd_test: check Dave's balance in WaitNoErr

Resolves a flake.

lnd_test.go

@@ -7945,16 +7945,25 @@ func testDataLossProtection(net *lntest.NetworkHarness, t *harnessTest) {
 	mineBlocks(t, net, 1, 1)
 	assertNodeNumChannels(t, dave, 0)
 
-	ctxt, _ = context.WithTimeout(ctxb, defaultTimeout)
+	err = lntest.WaitNoError(func() error {
-	daveBalResp, err := dave.WalletBalance(ctxt, balReq)
+		ctxt, _ = context.WithTimeout(ctxb, defaultTimeout)
-	if err != nil {
+		daveBalResp, err := dave.WalletBalance(ctxt, balReq)
-		t.Fatalf("unable to get dave's balance: %v", err)
+		if err != nil {
-	}
+			return fmt.Errorf("unable to get dave's balance: %v",
+				err)
+		}
 
-	daveBalance := daveBalResp.ConfirmedBalance
+		daveBalance := daveBalResp.ConfirmedBalance
-	if daveBalance <= daveStartingBalance {
+		if daveBalance <= daveStartingBalance {
-		t.Fatalf("expected dave to have balance above %d, intead had %v",
+			return fmt.Errorf("expected dave to have balance "+
-			daveStartingBalance, daveBalance)
+				"above %d, intead had %v", daveStartingBalance,
+				daveBalance)
+		}
+
+		return nil
+	}, time.Second*15)
+	if err != nil {
+		t.Fatalf("%v", err)
 	}
 }