}

        assertEquals(size, totalRead);
        assertArrayEquals(data, buffer);
    }

    @Test
    @DisplayName("Read handles partial reads from underlying stream")
    void readHandlesPartialReads() throws IOException {
        // Create message with data
        byte[] data = new byte[] { 1, 2, 3, 4, 5 };
        InputStream in = new ByteArrayInputStream(concat(messageHeader(5), data));