Function TestCodecKey

pkg/util/codec/codec_test.go:37–102

(t *testing.T)

)

func TestCodecKey(t *testing.T) {
	table := []struct {
		Input  []types.Datum
		Expect []types.Datum
	}{
		{
			types.MakeDatums(int64(1)),
			types.MakeDatums(int64(1)),
		},

		{
			types.MakeDatums(float32(1), float64(3.15), []byte("123"), "123"),
			types.MakeDatums(float64(1), float64(3.15), []byte("123"), []byte("123")),
		},
		{
			types.MakeDatums(uint64(1), float64(3.15), []byte("123"), int64(-1)),
			types.MakeDatums(uint64(1), float64(3.15), []byte("123"), int64(-1)),
		},

		{
			types.MakeDatums(true, false),
			types.MakeDatums(int64(1), int64(0)),
		},

		{
			types.MakeDatums(nil),
			types.MakeDatums(nil),
		},

		{
			types.MakeDatums(types.NewBinaryLiteralFromUint(100, -1), types.NewBinaryLiteralFromUint(100, 4)),
			types.MakeDatums(uint64(100), uint64(100)),
		},

		{
			types.MakeDatums(types.Enum{Name: "a", Value: 1}, types.Set{Name: "a", Value: 1}),
			types.MakeDatums(uint64(1), uint64(1)),
		},
	}
	typeCtx := types.DefaultStmtNoWarningContext.WithLocation(time.Local)
	for i, datums := range table {
		comment := fmt.Sprintf("%d %v", i, datums)
		b, err := EncodeKey(typeCtx.Location(), nil, datums.Input...)
		require.NoError(t, err, comment)

		args, err := Decode(b, 1)
		require.NoError(t, err, comment)
		require.Equal(t, datums.Expect, args, comment)

		b, err = EncodeValue(typeCtx.Location(), nil, datums.Input...)
		require.NoError(t, err, comment)

		size, err := estimateValuesSize(typeCtx, datums.Input)
		require.NoError(t, err, comment)
		require.Len(t, b, size, comment)

		args, err = Decode(b, 1)
		require.NoError(t, err, comment)