itests: Enable harmonydb in harmonydb tests

This commit is contained in:
Łukasz Magiera 2023-11-23 21:51:34 +01:00
parent 687ad3ae53
commit e9d663c997
5 changed files with 12 additions and 0 deletions

View File

@ -16,6 +16,7 @@ func withSetup(t *testing.T, f func(*kit.TestMiner)) {
_, miner, _ := kit.EnsembleMinimal(t, _, miner, _ := kit.EnsembleMinimal(t,
kit.LatestActorsAt(-1), kit.LatestActorsAt(-1),
kit.MockProofs(), kit.MockProofs(),
kit.WithSectorIndexDB(),
) )
f(miner) f(miner)

View File

@ -30,6 +30,7 @@ func withDbSetup(t *testing.T, f func(*kit.TestMiner)) {
_, miner, _ := kit.EnsembleMinimal(t, _, miner, _ := kit.EnsembleMinimal(t,
kit.LatestActorsAt(-1), kit.LatestActorsAt(-1),
kit.MockProofs(), kit.MockProofs(),
kit.WithSectorIndexDB(),
) )
logging.SetLogLevel("harmonytask", "debug") logging.SetLogLevel("harmonytask", "debug")

View File

@ -607,6 +607,7 @@ func (n *Ensemble) Start() *Ensemble {
cfg.Subsystems.EnableMining = m.options.subsystems.Has(SMining) cfg.Subsystems.EnableMining = m.options.subsystems.Has(SMining)
cfg.Subsystems.EnableSealing = m.options.subsystems.Has(SSealing) cfg.Subsystems.EnableSealing = m.options.subsystems.Has(SSealing)
cfg.Subsystems.EnableSectorStorage = m.options.subsystems.Has(SSectorStorage) cfg.Subsystems.EnableSectorStorage = m.options.subsystems.Has(SSectorStorage)
cfg.Subsystems.EnableSectorIndexDB = m.options.subsystems.Has(SHarmony)
cfg.Dealmaking.MaxStagingDealsBytes = m.options.maxStagingDealsBytes cfg.Dealmaking.MaxStagingDealsBytes = m.options.maxStagingDealsBytes
if m.options.mainMiner != nil { if m.options.mainMiner != nil {

View File

@ -37,6 +37,8 @@ const (
SSealing SSealing
SSectorStorage SSectorStorage
SHarmony
MinerSubsystems = iota MinerSubsystems = iota
) )

View File

@ -89,6 +89,13 @@ func WithAllSubsystems() NodeOpt {
} }
} }
func WithSectorIndexDB() NodeOpt {
return func(opts *nodeOpts) error {
opts.subsystems = opts.subsystems.Add(SHarmony)
return nil
}
}
func WithSubsystems(systems ...MinerSubsystem) NodeOpt { func WithSubsystems(systems ...MinerSubsystem) NodeOpt {
return func(opts *nodeOpts) error { return func(opts *nodeOpts) error {
for _, s := range systems { for _, s := range systems {