when(mockFileInfo.getFileInformationLevel()).thenReturn((byte) FileInformation.FILE_BASIC_INFO);
        when(mockFileInfo.encode(any(byte[].class), anyInt())).thenReturn(50);

        trans2SetFileInfo = new Trans2SetFileInformation(config, TEST_FID, mockFileInfo);

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

        }

        @Test
        @DisplayName("Should decode at different buffer offset")
        void testDecodeWithOffset() throws SMBProtocolDecodingException {
            byte[] buffer = new byte[256];
            int offset = 100;
            byte[] responseData = createValidCopyChunkResponse(10, 131072, 1310720);
            System.arraycopy(responseData, 0, buffer, offset, responseData.length);

        assertEquals(1, response.setupCount);
    }

    // Helper method to write a 4-byte integer to a byte array.
    private void writeInt4(int val, byte[] dst, int dstIndex) {
        dst[dstIndex] = (byte) val;
        dst[dstIndex + 1] = (byte) (val >> 8);
        dst[dstIndex + 2] = (byte) (val >> 16);
        dst[dstIndex + 3] = (byte) (val >> 24);
    }

        Smb2SessionSetupResponse resp = newResponse();

        byte[] buf = new byte[256];
        int headerStart = 0;
        long expectedSessionId = 0x1122334455667788L;
        buildHeader(buf, headerStart, NtStatus.NT_STATUS_MORE_PROCESSING_REQUIRED, 0x0001, expectedSessionId);

        int bodyStart = headerStart + Smb2Constants.SMB2_HEADER_LENGTH;
        byte[] blob = new byte[] { 10, 20, 30, 40, 50 };

import (
	"bufio"
	"bytes"
	"fmt"
	"io"
	"sync/atomic"
	"testing"
)

var (
	smallInput  = make([]byte, 1024*12)       // 12K
	mediumInput = make([]byte, 1024*1024*12)  // 12MB
	largeInput  = make([]byte, 1024*1024*128) // 128MB
)

func TestScanner(t *testing.T) {
	t.Skip("Unstable test")
	data := []byte("abcdefghijklmnopqrstuvwxyz0123456789")

	var i int
	r := bytes.NewReader(data)
	scanner := newScanner(r)

        int offset = 10;
        byte[] buffer = new byte[34]; // 10 offset + 16 header + 8 data
        int bufferIndex = offset;

        // Set up test values
        int namedPipeState = 0x02;
        int readDataAvailable = 256;
        int numberOfMessages = 5;
        int messageLength = 128;
        byte[] testData = { (byte) 0xFF, (byte) 0xEE, (byte) 0xDD, (byte) 0xCC, (byte) 0xBB, (byte) 0xAA, (byte) 0x99, (byte) 0x88 };

        assertEquals(0, emptyRead);
        assertEquals(0, fullRead);
    }

    /**
     * Create a test byte array with specified size and pattern
     */
    private byte[] createTestData(int size) {
        byte[] data = new byte[size];
        for (int i = 0; i < size; i++) {
            data[i] = (byte) (i % 256);
        }
        return data;
    }

        byte subCmd = (byte) getPrivateField(cmd, "subCommand");
        assertEquals(subCmd, buf[0]);
        assertEquals((byte) 0x00, buf[1]);
    }

    @Test
    @DisplayName("writeParametersWireFormat writes referral level and path")
    void testWriteParams() throws Exception {
        Trans2GetDfsReferral cmd = new Trans2GetDfsReferral("abc");
        setPrivateField(cmd, "maxReferralLevel", 0x12AB);

            byte[] testContext =
                    new byte[] { (byte) 0x11, (byte) 0x22, (byte) 0x33, (byte) 0x44, (byte) 0x55, (byte) 0x66, (byte) 0x77, (byte) 0x88 };
            message.setContext(testContext);
            byte[] testTargetInfo = new byte[] { 0x10, 0x20, 0x30 };
            message.setTargetInformation(testTargetInfo);

        byte[] src = new byte[10];
        int srcIndex = 0;

        // Simulate groupName = false, nodeType = 0
        src[srcIndex] = (byte) 0x00;
        src[srcIndex + 1] = (byte) 0x00;

        // Simulate address = 0.0.0.0
        src[srcIndex + 2] = (byte) 0x00;
        src[srcIndex + 3] = (byte) 0x00;
        src[srcIndex + 4] = (byte) 0x00;
        src[srcIndex + 5] = (byte) 0x00;