Merge pull request #3007 from halseth/lnd-test-balance-flake

lnd_test: check Dave's balance in WaitNoErr
This commit is contained in:
Olaoluwa Osuntokun 2019-04-24 17:05:12 -07:00 committed by GitHub
commit 7ba1344468
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -7984,16 +7984,25 @@ func testDataLossProtection(net *lntest.NetworkHarness, t *harnessTest) {
mineBlocks(t, net, 1, 1) mineBlocks(t, net, 1, 1)
assertNodeNumChannels(t, dave, 0) 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)
} }
} }