Function TestCutRow

pkg/tablecodec/tablecodec_test.go:304–354

(t *testing.T)

}

func TestCutRow(t *testing.T) {
	var err error
	c1 := &column{id: 1, tp: types.NewFieldType(mysql.TypeLonglong)}
	c2 := &column{id: 2, tp: types.NewFieldType(mysql.TypeVarchar)}
	c3 := &column{id: 3, tp: types.NewFieldType(mysql.TypeNewDecimal)}
	cols := []*column{c1, c2, c3}

	row := make([]types.Datum, 3)
	row[0] = types.NewIntDatum(100)
	row[1] = types.NewBytesDatum([]byte("abc"))
	row[2] = types.NewDecimalDatum(types.NewDecFromInt(1))

	sc := stmtctx.NewStmtCtxWithTimeZone(time.UTC)
	data := make([][]byte, 3)
	data[0], err = EncodeValue(sc.TimeZone(), nil, row[0])
	require.NoError(t, err)
	data[1], err = EncodeValue(sc.TimeZone(), nil, row[1])
	require.NoError(t, err)
	data[2], err = EncodeValue(sc.TimeZone(), nil, row[2])
	require.NoError(t, err)
	// Encode
	colIDs := make([]int64, 0, 3)
	for _, col := range cols {
		colIDs = append(colIDs, col.id)
	}
	bs, err := EncodeOldRow(sc.TimeZone(), row, colIDs, nil, nil)
	require.NoError(t, err)
	require.NotNil(t, bs)

	// Decode
	colMap := make(map[int64]int, 3)
	for i, col := range cols {
		colMap[col.id] = i
	}
	r, err := CutRowNew(bs, colMap)
	require.NoError(t, err)
	require.NotNil(t, r)
	require.Len(t, r, 3)
	// Compare cut row and original row
	for i := range colIDs {
		require.Equal(t, data[i], r[i])
	}
	bs = []byte{codec.NilFlag}
	r, err = CutRowNew(bs, colMap)
	require.NoError(t, err)
	require.Nil(t, r)
	bs = nil
	r, err = CutRowNew(bs, colMap)
	require.NoError(t, err)
	require.Nil(t, r)
}

func TestCutKeyNew(t *testing.T) {
	values := []types.Datum{types.NewIntDatum(1), types.NewBytesDatum([]byte("abc")), types.NewFloat64Datum(5.5)}