From 033ca7d47f43284a1aa3754259a6ec0561b02840 Mon Sep 17 00:00:00 2001 From: Yuli Khodorkovskiy Date: Tue, 6 Jul 2021 21:35:21 -0400 Subject: [PATCH] Fix unexpected EOF failure for StartupMessage --- backend_test.go | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/backend_test.go b/backend_test.go index 5e9a2ac5..708f1280 100644 --- a/backend_test.go +++ b/backend_test.go @@ -50,8 +50,12 @@ func TestBackendReceiveUnexpectedEOF(t *testing.T) { assert.Nil(t, msg) assert.Equal(t, io.ErrUnexpectedEOF, err) - // Receive FE msg - server.push([]byte{'F', 0, 0, 0, 6}) + // Receive StartupMessage msg + dst := []byte{} + dst = pgio.AppendUint32(dst, 1000) // tell the backend we expect 1000 bytes to be read + dst = pgio.AppendUint32(dst, 1) // only send 1 byte + server.push(dst) + msg, err = backend.ReceiveStartupMessage() assert.Nil(t, msg) assert.Equal(t, io.ErrUnexpectedEOF, err)