T[] castBuffer = (T[]) buffer;
    sort(castBuffer, 0, bufferSize, comparator);
    if (bufferSize > k) {
      Arrays.fill(buffer, k, buffer.length, null);
      bufferSize = k;
      threshold = buffer[k - 1];
    }
    // Up to bufferSize, all elements of buffer are real Ts (not null unless T includes null)
    T[] topK = Arrays.copyOf(castBuffer, bufferSize);
    // we have to support null elements, so no ImmutableList for us

        // Given
        smbComTreeDisconnect = new SmbComTreeDisconnect(config);
        int[] bufferSizes = { 0, 10, 50, 100, 1024 };

        for (int bufferSize : bufferSizes) {
            byte[] dst = new byte[bufferSize];

            // When
            int result = smbComTreeDisconnect.writeParameterWordsWireFormat(dst, 0);

            // Then

        @DisplayName("Should handle various buffer configurations")
        void testVariousBufferConfigurations(int bufferSize, int offset, int length) {
            byte[] buffer = new byte[bufferSize];
            FileFsSizeInformation info = new FileFsSizeInformation();

            if (offset + length <= bufferSize && length >= 24) {
                assertDoesNotThrow(() -> {
                    info.decode(buffer, offset, length);

        int bufferSize;
        if (initialCapacity > 0 && initialCapacity <= MAX_BUF_SIZE) {
            bufferSize = initialCapacity;
        } else {
            bufferSize = DEFAULT_BUF_SIZE;
        }
        char[] buf = new char[bufferSize];
        int size = 0;
        int len;
        try {
            while ((len = reader.read(buf, size, bufferSize - size)) != -1) {

    T[] castBuffer = (T[]) buffer;
    sort(castBuffer, 0, bufferSize, comparator);
    if (bufferSize > k) {
      Arrays.fill(buffer, k, buffer.length, null);
      bufferSize = k;
      threshold = buffer[k - 1];
    }
    // Up to bufferSize, all elements of buffer are real Ts (not null unless T includes null)
    T[] topK = Arrays.copyOf(castBuffer, bufferSize);
    // we have to support null elements, so no ImmutableList for us

        String testPath = "\\server\\share";
        byte[] pathBytes = testPath.getBytes(java.nio.charset.StandardCharsets.UTF_16LE);
        int bufferSize = stringDataStart + pathBytes.length + 2; // +2 for null terminator
        byte[] buffer = new byte[bufferSize];

        // Path consumed (2 bytes) - value: 10 * 2 = 20
        SMBUtil.writeInt2(20, buffer, 0);

        // Number of referrals (2 bytes) - value: 1

      }

      OPCODE_CONTROL_CLOSE -> {
        var code = CLOSE_NO_STATUS_CODE
        var reason = ""
        val bufferSize = controlFrameBuffer.size
        if (bufferSize == 1L) {
          throw ProtocolException("Malformed close payload length of 1.")
        } else if (bufferSize != 0L) {
          code = controlFrameBuffer.readShort().toInt()
          reason = controlFrameBuffer.readUtf8()

     * @return The buffer size
     */
    public int getBufferSize() {
        return bufferSize;
    }

    /**
     * Sets the buffer size for data transfers.
     *
     * @param bufferSize The buffer size
     */
    public void setBufferSize(final int bufferSize) {
        this.bufferSize = bufferSize;
    }

    /**
     * Gets the local IP address for passive mode FTP.
     *

        @DisplayName("Should handle maximum buffer index")
        void testMaxBufferIndex() throws SMBProtocolDecodingException {
            // Given
            int bufferSize = 10000;
            byte[] buffer = new byte[bufferSize];
            int bufferIndex = bufferSize - 4;
            SMBUtil.writeInt2(4, buffer, bufferIndex);
            SMBUtil.writeInt2(0, buffer, bufferIndex + 2);

            // When

    void testWriteBytesWireFormatWithDifferentBufferSizes(int bufferSize) {
        request = new Smb2QueryInfoRequest(mockConfig, testFileId);

        if (bufferSize > 0) {
            Encodable mockInputBuffer = mock(Encodable.class);
            when(mockInputBuffer.encode(any(byte[].class), anyInt())).thenReturn(bufferSize);
            when(mockInputBuffer.size()).thenReturn(bufferSize);

            try {