testutil

Documentation

Index

• func CompareUnorderedStringSlice(a []string, b []string) bool
• func DatumEqual(t testing.TB, expected, actual types.Datum, msgAndArgs ...any)
• func HandleEqual(t testing.TB, expected, actual kv.Handle, msgAndArgs ...any)
• func MaskSortHandles(handles []int64, shardBitsCount int, fieldType byte) []int64
• func MustNewCommonHandle(t *testing.T, values ...any) kv.Handle
• func RandStringRunes(n int) string
• type LogHook
  • func WithLogHook(ctx context.Context, t *testing.T, msgFilter string) (newCtx context.Context, hook *LogHook)
  • func (h *LogHook) Check(entry zapcore.Entry, ce *zapcore.CheckedEntry) *zapcore.CheckedEntry
  • func (h *LogHook) CheckLogCount(t *testing.T, expected int)
  • func (h *LogHook) Write(entry zapcore.Entry, fields []zapcore.Field) error

Functions

func CompareUnorderedStringSlice

func CompareUnorderedStringSlice(a []string, b []string) bool

CompareUnorderedStringSlice compare two string slices. If a and b is exactly the same except the order, it returns true. In otherwise return false.

func DatumEqual

func DatumEqual(t testing.TB, expected, actual types.Datum, msgAndArgs ...any)

DatumEqual verifies that the actual value is equal to the expected value. For string datum, they are compared by the binary collation.

func HandleEqual

func HandleEqual(t testing.TB, expected, actual kv.Handle, msgAndArgs ...any)

HandleEqual verifies that the actual handle is equal to the expected handle.

func MaskSortHandles

func MaskSortHandles(handles []int64, shardBitsCount int, fieldType byte) []int64

MaskSortHandles sorts the handles by lowest (fieldTypeBits - 1 - shardBitsCount) bits.

func MustNewCommonHandle

func MustNewCommonHandle(t *testing.T, values ...any) kv.Handle

MustNewCommonHandle create a common handle with given values.

func RandStringRunes

func RandStringRunes(n int) string

RandStringRunes generate random string of length n.

Types

type LogHook

type LogHook struct {
	zapcore.Core
	Logs []logEntry
	// contains filtered or unexported fields
}

LogHook captures logs, mainly for testing

func WithLogHook

func WithLogHook(ctx context.Context, t *testing.T, msgFilter string) (newCtx context.Context, hook *LogHook)

WithLogHook is a helper function to use with LogHook. It returns a context whose logger is replaced with the hook.

func (*LogHook) Check

func (h *LogHook) Check(entry zapcore.Entry, ce *zapcore.CheckedEntry) *zapcore.CheckedEntry

Check implements the string filter

func (*LogHook) CheckLogCount

func (h *LogHook) CheckLogCount(t *testing.T, expected int)

CheckLogCount is a helper function to assert the number of logs captured

func (*LogHook) Write

func (h *LogHook) Write(entry zapcore.Entry, fields []zapcore.Field) error

Write captures the log and save it