diff --git a/lntest/harness.go b/lntest/harness.go index bec7eb4c..9d878f54 100644 --- a/lntest/harness.go +++ b/lntest/harness.go @@ -1301,7 +1301,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) } @@ -1312,7 +1312,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 613d47ec..65415277 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)