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

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

    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)
        testFileId = new byte[16];
        for (int i = 0; i < 16; i++) {
            testFileId[i] = (byte) (i + 1);
        }

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

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

    @BeforeEach
    void setUp() {
        // Create a fresh pool instance for each test
        pool = new SmbTransportPoolImpl();

        // Setup default mock behaviors
        when(ctx.getConfig()).thenReturn(config);
        when(ctx.getNameServiceClient()).thenReturn(nameSvc);
        when(ctx.getCredentials()).thenReturn(creds);
        when(ctx.getTransportPool()).thenReturn(pool);

							}
							for offset < len(splitDataShards) {
								newConfig := false
								for shardIdx, shard := range splitFilled[:k] {
									if shardConfig[shardIdx] != shard[offset] {
										newConfig = true
										break
									}
								}
								if newConfig {
									if offset > startOffset && reconstructAbleConfig {

        final UniAddress dc = new UniAddress(addr);
        try (SmbTransportInternal trans = tf.getTransportPool()
                .getSmbTransport(tf, dc, 0, false, tf.hasDefaultCredentials() && tf.getConfig().isIpcSigningEnforced())
                .unwrap(SmbTransportInternal.class)) {
            if (!tf.hasDefaultCredentials()) {
                trans.ensureConnected();
                log.warn("""