github.com/machinefi/w3bstream@v1.6.5-rc9.0.20240426031326-b8c7c4876e72/pkg/test/patch_modules/stragety.go (about)

     1  package patch_modules
     2  
     3  import (
     4  	"context"
     5  
     6  	"github.com/agiledragon/gomonkey/v2"
     7  
     8  	"github.com/machinefi/w3bstream/pkg/models"
     9  	"github.com/machinefi/w3bstream/pkg/modules/strategy"
    10  )
    11  
    12  func StrategyRemove(patch *gomonkey.Patches, err error) *gomonkey.Patches {
    13  	return patch.ApplyFunc(
    14  		strategy.Remove,
    15  		func(_ context.Context, _ *strategy.CondArgs) error { return err },
    16  	)
    17  }
    18  
    19  func StrategyBatchCreate(patch *gomonkey.Patches, err error) *gomonkey.Patches {
    20  	return patch.ApplyFunc(
    21  		strategy.BatchCreate,
    22  		func(_ context.Context, _ []models.Strategy) error { return err },
    23  	)
    24  }