byte[] testBuffer = new byte[8];
            ByteBuffer bb = ByteBuffer.wrap(testBuffer).order(ByteOrder.LITTLE_ENDIAN);
            bb.putShort((short) 0); // pathConsumed (will be divided by 2)
            bb.putShort((short) 0); // numReferrals
            bb.putShort((short) 0); // tflags
            bb.putShort((short) 0); // tflags high bytes (skipped in decode)

  TestByteSource(byte[] bytes, TestOption... options) {
    this.bytes = checkNotNull(bytes);
    this.options = ImmutableSet.copyOf(options);
  }

  @Override
  public boolean wasStreamOpened() {
    return inputStreamOpened;
  }

  @Override
  public boolean wasStreamClosed() {
    return inputStreamClosed;
  }

  @Override
  public InputStream openStream() throws IOException {

  TestByteSource(byte[] bytes, TestOption... options) {
    this.bytes = checkNotNull(bytes);
    this.options = ImmutableSet.copyOf(options);
  }

  @Override
  public boolean wasStreamOpened() {
    return inputStreamOpened;
  }

  @Override
  public boolean wasStreamClosed() {
    return inputStreamClosed;
  }

  @Override
  public InputStream openStream() throws IOException {

        // Prepare a buffer with sample data.
        // 2 bytes for fileAttributes, 4 bytes for lastWriteTime, 4 bytes for fileSize
        byte[] buffer = new byte[20];
        // File Attributes: 0x0010 (Directory)
        buffer[0] = 0x10;
        buffer[1] = 0x00;
        // Last Write Time (UTime): A sample timestamp in milliseconds

        assertEquals((byte) 0xCC, buffer[4]);
        assertEquals((byte) 0xCC, buffer[5]);
        assertEquals((byte) 0xCC, buffer[6]);
        assertEquals((byte) 0xCC, buffer[7]);

        // FileId (16 bytes) - should match our test file ID
        byte[] wireFileId = new byte[16];
        System.arraycopy(buffer, 8, wireFileId, 0, 16);
        assertArrayEquals(testFileId, wireFileId);

class Trans2QueryFSInformationTest {

    /** Small helper to create a byte buffer larger than the maximum expected
     *  wire format so that we can observe only the bytes written by a method.
     */
    private static byte[] newBuffer(int length) {
        // initialise with distinct pattern to detect unused trailing bytes
        byte[] buf = new byte[length];
        Arrays.fill(buf, (byte) 0xFF);
        return buf;
    }

    @Test

        // Write version major (1 byte)
        buffer[bufferIndex++] = (byte) versionMajor;

        // Write version minor (1 byte)
        buffer[bufferIndex++] = (byte) versionMinor;

        // Write type (4 bytes)
        SMBUtil.writeInt4(serverType, buffer, bufferIndex);
        bufferIndex += 4;

        // Write comment offset (4 bytes)

            // Given
            byte[] buffer = new byte[60];
            int bufferIndex = 0;

            // Structure Size (2 bytes) - must be 60
            SMBUtil.writeInt2(60, buffer, bufferIndex);
            // Flags (2 bytes)
            SMBUtil.writeInt2(SMB2_CLOSE_FLAG_POSTQUERY_ATTIB, buffer, bufferIndex + 2);
            // Reserved (4 bytes)
            SMBUtil.writeInt4(0, buffer, bufferIndex + 4);

    }

    @Test
    @DisplayName("Test decode reads correct number of bytes")
    void testDecodeReadsCorrectBytes() throws SMBProtocolDecodingException {
        // Prepare test data
        byte[] buffer = new byte[22];

        // Fill with pattern to verify we read correct bytes
        for (int i = 0; i < buffer.length; i++) {
            buffer[i] = (byte) (i + 1);
        }

        // Decode

        // MECH_OID_LEN (2 bytes big-endian)
        buf[i++] = (byte) ((oidLen >>> 8) & 0xFF);
        buf[i++] = (byte) (oidLen & 0xFF);
        // MECH_OID
        System.arraycopy(oidDer, 0, buf, i, oidLen);
        i += oidLen;
        // NAME_LEN (4 bytes big-endian)
        buf[i++] = (byte) ((nameLen >>> 24) & 0xFF);
        buf[i++] = (byte) ((nameLen >>> 16) & 0xFF);
        buf[i++] = (byte) ((nameLen >>> 8) & 0xFF);