}

    @Test
    void testWriteDataWireFormat() {
        // Given
        // writeInt8 writes 8 bytes but needs to access dst[dstIndex+7]
        // Total bytes written: 8+8+8+8+2+6=40
        // But writeInt8 at position 34 needs to access position 34+7=41
        // So we need at least 42 bytes in the array
        byte[] dst = new byte[42];

        // When

     */
    public void setLength(final int length) {
        deferred.length = length;
    }

    /**
     * Advances the buffer index by the specified number of bytes.
     *
     * @param n the number of bytes to advance
     */
    public void advance(final int n) {
        index += n;
        if (index - start > deferred.length) {
            deferred.length = index - start;
        }

        // Prepare test data
        byte[] buffer = new byte[64];
        int bufferIndex = 10;

        // Set capabilities (4 bytes)
        int testCapabilities = 0x12345678;
        SMBUtil.writeInt4(testCapabilities, buffer, bufferIndex);

        // Set server GUID (16 bytes)
        byte[] testGuid = new byte[16];
        for (int i = 0; i < 16; i++) {
            testGuid[i] = (byte) (i + 1);
        }

    /**
     * Reads up to b.length bytes of data from this input stream into an array of bytes.
     *
     * @throws IOException if a network error occurs
     */

    @Override
    public int read(final byte[] b) throws IOException {
        return read(b, 0, b.length);
    }

    /**
     * Reads up to len bytes of data from this input stream into an array of bytes.
     *

    PrimitiveSink sink = mock(PrimitiveSink.class);
    OutputStream out = Funnels.asOutputStream(sink);
    byte[] bytes = {1, 2, 3, 4};
    out.write(255);
    out.write(bytes);
    out.write(bytes, 1, 2);
    verify(sink).putByte((byte) 255);
    verify(sink).putBytes(bytes);
    verify(sink).putBytes(bytes, 1, 2);
  }

  public void testSerialization() {
    assertSame(

                bytes[byteIndex++] = (byte) (0xE0 | (c >> 12));
                bytes[byteIndex++] = (byte) (0x80 | ((c >> 6) & 0x3F));
                bytes[byteIndex++] = (byte) (0x80 | (c & 0x3F));
            }
        }

        return Arrays.copyOf(bytes, byteIndex);
    }

    /**
     * Convert byte array to char array (UTF-8)
     */
    private char[] bytesToChars(byte[] bytes) {
        if (bytes == null) {

    public Smb2LeaseKey() {
        this.key = new byte[LEASE_KEY_SIZE];
        RANDOM.nextBytes(this.key);
    }

    /**
     * Create a lease key from existing bytes
     *
     * @param key 16-byte array
     * @throws IllegalArgumentException if key is not 16 bytes
     */
    public Smb2LeaseKey(byte[] key) {
        if (key == null) {
            throw new IllegalArgumentException("Lease key cannot be null");
        }

        // Test with zero
        byte[] zeroBytes = new byte[] { 0x00, 0x00, 0x00, 0x00 };
        AvFlags zeroAvFlags = new AvFlags(zeroBytes);
        assertNotNull(zeroAvFlags, "AvFlags object should not be null for zero bytes");
        assertEquals(0, zeroAvFlags.getFlags(), "Flags should be 0 for zero bytes");

// suffixSaver is an io.Writer & io.ReaderAt that remembers the last 0
// to 'keep' bytes of data written to it. Call Suffix to get the
// suffix bytes.
type suffixSaver struct {
	keep  int
	buf   []byte
	start int
	size  int64
}

func (ss *suffixSaver) Size() int64 { return ss.size }

var errDiscardedBytes = errors.New("ReadAt of discarded bytes")

func (ss *suffixSaver) ReadAt(p []byte, off int64) (n int, err error) {

        // Verify total data count (4 bytes at position 7)
        assertEquals(200, SMBUtil.readInt4(dst, 7));

        // Verify max parameter count (4 bytes at position 11)
        assertEquals(1024, SMBUtil.readInt4(dst, 11));

        // Verify max data count (4 bytes at position 15)
        assertEquals(2048, SMBUtil.readInt4(dst, 15));

        // Verify parameter count (4 bytes at position 19)