mocks

type MockFairnessPolicy ¶

type MockFairnessPolicy struct {
	TypedNameV   plugin.TypedName
	NewStateFunc func(ctx context.Context) any
	PickFunc     func(ctx context.Context, flowGroup flowcontrol.PriorityBandAccessor) (flowcontrol.FlowQueueAccessor, error)
}

MockFairnessPolicy is a behavioral mock for the FairnessPolicy interface. Simple accessors are configured with public value fields (e.g., NameV). Complex methods with logic are configured with function fields (e.g., PickFunc).

func (m *MockFairnessPolicy) NewState(ctx context.Context) any

func (m *MockFairnessPolicy) Pick(ctx context.Context, flowGroup flowcontrol.PriorityBandAccessor) (flowcontrol.FlowQueueAccessor, error)

func (m *MockFairnessPolicy) TypedName() plugin.TypedName

type MockFlowControlRequest ¶

type MockFlowControlRequest struct {
	FlowKeyV             flowcontrol.FlowKey
	ByteSizeV            uint64
	InferenceRequestV    *scheduling.InferenceRequest
	ReceivedTimestampV   time.Time
	InitialEffectiveTTLV time.Duration
	IDV                  string
	MetadataV            map[string]any
	InferencePoolNameV   string
	ModelNameV           string
	TargetModelNameV     string
}

MockFlowControlRequest provides a mock implementation of the FlowControlRequest interface.

func NewMockFlowControlRequest(
	byteSize uint64,
	id string,
	key flowcontrol.FlowKey,
	opts ...MockRequestOption,
) *MockFlowControlRequest

func (m *MockFlowControlRequest) ByteSize() uint64

func (m *MockFlowControlRequest) FlowKey() flowcontrol.FlowKey

func (m *MockFlowControlRequest) GetMetadata() map[string]any

func (m *MockFlowControlRequest) ID() string

func (m *MockFlowControlRequest) InferencePoolName() string

func (m *MockFlowControlRequest) InferenceRequest() *scheduling.InferenceRequest

func (m *MockFlowControlRequest) InitialEffectiveTTL() time.Duration

func (m *MockFlowControlRequest) ModelName() string

func (m *MockFlowControlRequest) ReceivedTimestamp() time.Time

func (m *MockFlowControlRequest) TargetModelName() string

type MockFlowQueueAccessor ¶

type MockFlowQueueAccessor struct {
	NameV           string
	LenV            int
	ByteSizeV       uint64
	PeekHeadV       flowcontrol.QueueItemAccessor
	PeekTailV       flowcontrol.QueueItemAccessor
	FlowKeyV        flowcontrol.FlowKey
	OrderingPolicyV flowcontrol.OrderingPolicy
	CapabilitiesV   []flowcontrol.QueueCapability
}

MockFlowQueueAccessor is a simple stub mock for the FlowQueueAccessor interface. It is used for tests that require static, predictable return values from a queue accessor. For complex, stateful queue behavior, use the mock in ../../contracts/mocks.MockManagedQueue.

func (m *MockFlowQueueAccessor) ByteSize() uint64

func (m *MockFlowQueueAccessor) Capabilities() []flowcontrol.QueueCapability

func (m *MockFlowQueueAccessor) FlowKey() flowcontrol.FlowKey

func (m *MockFlowQueueAccessor) Len() int

func (m *MockFlowQueueAccessor) Name() string

func (m *MockFlowQueueAccessor) OrderingPolicy() flowcontrol.OrderingPolicy

func (m *MockFlowQueueAccessor) PeekHead() flowcontrol.QueueItemAccessor

func (m *MockFlowQueueAccessor) PeekTail() flowcontrol.QueueItemAccessor

type MockOrderingPolicy ¶

type MockOrderingPolicy struct {
	TypedNameV                 plugin.TypedName
	LessFunc                   func(a, b flowcontrol.QueueItemAccessor) bool
	RequiredQueueCapabilitiesV []flowcontrol.QueueCapability
}

