utils

Documentation

Index

• func MakePredicateComparisonColumn(columnName string, ...) *api_service_protos.TPredicate_Comparison
• func MakePredicateIsNotNullColumn(columnName string) *api_service_protos.TPredicate_IsNotNull
• func MakePredicateIsNullColumn(columnName string) *api_service_protos.TPredicate_IsNull
• func MustProtobufDifference(expected, actual proto.Message) string
• func NewInt32IDArrayBuilder(pool memory.Allocator) *array.Int32Builder
• func NewInt64IDArrayBuilder(pool memory.Allocator) *array.Int64Builder
• func ProtobufDifference(expected, actual proto.Message) (string, error)
• type ArrowIDBuilder
• type Record
  • func (r *Record[ID, IDBUILDER]) MatchRecord(t *testing.T, receivedRecord arrow.Record, ...)
• type Table
  • func (tb *Table[T, K]) MatchRecords(t *testing.T, records []arrow.Record, schema *api_service_protos.TSchema)
  • func (tb *Table[_, _]) MatchSchema(t *testing.T, schema *api_service_protos.TSchema)
• type TableIDTypes
• type TableSchema

Functions

func MakePredicateComparisonColumn

func MakePredicateComparisonColumn(
	columnName string,
	operation api_service_protos.TPredicate_TComparison_EOperation,
	value *Ydb.TypedValue,
) *api_service_protos.TPredicate_Comparison

func MakePredicateIsNotNullColumn

func MakePredicateIsNotNullColumn(columnName string) *api_service_protos.TPredicate_IsNotNull

func MakePredicateIsNullColumn

func MakePredicateIsNullColumn(columnName string) *api_service_protos.TPredicate_IsNull

func MustProtobufDifference

func MustProtobufDifference(expected, actual proto.Message) string

func NewInt32IDArrayBuilder

func NewInt32IDArrayBuilder(pool memory.Allocator) *array.Int32Builder

func NewInt64IDArrayBuilder

func NewInt64IDArrayBuilder(pool memory.Allocator) *array.Int64Builder

func ProtobufDifference

func ProtobufDifference(expected, actual proto.Message) (string, error)

Types

type ArrowIDBuilder

type ArrowIDBuilder[ID TableIDTypes] interface {
	*array.Int64Builder | *array.Int32Builder
	Append(ID)
	NewArray() arrow.Array
	Release()
}

type Record

type Record[ID TableIDTypes, IDBUILDER ArrowIDBuilder[ID]] struct {
	Columns map[string]any
}

Record is somewhat equivalent to arrow.Record. Store columns in map because order of columns in some datasource is undefined. (i.e. in YDB - https://st.yandex-team.ru/KIKIMR-20836)

func (*Record[ID, IDBUILDER]) MatchRecord

func (r *Record[ID, IDBUILDER]) MatchRecord(
	t *testing.T,
	receivedRecord arrow.Record,
	receivedSchema *api_service_protos.TSchema,
	idArrBuilder IDBUILDER)

type Table

type Table[T TableIDTypes, K ArrowIDBuilder[T]] struct {
	Name                  string
	Schema                *TableSchema
	Records               []*Record[T, K] // Large tables may consist of multiple records
	IDArrayBuilderFactory func() K
}

func (*Table[T, K]) MatchRecords

func (tb *Table[T, K]) MatchRecords(t *testing.T, records []arrow.Record, schema *api_service_protos.TSchema)

func (*Table[_, _]) MatchSchema

func (tb *Table[_, _]) MatchSchema(t *testing.T, schema *api_service_protos.TSchema)

type TableIDTypes

type TableIDTypes interface {
	int32 | int64
}

type TableSchema

type TableSchema struct {
	Columns map[string]*Ydb.Type
}