mocks

type MockEventStorage struct {
	EmptyCall            func() bool
	CountCall            func() int64
	PopNCall             func(n int64) ([]dtos.EventDTO, error)
	PopNWithMetadataCall func(n int64) ([]dtos.QueueStoredEventDTO, error)
	PushCall             func(event dtos.EventDTO, size int) error
	DropCall             func(size *int64) error
}

type MockImpressionStorage struct {
	EmptyCall            func() bool
	CountCall            func() int64
	LogImpressionsCall   func(impressions []dtos.Impression) error
	PopNCall             func(n int64) ([]dtos.Impression, error)
	PopNWithMetadataCall func(n int64) ([]dtos.ImpressionQueueObject, error)
	DropCall             func(size *int64) error
}

type MockMetricStorage struct {
	IncCounterCall               func(key string)
	IncLatencyCall               func(metricName string, index int)
	PutGaugeCall                 func(key string, gauge float64)
	PopGaugesCall                func() []dtos.GaugeDTO
	PopLatenciesCall             func() []dtos.LatenciesDTO
	PopCountersCall              func() []dtos.CounterDTO
	PeekCountersCall             func() map[string]int64
	PeekLatenciesCall            func() map[string][]int64
	PopGaugesWithMetadataCall    func() (*dtos.GaugeDataBulk, error)
	PopCountersWithMetadataCall  func() (*dtos.CounterDataBulk, error)
	PopLatenciesWithMetadataCall func() (*dtos.LatencyDataBulk, error)
}

type MockSegmentStorage struct {
	ChangeNumberCall       func(segmentName string) (int64, error)
	KeysCall               func(segmentName string) *set.ThreadUnsafeSet
	UpdateCall             func(name string, toAdd *set.ThreadUnsafeSet, toRemove *set.ThreadUnsafeSet, changeNumber int64) error
	SegmentContainsKeyCall func(segmentName string, key string) (bool, error)
	SetChangeNumberCall    func(segmentName string, till int64) error
	CountRemovedKeysCall   func(segmentName string) int64
}

type MockSplitStorage struct {
	AllCall               func() []dtos.SplitDTO
	ChangeNumberCall      func() (int64, error)
	FetchManyCall         func(splitNames []string) map[string]*dtos.SplitDTO
	KillLocallyCall       func(splitName string, defaultTreatment string, changeNumber int64)
	PutManyCall           func(splits []dtos.SplitDTO, changeNumber int64)
	RemoveCall            func(splitName string)
	SegmentNamesCall      func() *set.ThreadUnsafeSet
	SetChangeNumberCall   func(changeNumber int64) error
	SplitCall             func(splitName string) *dtos.SplitDTO
	SplitNamesCall        func() []string
	TrafficTypeExistsCall func(trafficType string) bool
}