// Set system property for handle state directory
        System.setProperty("jcifs.smb.client.handleStateDirectory", tempDir.toString());

        when(mockContext.getConfig()).thenReturn(mockConfig);

        manager = new PersistentHandleManager(mockContext);
    }

    @AfterEach
    public void tearDown() throws IOException {
        if (manager != null) {
            manager.shutdown();
        }

    private Configuration mockConfig;
    private CIFSContext mockContext;
    private Smb2TreeConnectRequest request;
    private static final String TEST_PATH = "\\\\server\\share";

    @BeforeEach
    void setUp() {
        mockConfig = mock(Configuration.class);
        mockContext = mock(CIFSContext.class);
        when(mockContext.getConfig()).thenReturn(mockConfig);

    @Mock
    private Configuration mockConfig;

    @Mock
    private CIFSContext mockContext;

    private byte[] testFileId;
    private byte[] outputBuffer;
    private Smb2ReadRequest request;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockContext.getConfig()).thenReturn(mockConfig);

        testFileId = new byte[16];

class Smb2FlushRequestTest {

    private Configuration mockConfig;
    private CIFSContext mockContext;
    private byte[] testFileId;
    private Smb2FlushRequest request;

    @BeforeEach
    void setUp() {
        mockConfig = mock(Configuration.class);
        mockContext = mock(CIFSContext.class);
        when(mockContext.getConfig()).thenReturn(mockConfig);

        // Create a test file ID (16 bytes)

        multiConfig = new PropertyConfiguration(props);

        // Mock context and session for ChannelManager
        when(mockContext.getConfig()).thenReturn(multiConfig);
        when(mockSession.getSessionKey()).thenReturn(new byte[16]);

        channelManager = new ChannelManager(mockContext, mockSession);
    }

    @Test
    void testMultiChannelInitialization() throws Exception {

@DisplayName("LeaseManager Tests")
class LeaseManagerTest {

    private LeaseManager leaseManager;
    private CIFSContext mockContext;
    private Configuration mockConfig;

    @BeforeEach
    void setUp() {
        mockContext = mock(CIFSContext.class);
        mockConfig = mock(Configuration.class);
        when(mockContext.getConfig()).thenReturn(mockConfig);
        when(mockConfig.getLeaseTimeout()).thenReturn(30000);

        Configuration mockConfig = mock(Configuration.class);
        CIFSContext mockContext = mock(CIFSContext.class);
        when(mockContext.getConfig()).thenReturn(mockConfig);
        Smb2TreeDisconnectRequest request = new Smb2TreeDisconnectRequest(mockConfig);

        // When
        Smb2TreeDisconnectResponse response = request.createResponse(mockContext, request);

        // Then
        assertNotNull(response);

    private Configuration mockConfig;

    @Mock
    private CIFSContext mockContext;

    @Mock
    private Smb2Lock mockLock;

    private byte[] testFileId;
    private Smb2Lock[] testLocks;
    private Smb2LockRequest request;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockContext.getConfig()).thenReturn(mockConfig);

        testFileId = new byte[16];

        Smb2IoctlRequest request = new Smb2IoctlRequest(mockConfig, TEST_CONTROL_CODE, TEST_FILE_ID, TEST_OUTPUT_BUFFER);

        Smb2IoctlResponse response = request.createResponse(mockContext, request);

        assertNotNull(response);
        verify(mockContext).getConfig();
    }

    @Test
    @DisplayName("Test setFlags method")
    void testSetFlags() {
        setupMockConfig();

class Smb2QueryInfoRequestTest {

    private Configuration mockConfig;
    private CIFSContext mockContext;
    private Smb2QueryInfoRequest request;
    private byte[] testFileId;

    @BeforeEach
    void setUp() {
        mockConfig = mock(Configuration.class);
        mockContext = mock(CIFSContext.class);
        when(mockContext.getConfig()).thenReturn(mockConfig);
        when(mockConfig.getMaximumBufferSize()).thenReturn(65536);