diff --git a/lntest/harness.go b/lntest/harness.go index 70ab825c..7b8f4036 100644 --- a/lntest/harness.go +++ b/lntest/harness.go @@ -1279,7 +1279,7 @@ func (n *NetworkHarness) sendCoins(ctx context.Context, amt btcutil.Amount, // the target node's unconfirmed balance reflects the expected balance // and exit. if !confirmed { - expectedBalance := initialBalance.UnconfirmedBalance + int64(amt) + expectedBalance := btcutil.Amount(initialBalance.UnconfirmedBalance) + amt return target.WaitForBalance(expectedBalance, false) } @@ -1290,7 +1290,7 @@ func (n *NetworkHarness) sendCoins(ctx context.Context, amt btcutil.Amount, return err } - expectedBalance := initialBalance.ConfirmedBalance + int64(amt) + expectedBalance := btcutil.Amount(initialBalance.ConfirmedBalance) + amt return target.WaitForBalance(expectedBalance, true) } diff --git a/lntest/node.go b/lntest/node.go index 55581367..8cd0d6f1 100644 --- a/lntest/node.go +++ b/lntest/node.go @@ -915,7 +915,7 @@ func (hn *HarnessNode) WaitForBlockchainSync(ctx context.Context) error { // WaitForBalance waits until the node sees the expected confirmed/unconfirmed // balance within their wallet. -func (hn *HarnessNode) WaitForBalance(expectedBalance int64, confirmed bool) error { +func (hn *HarnessNode) WaitForBalance(expectedBalance btcutil.Amount, confirmed bool) error { ctx := context.Background() req := &lnrpc.WalletBalanceRequest{} @@ -928,11 +928,11 @@ func (hn *HarnessNode) WaitForBalance(expectedBalance int64, confirmed bool) err if confirmed { lastBalance = btcutil.Amount(balance.ConfirmedBalance) - return balance.ConfirmedBalance == expectedBalance + return btcutil.Amount(balance.ConfirmedBalance) == expectedBalance } lastBalance = btcutil.Amount(balance.UnconfirmedBalance) - return balance.UnconfirmedBalance == expectedBalance + return btcutil.Amount(balance.UnconfirmedBalance) == expectedBalance } err := WaitPredicate(doesBalanceMatch, 30*time.Second)