testutil

package

v2.12.0 Latest Latest Go to latest Published: Jan 8, 2026 License: MIT Imports: 18 Imported by: 0

Details

Package testutil provides basic utilities for testing the client.

This section is empty.

This section is empty.

func StartMockFlightServer(t *testing.T, blobSize int64) *flight.Server

type BlobTicket struct {
	Name string
	Size int64
}

func BlobTicketFromJSONBytes(jsBytes []byte) (*BlobTicket, error)

func NewBlobTicket(size int64) *BlobTicket

func (bt *BlobTicket) ToJSONBytes() []byte

func (bt *BlobTicket) ToJSONString() string

type ErrorMessageMockReader struct {
	Counter      int
	ErrorMessage string
}

func (emmr *ErrorMessageMockReader) Message() (*ipc.Message, error)

func (emmr *ErrorMessageMockReader) Release()

func (emmr *ErrorMessageMockReader) Retain()

type MockFlightServer struct {
	BlobSize int64
	Records  map[string][]arrow.Record
	flight.BaseFlightServer
}

func (f *MockFlightServer) DoGet(tkt *flight.Ticket, fs flight.FlightService_DoGetServer) error

func (f *MockFlightServer) MakeBlobRecords(name string, size int64) []arrow.Record

func (f *MockFlightServer) MakeConstantRecords() []arrow.Record

type SQLQueryTicket struct {
	Database  string
	QueryType string
	SQLQuery  string
}

func NewSQLQueryTicket(database string, queryType string, query string) *SQLQueryTicket

func SQLQueryTicketFromJSONBytes(jsBytes []byte) (*SQLQueryTicket, error)

type ServAuth struct{}

func (a *ServAuth) Authenticate(c flight.AuthConn) error

func (a *ServAuth) IsValid(token string) (any, error)