e2e: fix TestBlocklist

SiaFoundation · Mar 19, 2024 · 91b482a · 91b482a
1 parent 8799c23
commit 91b482a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/autopilot/autopilot.go b/autopilot/autopilot.go
@@ -148,7 +148,7 @@ func New(id string, bus Bus, workers []Worker, logger *zap.Logger, heartbeat tim
 	}
 
 	ap.s = scanner
-	ap.c = contractor.New(bus, ap.logger, revisionSubmissionBuffer, revisionBroadcastInterval)
+	ap.c = contractor.New(bus, bus, ap.logger, revisionSubmissionBuffer, revisionBroadcastInterval)
 	ap.m = newMigrator(ap, migrationHealthCutoff, migratorParallelSlabsPerWorker)
 	ap.a = newAccounts(ap, ap.bus, ap.bus, ap.workers, ap.logger, accountsRefillInterval)
 

diff --git a/autopilot/contractor/contractor.go b/autopilot/contractor/contractor.go
@@ -173,10 +173,11 @@ type (
 	}
 )
 
-func New(alerter alerts.Alerter, logger *zap.SugaredLogger, revisionSubmissionBuffer uint64, revisionBroadcastInterval time.Duration) *Contractor {
+func New(bus Bus, alerter alerts.Alerter, logger *zap.SugaredLogger, revisionSubmissionBuffer uint64, revisionBroadcastInterval time.Duration) *Contractor {
 	logger = logger.Named("contractor")
 	ctx, cancel := context.WithCancel(context.Background())
 	return &Contractor{
+		bus:     bus,
 		alerter: alerter,
 		churn:   newAccumulatedChurn(),
 		logger:  logger,