MockOrderingPolicy is a behavioral mock for the OrderingPolicy interface. Simple accessors are configured with public value fields (e.g., TypedNameV). Complex methods with logic are configured with function fields (e.g., LessFunc).

func (m *MockOrderingPolicy) Less(a, b flowcontrol.QueueItemAccessor) bool

func (m *MockOrderingPolicy) RequiredQueueCapabilities() []flowcontrol.QueueCapability

func (m *MockOrderingPolicy) TypedName() plugin.TypedName

type MockPriorityBandAccessor ¶

type MockPriorityBandAccessor struct {
	PriorityV         int
	PolicyStateV      any
	FlowKeysFunc      func() []flowcontrol.FlowKey
	QueueFunc         func(flowID string) flowcontrol.FlowQueueAccessor
	IterateQueuesFunc func(callback func(flow flowcontrol.FlowQueueAccessor) (keepIterating bool))
}

MockPriorityBandAccessor is a behavioral mock for the PriorityBandAccessor interface. Simple accessors are configured with public value fields (e.g., PriorityV). Complex methods with logic are configured with function fields (e.g., IterateQueuesFunc).

Convention: Fields suffixed with 'V' (e.g., PriorityV) are static Value return fields. This avoids collision with the interface method of the same name.

func (m *MockPriorityBandAccessor) FlowKeys() []flowcontrol.FlowKey

func (m *MockPriorityBandAccessor) IterateQueues(callback func(flow flowcontrol.FlowQueueAccessor) bool)

func (m *MockPriorityBandAccessor) PolicyState() any

func (m *MockPriorityBandAccessor) Priority() int

func (m *MockPriorityBandAccessor) Queue(id string) flowcontrol.FlowQueueAccessor

type MockQueueItemAccessor ¶

type MockQueueItemAccessor struct {
	EnqueueTimeV     time.Time
	EffectiveTTLV    time.Duration
	OriginalRequestV flowcontrol.FlowControlRequest
	HandleV          flowcontrol.QueueItemHandle
}

MockQueueItemAccessor provides a mock implementation of the QueueItemAccessor interface.

func NewMockQueueItemAccessor(
	byteSize uint64,
	reqID string,
	key flowcontrol.FlowKey,
	opts ...MockRequestOption,
) *MockQueueItemAccessor

func (m *MockQueueItemAccessor) EffectiveTTL() time.Duration

func (m *MockQueueItemAccessor) EnqueueTime() time.Time

func (m *MockQueueItemAccessor) Handle() flowcontrol.QueueItemHandle

func (m *MockQueueItemAccessor) OriginalRequest() flowcontrol.FlowControlRequest

func (m *MockQueueItemAccessor) SetHandle(handle flowcontrol.QueueItemHandle)

type MockQueueItemHandle ¶

type MockQueueItemHandle struct {
	RawHandle      any
	IsInvalidatedV bool
}

MockQueueItemHandle provides a mock implementation of the QueueItemHandle interface.

func (m *MockQueueItemHandle) Handle() any

func (m *MockQueueItemHandle) Invalidate()

func (m *MockQueueItemHandle) IsInvalidated() bool

type MockRequestOption ¶

type MockRequestOption func(*MockFlowControlRequest)

MockRequestOption is a functional option for configuring a MockFlowControlRequest.

type MockSaturationDetector ¶

type MockSaturationDetector struct {
	TypedNameV   plugin.TypedName
	IsSaturatedV bool
	SaturationV  float64
}

MockSaturationDetector is a behavioral mock for the SaturationDetector interface.

func (m *MockSaturationDetector) IsSaturated() bool

func (m *MockSaturationDetector) LastCheckTime() time.Time

func (m *MockSaturationDetector) Saturation(ctx context.Context, endpoints []datalayer.Endpoint) float64

func (m *MockSaturationDetector) TypedName() plugin.TypedName