Function TestPartialQueryCodec_Encode

codecs/partial_codecs_test.go:72–151

(t *testing.T)

}

func TestPartialQueryCodec_Encode(t *testing.T) {
	tests := []struct {
		name string
		msg  *message.Query
	}{
		{
			name: "simple",
			msg: &message.Query{
				Query: "select * from table1",
			},
		},
		{
			name: "consistency",
			msg: &message.Query{
				Query: "select * from table2",
				Options: &message.QueryOptions{
					Consistency: primitive.ConsistencyLevelEachQuorum,
				},
			},
		},
		{
			name: "parameters",
			msg: &message.Query{
				Query: "select * from table2",
				Options: &message.QueryOptions{
					Consistency: primitive.ConsistencyLevelEachQuorum,
					PositionalValues: []*primitive.Value{
						{
							Type:     primitive.ValueTypeRegular,
							Contents: []byte{0x1, 0x2, 0x3},
						},
						{
							Type:     primitive.ValueTypeRegular,
							Contents: []byte{0x4, 0x5, 0x6},
						},
					},
					SkipMetadata:    true,
					PageSize:        123,
					PageSizeInBytes: false,
					PagingState:     []byte{0xa, 0xb, 0xc},
					Keyspace:        "keyspace1",
				},
			},
		},
	}

	codec := &partialQueryCodec{}
	version := primitive.ProtocolVersion4

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			expectedLen, err := builtinQueryCodec.EncodedLength(tt.msg, version)
			assert.NoError(t, err, "unexpected error calculating encoded length with built-in query codec")

			var expectedBytes bytes.Buffer
			err = builtinQueryCodec.Encode(tt.msg, &expectedBytes, version)
			require.NoError(t, err, "unexpected error encoding with built-in query codec")