testutil

type AggTestCase struct {
	Ctx              sessionctx.Context
	ExecType         string
	AggFunc          string
	GroupByNDV       int
	Rows             int
	Concurrency      int
	DataSourceSorted bool
	HasDistinct      bool
}

type LimitCase struct {
	Ctx                   sessionctx.Context
	ChildUsedSchema       []bool
	Rows                  int
	Offset                int
	Count                 int
	UsingInlineProjection bool
}

type MockActionOnExceed struct {
	memory.BaseOOMAction
	// contains filtered or unexported fields
}

type MockDataPhysicalPlan struct {
	MockPhysicalPlan
	DataSchema *expression.Schema
	Exec       exec.Executor
}

type MockDataSource struct {
	GenData []*chunk.Chunk
	Chunks  []*chunk.Chunk
	P       MockDataSourceParameters
	exec.BaseExecutor
	ChunkPtr int
}

type MockDataSourceParameters struct {
	Ctx         sessionctx.Context
	DataSchema  *expression.Schema
	GenDataFunc func(row int, typ *types.FieldType) any
	Ndvs        []int
	Orders      []bool

	// Sometimes, user wants to manually provide test data
	// and he can save provided test data at here.
	Datums [][]any

	Rows   int
	HasSel bool
}

type MockPhysicalPlan interface {
	plannercore.PhysicalPlan
	GetExecutor() exec.Executor
}

type SortCase struct {
	Ctx        sessionctx.Context
	OrderByIdx []int
	Ndvs       []int
	Rows       int
}

type WindowTestCase struct {
	Ctx              sessionctx.Context
	Frame            *logicalop.WindowFrame
	WindowFunc       string
	RawDataSmall     string
	Columns          []*expression.Column
	NumFunc          int
	Ndv              int
	Rows             int
	Concurrency      int
	Pipelined        int
	DataSourceSorted bool
}