private AutoCloseable mocks;

    @BeforeEach
    public void setUp() throws Exception {
        mocks = MockitoAnnotations.openMocks(this);

        when(context.getConfig()).thenReturn(config);
        when(config.isDirectoryNotificationsEnabled()).thenReturn(false);

        // Create real lease manager for integration testing
        baseLeaseManager = new LeaseManager(context);

        when(mockHandle.getFileId()).thenReturn(new byte[16]);

        when(mockTree.getReceiveBufferSize()).thenReturn(0x10000);
        when(mockTree.getMaximumBufferSize()).thenReturn(0x10000);
        when(mockTree.getConfig()).thenReturn(mockConfig);
        when(mockTree.isConnected()).thenReturn(true);
        when(mockTree.getTreeId()).thenReturn(1L);

        when(mockFile.getType()).thenReturn(SmbConstants.TYPE_FILESYSTEM);
    }

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

                new BaseContext(new PropertyConfiguration(System.getProperties())), "domain", "user", "password");
        when(mockCifsContext.getCredentials()).thenReturn(creds);
        when(mockCifsContext.getConfig()).thenReturn(new PropertyConfiguration(System.getProperties()));
        when(mockCifsContext.getNameServiceClient()).thenReturn(mockNameServiceClient);

        ntlmSsp = new NtlmSsp();

        // Use lenient stubbing to avoid UnnecessaryStubbing errors for tests that don't need all mocks
        lenient().when(mockCifsContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockConfig.getDefaultDomain()).thenReturn("DOMAIN");
        lenient().when(mockConfig.isUseUnicode()).thenReturn(true);

        void testConstructor() {
            Smb2EchoResponse response = new Smb2EchoResponse(mockConfig);
            assertNotNull(response);
            assertEquals(mockConfig, response.getConfig());
        }

        @Test
        @DisplayName("Should extend ServerMessageBlock2Response")
        void testInheritance() {
            assertTrue(echoResponse instanceof ServerMessageBlock2Response);
        }

        when(mockConfig.getMaximumVersion()).thenReturn(DialectVersion.SMB311);
        when(mockConfig.getMachineId()).thenReturn(testMachineId);
        when(mockConfig.getRandom()).thenReturn(mockRandom);
        when(mockContext.getConfig()).thenReturn(mockConfig);

        // Mock random for predictable salt generation
        doAnswer(invocation -> {
            byte[] buffer = invocation.getArgument(0);

        void testConstructorWithConfig() {
            ServerMessageBlock2 msg = new TestServerMessageBlock2(mockConfig);
            assertNotNull(msg);
            assertEquals(mockConfig, msg.getConfig());
            assertEquals(0, msg.getCommand());
        }

        @Test
        @DisplayName("Should create with configuration and command")
        void testConstructorWithConfigAndCommand() {

    @Override
    public WebConfigDbm asDBMeta() {
        return WebConfigDbm.getInstance();
    }

    @Override
    public String asTableDbName() {
        return "web_config";
    }

    // ===================================================================================
    //                                                                              Source

        mockContext = Mockito.mock(CIFSContext.class);
        mockConfig = Mockito.mock(Configuration.class);
        mockCredentials = Mockito.mock(Credentials.class);

        Mockito.when(mockContext.getConfig()).thenReturn(mockConfig);
        Mockito.when(mockContext.getCredentials()).thenReturn(mockCredentials);

        // Mock CredentialsInternal for session tests