void testCompleteWireFormatWriting() {
        trans2QueryFSInfo = new Trans2QueryFSInformation(config, FileSystemInformation.FS_SIZE_INFO);

        byte[] setupBuffer = new byte[256];
        byte[] paramBuffer = new byte[256];
        byte[] dataBuffer = new byte[256];

        int setupWritten = trans2QueryFSInfo.writeSetupWireFormat(setupBuffer, 0);
        int paramWritten = trans2QueryFSInfo.writeParametersWireFormat(paramBuffer, 0);

    }

    @Test
    @DisplayName("Should write empty bytes to wire format")
    void testWriteBytesWireFormat() {
        // Given
        byte[] buffer = new byte[256];
        int offset = 10;

        // When
        int bytesWritten = response.writeBytesWireFormat(buffer, offset);

        // Then
        assertEquals(0, bytesWritten);
    }

    @Test

        // Test with binary data containing all byte values
        byte[] data = new byte[256];
        for (int i = 0; i < 256; i++) {
            data[i] = (byte) i;
        }

        ContentCache cache = new ContentCache(data);

        try (InputStream stream = cache.getInputStream()) {
            for (int i = 0; i < 256; i++) {
                int byteValue = stream.read();

            when(mockLock.encode(any(byte[].class), anyInt())).thenReturn(24);

            Smb2LockRequest req = new Smb2LockRequest(mockConfig, testFileId, locks);
            byte[] buffer = new byte[256];

            int bytesWritten = req.writeBytesWireFormat(buffer, 0);

            // Verify structure
            assertEquals(48, SMBUtil.readInt2(buffer, 0)); // Structure size

        private byte[] createCompoundMessage() {
            byte[] buffer = new byte[256];
            System.arraycopy(SMBUtil.SMB2_HEADER, 0, buffer, 0, 4);
            SMBUtil.writeInt4(128, buffer, 20); // next command offset
            return buffer;
        }

        private byte[] createMisalignedCompoundMessage() {
            byte[] buffer = new byte[256];
            System.arraycopy(SMBUtil.SMB2_HEADER, 0, buffer, 0, 4);

        netShareEnum = new NetShareEnum(realConfig);

        // Test with offset 0
        byte[] dst1 = new byte[256];
        int bytesWritten1 = netShareEnum.writeParametersWireFormat(dst1, 0);

        // Test with offset 100
        byte[] dst2 = new byte[256];
        int bytesWritten2 = netShareEnum.writeParametersWireFormat(dst2, 100);

        // Same number of bytes should be written

     * @param klen the length of the key to use
     */
    public void init(final byte[] key, final int ki, final int klen) {
        s = new byte[256];

        for (i = 0; i < 256; i++) {
            s[i] = (byte) i;
        }

        for (i = j = 0; i < 256; i++) {
            j = j + key[ki + i % klen] + s[i] & 0xff;
            final byte t = s[i];
            s[i] = s[j];
            s[j] = t;
        }

    }

    @Test
    @DisplayName("Test writeParameterWordsWireFormat for primary NT transaction")
    void testWriteParameterWordsWireFormatPrimary() {
        byte[] dst = new byte[256];

        // Set up transaction parameters
        transaction.setMaxSetupCount((byte) 2);
        transaction.setTotalParameterCount(100);
        transaction.setTotalDataCount(200);

        void testConcurrentEncoding() {
            TestCreateContextRequest request = new TestCreateContextRequest("CONCURRENT".getBytes(StandardCharsets.UTF_8));

            byte[] buffer1 = new byte[256];
            byte[] buffer2 = new byte[256];

            // Encode to different buffers
            int result1 = request.encode(buffer1, 0);
            int result2 = request.encode(buffer2, 50);

        @Test
        @DisplayName("Should handle empty data in wire format")
        void testWriteBytesWireFormatEmptyData() {
            request.setData(new byte[0], 0, 0);

            byte[] buffer = new byte[256];
            int bytesWritten = request.writeBytesWireFormat(buffer, 0);

            assertEquals(48, bytesWritten); // Only header, no data
            assertEquals(49, SMBUtil.readInt2(buffer, 0)); // Structure size