From 414894348aa9e2e98c6eaa17f9257edd23e6e10f Mon Sep 17 00:00:00 2001
From: Olaoluwa Osuntokun <laolu32@gmail.com>
Date: Thu, 7 Nov 2024 18:54:33 -0800
Subject: [PATCH] sweep: update BudgetInputSet.Budget() to factor in extra
 budget

In this commit, we update the `Budget()` call to factor in the
`extraBudget` value. Otherwise, when we go to intialize the fee
function, we won't factor in the extra budget, and will determine that
we can't broadcast/bump.
---
 sweep/tx_input_set.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/sweep/tx_input_set.go b/sweep/tx_input_set.go
index 3a95fff2f..ce144a8eb 100644
--- a/sweep/tx_input_set.go
+++ b/sweep/tx_input_set.go
@@ -373,7 +373,9 @@ func (b *BudgetInputSet) Budget() btcutil.Amount {
 		budget += input.params.Budget
 	}
 
-	return budget
+	// We'll also tack on the extra budget which will eventually be
+	// accounted for by the wallet txns when we're broadcasting.
+	return budget + b.extraBudget
 }
 
 // DeadlineHeight returns the deadline height of the set.