// Test with a positive value
        buffer.putShort(0, (short) 0x1234);
        assertEquals(0x1234, NtlmMessage.readUShort(data, 0), "Should read positive UShort correctly.");

        // Test with zero
        buffer.putShort(0, (short) 0);
        assertEquals(0, NtlmMessage.readUShort(data, 0), "Should read zero UShort correctly.");

        // Test with max short value (unsigned interpretation)

        dest[offset + 3] = (byte) (ulong >> 24 & 0xff);
    }

    static void writeUShort(final byte[] dest, final int offset, final int ushort) {
        dest[offset] = (byte) (ushort & 0xff);
        dest[offset + 1] = (byte) (ushort >> 8 & 0xff);
    }

    static int writeSecurityBuffer(final byte[] dest, final int offset, final byte[] src) {
        final int length = src != null ? src.length : 0;

        dest[offset + 3] = (byte) (ulong >> 24 & 0xff);
    }

    static void writeUShort(final byte[] dest, final int offset, final int ushort) {
        dest[offset] = (byte) (ushort & 0xff);
        dest[offset + 1] = (byte) (ushort >> 8 & 0xff);
    }

    static void writeSecurityBuffer(final byte[] dest, final int offset, final int bodyOffset, final byte[] src) {

        assertEquals(0xBBAA, result, "UShort should be little‑endian");
    }

    @Test
    @DisplayName("readUShort throws on too short array")
    void testReadUShortOutOfBounds() {
        byte[] a = new byte[] { 0x02 };
        assertThrows(IndexOutOfBoundsException.class, () -\u003e NtlmMessage.readUShort(a, 0));
    }

    @Test
    @DisplayName("readULong throws on too short array")
    void testReadULongOutOfBounds() {

            bb.putShort((short) 20); // pathConsumed
            bb.putShort((short) 1); // numReferrals
            bb.putShort((short) 15); // tflags
            bb.putShort((short) 0); // tflags high bytes

            // Simple v1 referral
            bb.putShort((short) 1); // version
            bb.putShort((short) 16); // size
            bb.putShort((short) 1); // serverType

        bb.putShort((short) 3);
        bb.putShort((short) 65535); // max unsigned short
        bb.putShort((short) 65535);
        bb.putShort((short) 65535);
        bb.putShort((short) 65535);
        bb.putShort((short) 65535);
        bb.putShort((short) 0);
        bb.putShort((short) 0);
        bb.putShort((short) 0);

        int data1 = (int) (mostSig >>> 32); // first 4 bytes
        short data2 = (short) (mostSig >>> 16); // next 2 bytes
        short data3 = (short) mostSig; // next 2 bytes

        // Write in little-endian format as specified by MS-SMB2
        bb.putInt(data1); // data1 (4 bytes, little-endian)
        bb.putShort(data2); // data2 (2 bytes, little-endian)
        bb.putShort(data3); // data3 (2 bytes, little-endian)

        ByteBuffer buffer = ByteBuffer.allocate(34);
        buffer.order(ByteOrder.LITTLE_ENDIAN);
        buffer.putShort((short) 0); // dialectIndex
        buffer.put((byte) 0x0F); // securityMode (user, encrypted, sigs enabled, sigs required)
        buffer.putShort((short) 50); // maxMpxCount
        buffer.putShort((short) 10); // maxNumberVcs
        buffer.putInt(8192); // maxBufferSize
        buffer.putInt(65536); // maxRawSize

        // Assert
        assertEquals(Short.MIN_VALUE, queryPolicy.level);
    }

    @Test
    void constructor_withMaximumShortValue() {
        // Arrange
        short level = Short.MAX_VALUE;

        // Act
        queryPolicy = new MsrpcQueryInformationPolicy(mockPolicyHandle, level, mockNdrObject);

        // Assert
        assertEquals(Short.MAX_VALUE, queryPolicy.level);
    }

        // DACL at offset 60
        buffer.position(60);
        buffer.put((byte) 0x02); // AclRevision
        buffer.put((byte) 0x00); // Padding
        buffer.putShort((short) 8); // AclSize
        buffer.putShort((short) 0); // AceCount
        buffer.putShort((short) 0); // Padding

        return buffer.array();
    }

    /**
     * Helper method to set error code using reflection
     */