testutil

Documentation

Index

• Variables
• func GetTestField(fname protoreflect.Name) protoreflect.FieldDescriptor
• func MakeTestCodec(fname protoreflect.Name, nonTerminal bool) (ormfield.Codec, error)
• func TestFieldSpecsGen(minLen, maxLen int) *rapid.Generator[[]TestFieldSpec]
• func TestKeyCodecGen(minLen, maxLen int) *rapid.Generator[TestKeyCodec]
• type TestFieldSpec
• type TestKeyCodec
  • func (k TestKeyCodec) Draw(t *rapid.T, id string) []protoreflect.Value

Variables

var GenA = rapid.Custom(func(t *rapid.T) *testpb.ExampleTable {
	a := &testpb.ExampleTable{}
	ref := a.ProtoReflect()
	for _, spec := range TestFieldSpecs {
		field := GetTestField(spec.FieldName)
		value := spec.Gen.Draw(t, string(spec.FieldName))
		if value != nil {
			ref.Set(field, protoreflect.ValueOf(value))
		}
	}
	return a
})

var TestFieldSpecs = []TestFieldSpec{
	{
		"u32",
		rapid.Uint32().AsAny(),
	},
	{
		"u64",
		rapid.Uint64().AsAny(),
	},
	{
		"str",
		rapid.String().Filter(func(x string) bool {

			return strings.IndexByte(x, 0) < 0
		}).AsAny(),
	},
	{
		"bz",
		rapid.SliceOfN(rapid.Byte(), 0, math.MaxUint32).AsAny(),
	},
	{
		"i32",
		rapid.Int32().AsAny(),
	},
	{
		"f32",
		rapid.Uint32().AsAny(),
	},
	{
		"s32",
		rapid.Int32().AsAny(),
	},
	{
		"sf32",
		rapid.Int32().AsAny(),
	},
	{
		"i64",
		rapid.Int64().AsAny(),
	},
	{
		"f64",
		rapid.Uint64().AsAny(),
	},
	{
		"s64",
		rapid.Int64().AsAny(),
	},
	{
		"sf64",
		rapid.Int64().AsAny(),
	},
	{
		"b",
		rapid.Bool().AsAny(),
	},
	{
		"ts",
		rapid.Custom(func(t *rapid.T) protoreflect.Message {
			isNil := rapid.Float32().Draw(t, "isNil")
			if isNil >= 0.95 {
				return nil
			}
			seconds := rapid.Int64Range(ormfield.TimestampSecondsMin, ormfield.TimestampSecondsMax).Draw(t, "seconds")
			nanos := rapid.Int32Range(0, ormfield.TimestampNanosMax).Draw(t, "nanos")
			return (&timestamppb.Timestamp{
				Seconds: seconds,
				Nanos:   nanos,
			}).ProtoReflect()
		}).AsAny(),
	},
	{
		"dur",
		rapid.Custom(func(t *rapid.T) protoreflect.Message {
			isNil := rapid.Float32().Draw(t, "isNil")
			if isNil >= 0.95 {
				return nil
			}
			seconds := rapid.Int64Range(ormfield.DurationNanosMin, ormfield.DurationNanosMax).Draw(t, "seconds")
			nanos := rapid.Int32Range(0, ormfield.DurationNanosMax).Draw(t, "nanos")
			if seconds < 0 {
				nanos = -nanos
			}
			return (&durationpb.Duration{
				Seconds: seconds,
				Nanos:   nanos,
			}).ProtoReflect()
		}).AsAny(),
	},
	{
		"e",
		rapid.Map(rapid.Int32(), func(x int32) protoreflect.EnumNumber {
			return protoreflect.EnumNumber(x)
		}).AsAny(),
	},
}

Functions

func GetTestField

func GetTestField(fname protoreflect.Name) protoreflect.FieldDescriptor

func MakeTestCodec

func MakeTestCodec(fname protoreflect.Name, nonTerminal bool) (ormfield.Codec, error)

func TestFieldSpecsGen

func TestFieldSpecsGen(minLen, maxLen int) *rapid.Generator[[]TestFieldSpec]

func TestKeyCodecGen

func TestKeyCodecGen(minLen, maxLen int) *rapid.Generator[TestKeyCodec]

Types

type TestFieldSpec

type TestFieldSpec struct {
	FieldName protoreflect.Name
	Gen       *rapid.Generator[any]
}

TestFieldSpec defines a test field against the testpb.ExampleTable message.

type TestKeyCodec

type TestKeyCodec struct {
	KeySpecs []TestFieldSpec
	Codec    *ormkv.KeyCodec
}

func (TestKeyCodec) Draw

func (k TestKeyCodec) Draw(t *rapid.T, id string) []protoreflect.Value