Function TestMessageRoundTrip

server/protocol_test.go:435–473

(t *testing.T)

}

func TestMessageRoundTrip(t *testing.T) {
	// Test that we can write a message and read it back
	testCases := []struct {
		name    string
		msgType byte
		data    []byte
	}{
		{"empty", 'Z', []byte{}},
		{"single byte", 'T', []byte{42}},
		{"text", 'Q', []byte("SELECT 1\x00")},
		{"binary", 'd', []byte{0x00, 0xFF, 0x01, 0xFE}},
	}

	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			var buf bytes.Buffer

			// Write
			err := wire.WriteMessage(&buf, tc.msgType, tc.data)
			if err != nil {
				t.Fatalf("wire.WriteMessage() error = %v", err)
			}

			// Read
			msgType, body, err := wire.ReadMessage(&buf)
			if err != nil {
				t.Fatalf("wire.ReadMessage() error = %v", err)
			}

			if msgType != tc.msgType {
				t.Errorf("msgType = %c, want %c", msgType, tc.msgType)
			}

			if !bytes.Equal(body, tc.data) {
				t.Errorf("body = %v, want %v", body, tc.data)
			}
		})
	}
}