diff --git a/contractcourt/chain_arbitrator_test.go b/contractcourt/chain_arbitrator_test.go index 9e90864d7..d0a476a84 100644 --- a/contractcourt/chain_arbitrator_test.go +++ b/contractcourt/chain_arbitrator_test.go @@ -86,7 +86,8 @@ func TestChainArbitratorRepublishCloses(t *testing.T) { published[tx.TxHash()]++ return nil }, - Clock: clock.NewDefaultClock(), + Clock: clock.NewDefaultClock(), + Budget: *DefaultBudgetConfig(), } chainArb := NewChainArbitrator( chainArbCfg, db, @@ -169,7 +170,8 @@ func TestResolveContract(t *testing.T) { PublishTx: func(tx *wire.MsgTx, _ string) error { return nil }, - Clock: clock.NewDefaultClock(), + Clock: clock.NewDefaultClock(), + Budget: *DefaultBudgetConfig(), } chainArb := NewChainArbitrator( chainArbCfg, db, diff --git a/contractcourt/channel_arbitrator_test.go b/contractcourt/channel_arbitrator_test.go index 8c72e02f6..03ba14810 100644 --- a/contractcourt/channel_arbitrator_test.go +++ b/contractcourt/channel_arbitrator_test.go @@ -390,6 +390,7 @@ func createTestChannelArbitrator(t *testing.T, log ArbitratorLog, return nil }, + Budget: *DefaultBudgetConfig(), } // We'll use the resolvedChan to synchronize on call to diff --git a/contractcourt/commit_sweep_resolver_test.go b/contractcourt/commit_sweep_resolver_test.go index e864fb608..5cff3a6d5 100644 --- a/contractcourt/commit_sweep_resolver_test.go +++ b/contractcourt/commit_sweep_resolver_test.go @@ -42,6 +42,7 @@ func newCommitSweepResolverTestContext(t *testing.T, ChainArbitratorConfig: ChainArbitratorConfig{ Notifier: notifier, Sweeper: sweeper, + Budget: *DefaultBudgetConfig(), }, PutResolverReport: func(_ kvdb.RwTx, _ *channeldb.ResolverReport) error { diff --git a/contractcourt/htlc_incoming_contest_resolver_test.go b/contractcourt/htlc_incoming_contest_resolver_test.go index cc3f9c934..7bfe285bd 100644 --- a/contractcourt/htlc_incoming_contest_resolver_test.go +++ b/contractcourt/htlc_incoming_contest_resolver_test.go @@ -351,6 +351,7 @@ func newIncomingResolverTestContext(t *testing.T, isExit bool) *incomingResolver return nil }, HtlcNotifier: htlcNotifier, + Budget: *DefaultBudgetConfig(), }, PutResolverReport: func(_ kvdb.RwTx, _ *channeldb.ResolverReport) error { diff --git a/contractcourt/htlc_outgoing_contest_resolver_test.go b/contractcourt/htlc_outgoing_contest_resolver_test.go index f83d17749..c788241d8 100644 --- a/contractcourt/htlc_outgoing_contest_resolver_test.go +++ b/contractcourt/htlc_outgoing_contest_resolver_test.go @@ -152,6 +152,7 @@ func newOutgoingResolverTestContext(t *testing.T) *outgoingResolverTestContext { return nil }, OnionProcessor: onionProcessor, + Budget: *DefaultBudgetConfig(), }, PutResolverReport: func(_ kvdb.RwTx, _ *channeldb.ResolverReport) error { diff --git a/contractcourt/htlc_success_resolver_test.go b/contractcourt/htlc_success_resolver_test.go index f1c401cc1..e18ef3383 100644 --- a/contractcourt/htlc_success_resolver_test.go +++ b/contractcourt/htlc_success_resolver_test.go @@ -88,6 +88,7 @@ func newHtlcResolverTestContext(t *testing.T, return nil }, HtlcNotifier: htlcNotifier, + Budget: *DefaultBudgetConfig(), }, PutResolverReport: func(_ kvdb.RwTx, report *channeldb.ResolverReport) error { diff --git a/contractcourt/htlc_timeout_resolver_test.go b/contractcourt/htlc_timeout_resolver_test.go index 6797a21f7..f4591281f 100644 --- a/contractcourt/htlc_timeout_resolver_test.go +++ b/contractcourt/htlc_timeout_resolver_test.go @@ -301,6 +301,7 @@ func TestHtlcTimeoutResolver(t *testing.T) { resolutionChan <- msgs[0] return nil }, + Budget: *DefaultBudgetConfig(), }, PutResolverReport: func(_ kvdb.RwTx, _ *channeldb.ResolverReport) error {