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);

    public static final byte TRANS2_SET_FILE_INFORMATION = (byte) 0x08;

    /**
     * Network share enumeration subcommand
     */
    public static final byte NET_SHARE_ENUM = (byte) 0x00;
    /**
     * Network server enumeration subcommand version 2
     */
    public static final byte NET_SERVER_ENUM2 = (byte) 0x68;
    /**
     * Network server enumeration subcommand version 3
     */

    void testSetFileId() {
        request = new Smb2QueryDirectoryRequest(mockConfig);
        byte[] fileId = new byte[16];
        for (int i = 0; i < 16; i++) {
            fileId[i] = (byte) (i + 1);
        }

        request.setFileId(fileId);

        // Verify by writing to buffer and checking the fileId position
        byte[] buffer = new byte[1024];
        request.writeBytesWireFormat(buffer, Smb2Constants.SMB2_HEADER_LENGTH);

     */
    public void setContext(final byte[] context) {
        this.context = context;
    }

    @Override
    public byte[] toByteArray() {
        try {
            final String targetName = getTarget();
            final byte[] challenge = getChallenge();
            byte[] context = getContext();
            final byte[] targetInformation = getTargetInformation();

    @Override
    int writeSetupWireFormat(final byte[] dst, final int dstIndex) {
        return 0;
    }

    @Override
    int writeParametersWireFormat(final byte[] dst, final int dstIndex) {
        return 0;
    }

    @Override
    int writeDataWireFormat(final byte[] dst, final int dstIndex) {
        return 0;
    }

    @Override
    int readSetupWireFormat(final byte[] buffer, final int bufferIndex, final int len) {

            assertEquals((byte) 0xFE, buffer[0]);
            assertEquals((byte) 'S', buffer[1]);
            assertEquals((byte) 'M', buffer[2]);
            assertEquals((byte) 'B', buffer[3]);
        }

        @Test
        @DisplayName("Should encode with signature when digest is set")
        void testEncodeWithSignature() {
            byte[] buffer = new byte[1024];

        // Create a large byte array (1MB)
        byte[] data = new byte[1024 * 1024];
        for (int i = 0; i < data.length; i++) {
            data[i] = (byte) (i % 256);
        }

        ContentCache cache = new ContentCache(data);

        try (InputStream stream = cache.getInputStream()) {
            assertNotNull(stream);
            byte[] buffer = new byte[data.length];

        final byte b0 = DECODE_TABLE[inData.charAt(inIndex)];
        final byte b1 = DECODE_TABLE[inData.charAt(inIndex + 1)];
        final byte b2 = DECODE_TABLE[inData.charAt(inIndex + 2)];
        final byte b3 = DECODE_TABLE[inData.charAt(inIndex + 3)];
        outData[outIndex] = (byte) (b0 << 2 & 0xfc | b1 >> 4 & 0x3);

                byte[] dst = new byte[buffer.size()];
                buffer.encode(dst, 0);

                // Verify SMBUtil.writeInt2 was called for referral level (with long parameter)
                mockedSMBUtil.verify(() -> SMBUtil.writeInt2(eq((long) maxReferralLevel), any(byte[].class), eq(0)));

            // Given
            byte[] originalBuffer = new byte[100];
            for (int i = 0; i < originalBuffer.length; i++) {
                originalBuffer[i] = (byte) i;
            }
            byte[] buffer = originalBuffer.clone();

            // When
            Method readParams = SmbComNtCancel.class.getDeclaredMethod("readParameterWordsWireFormat", byte[].class, int.class);