Function TestWriteMessage

server/protocol_test.go:186–232

(t *testing.T)

}

func TestWriteMessage(t *testing.T) {
	t.Run("write simple message", func(t *testing.T) {
		var buf bytes.Buffer

		data := []byte("test data")
		err := wire.WriteMessage(&buf, 'T', data)
		if err != nil {
			t.Fatalf("wire.WriteMessage() error = %v", err)
		}

		// Check message type
		if buf.Bytes()[0] != 'T' {
			t.Errorf("message type = %c, want T", buf.Bytes()[0])
		}

		// Check length (includes itself = 4)
		length := binary.BigEndian.Uint32(buf.Bytes()[1:5])
		if length != uint32(len(data)+4) {
			t.Errorf("length = %d, want %d", length, len(data)+4)
		}

		// Check data
		if !bytes.Equal(buf.Bytes()[5:], data) {
			t.Errorf("data = %v, want %v", buf.Bytes()[5:], data)
		}
	})

	t.Run("write empty message", func(t *testing.T) {
		var buf bytes.Buffer

		err := wire.WriteMessage(&buf, 'Z', []byte{})
		if err != nil {
			t.Fatalf("wire.WriteMessage() error = %v", err)
		}

		// Total should be 5 bytes: type (1) + length (4)
		if buf.Len() != 5 {
			t.Errorf("buffer length = %d, want 5", buf.Len())
		}

		// Length should be 4 (just the length field itself)
		length := binary.BigEndian.Uint32(buf.Bytes()[1:5])
		if length != 4 {
			t.Errorf("length = %d, want 4", length)
		}
	})
}

func TestWriteAuthOK(t *testing.T) {
	var buf bytes.Buffer