Addressing feedback

This commit is contained in:
Mukundan Kavanur Kidambi 2022-03-30 16:22:28 -07:00 committed by Jack Christensen
parent e145003288
commit c6ccb4b9a3

View File

@ -8,15 +8,11 @@ import (
)
func TestEncodeDecode(t *testing.T) {
src := pgproto3.CopyBothResponse{
OverallFormat: byte(1), // Just to differ from defaults
ColumnFormatCodes: []uint16{0, 1},
}
dstBytes := []byte{}
dstBytes = src.Encode(dstBytes)
dst := pgproto3.CopyBothResponse{}
err := dst.Decode(dstBytes[5:])
srcBytes := []byte{'W', 0x00, 0x00, 0x00, 0x0b, 0x01, 0x00, 0x02, 0x00, 0x00, 0x00, 0x01}
dstResp := pgproto3.CopyBothResponse{}
err := dstResp.Decode(srcBytes[5:])
assert.NoError(t, err, "No errors on decode")
assert.Equal(t, dst.OverallFormat, src.OverallFormat, "OverallFormat is decoded successfully")
assert.EqualValues(t, dst.ColumnFormatCodes, src.ColumnFormatCodes)
dstBytes := []byte{}
dstBytes = dstResp.Encode(dstBytes)
assert.EqualValues(t, srcBytes, dstBytes, "Expecting src & dest bytes to match")
}