private DirectoryLeaseManager directoryLeaseManager;
    private AutoCloseable mocks;

    @BeforeEach
    public void setUp() {
        mocks = MockitoAnnotations.openMocks(this);
        when(context.getConfig()).thenReturn(config);
        when(config.isDirectoryNotificationsEnabled()).thenReturn(false); // Disable for unit tests

        directoryLeaseManager = new DirectoryLeaseManager(context, baseLeaseManager);
    }

        Arrays.fill(testFileId, (byte) 0xAB);

        // Use lenient stubbing for default configuration values
        lenient().when(mockConfig.getNotifyBufferSize()).thenReturn(8192);
        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);

        request = new Smb2ChangeNotifyRequest(mockConfig, testFileId);
    }

    @Test
    @DisplayName("Should create request with configuration and file ID")

            Thread t = new Thread(r, "LeaseManager-Cleanup");
            t.setDaemon(true);
            return t;
        });

        // Schedule periodic cleanup
        long cleanupInterval = context.getConfig().getLeaseTimeout();
        if (cleanupInterval <= 0) {
            cleanupInterval = DEFAULT_LEASE_CLEANUP_INTERVAL;
        }

        final long finalCleanupInterval = cleanupInterval;

        props.setProperty("jcifs.smb.client.channelHealthCheckInterval", "5");

        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

        final Map<String, Object> newDataMap = new LinkedHashMap<>();
        for (final Map.Entry<String, Object> e : dataMap.entrySet()) {
            if (fieldConfigs.getConfig(e.getKey()).map(FieldConfigs.Config::isOverwrite).orElse(false)
                    && e.getValue() instanceof final Object[] values && values.length > 0) {
                newDataMap.put(e.getKey(), values[values.length - 1]);

        when(pipe.getPipeType()).thenReturn(SmbPipeResource.PIPE_TYPE_RDWR);
        when(pipe.getUncPath()).thenReturn("\\\\pipe\\\\my-pipe");

        // Setup tree with configuration
        when(tree.getConfig()).thenReturn(config);
        when(tree.getSendBufferSize()).thenReturn(65536);
        when(config.getPid()).thenReturn(12345);
        when(config.getSendBufferSize()).thenReturn(65536);

    private byte[] testFileId;

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

        testFileId = new byte[16];

    }

    @Override
    protected Smb2IoctlResponse createResponse(final CIFSContext tc, final ServerMessageBlock2Request<Smb2IoctlResponse> req) {
        return new Smb2IoctlResponse(tc.getConfig(), this.outputBuffer, this.controlCode);
    }

    /**
     * Set the IOCTL flags
     * @param flags the flags to set
     */
    public void setFlags(final int flags) {
        this.flags = flags;

        return shareMatch || addressMatch;
    }

    /**
     * Checks for expired registrations based on heartbeat timeouts.
     */
    private void checkHeartbeats() {
        long timeout = context.getConfig().getWitnessHeartbeatTimeout();

        for (WitnessRegistration registration : registrations.values()) {
            if (registration.isExpired(timeout)) {

    void testConstructor() {
        // Given & When
        Smb2FlushResponse flushResponse = new Smb2FlushResponse(mockConfig);

        // Then
        assertNotNull(flushResponse);
        // Cannot test getConfig() as it's protected
    }

    @Nested
    @DisplayName("writeBytesWireFormat tests")
    class WriteBytesWireFormatTests {

        @Test
        @DisplayName("Should return 0 for write bytes")