From c6ccb4b9a3dd75e628009594cd6adf8b47172d1a Mon Sep 17 00:00:00 2001
From: Mukundan Kavanur Kidambi <mukundan.kidambi@outreach.io>
Date: Wed, 30 Mar 2022 16:22:28 -0700
Subject: [PATCH] Addressing feedback

---
 copy_both_response_test.go | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/copy_both_response_test.go b/copy_both_response_test.go
index 7aa7da22..d9816fc8 100644
--- a/copy_both_response_test.go
+++ b/copy_both_response_test.go
@@ -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")
 }