Function TestPartialBatchCodec_Decode

codecs/partial_codecs_test.go:301–344

(t *testing.T)

}

func TestPartialBatchCodec_Decode(t *testing.T) {
	localSerialConsistency := primitive.ConsistencyLevelLocalSerial
	codec := &partialBatchCodec{}

	msg := &message.Batch{
		Type: primitive.BatchTypeLogged,
		Children: []*message.BatchChild{
			{
				Query: "SELECT * FROM table WHERE id = ?",
				Values: []*primitive.Value{
					{Type: primitive.ValueTypeRegular, Contents: []byte{0x01, 0x02, 0x03}},
					{Type: primitive.ValueTypeNull},
				},
			},
			{Id: []byte{0x01, 0x02, 0x03}},
		},
		Consistency:       primitive.ConsistencyLevelQuorum,
		SerialConsistency: &localSerialConsistency,
	}

	var buf bytes.Buffer

	err := builtinBatchCodec.Encode(msg, &buf, primitive.ProtocolVersion4)
	require.NoError(t, err)

	batch, err := codec.Decode(NewFrameBodyReader(buf.Bytes()), primitive.ProtocolVersion4)
	require.NoError(t, err)

	assert.False(t, batch.IsResponse())
	assert.Equal(t, primitive.OpCodeBatch, batch.GetOpCode())
	assert.Equal(t, fmt.Sprintf("BATCH (%d statements)", len(msg.Children)), batch.(*PartialBatch).String())

	require.Len(t, batch.(*PartialBatch).Queries, 2)
	assert.Equal(t, "SELECT * FROM table WHERE id = ?", batch.(*PartialBatch).Queries[0].QueryOrId)
	assert.Equal(t, []byte{0x0, 0x2, 0x0, 0x0, 0x0, 0x3, 0x1, 0x2, 0x3, 0xff, 0xff, 0xff, 0xff},
		batch.(*PartialBatch).Queries[0].Values)

	assert.Equal(t, []byte{0x1, 0x2, 0x3}, batch.(*PartialBatch).Queries[1].QueryOrId)
	assert.Equal(t, []byte{0x0, 0x0}, batch.(*PartialBatch).Queries[1].Values)

	assert.Equal(t, msg.Type, batch.(*PartialBatch).Type)
	assert.Equal(t, msg.Consistency, batch.(*PartialBatch).Consistency)
	assert.Equal(t, []byte{0x10, 0x0, 0x9}, batch.(*PartialBatch).Parameters)
}

func TestPartialBatchCodec_Decode_Invalid(t *testing.T) {
	tests := []struct {