}

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

               resourceName.equals(registration.getServerAddress().getHostAddress());
    }
    
    private void checkHeartbeats() {
        long timeout = context.getConfig().getWitnessHeartbeatTimeout();
        
        for (WitnessRegistration registration : registrations.values()) {
            if (registration.isExpired(timeout)) {

        return this.rdmaPort;
    }

    @Override
    public boolean isUseWitness() {
        return this.useWitness;
    }

    @Override
    public long getWitnessHeartbeatTimeout() {
        return this.witnessHeartbeatTimeout;
    }

    @Override
    public long getWitnessRegistrationTimeout() {
        return this.witnessRegistrationTimeout;
    }

        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockConfig.isUseWitness()).thenReturn(true);
        lenient().when(mockConfig.getWitnessHeartbeatTimeout()).thenReturn(120000L);
        lenient().when(mockConfig.getWitnessRegistrationTimeout()).thenReturn(300000L);
        lenient().when(mockConfig.getWitnessReconnectDelay()).thenReturn(1000L);

        // Setup mock configuration with lenient stubbing
        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockConfig.getWitnessHeartbeatTimeout()).thenReturn(120000L);
        lenient().when(mockConfig.getWitnessRegistrationTimeout()).thenReturn(300000L);
        lenient().when(mockConfig.getWitnessReconnectDelay()).thenReturn(1000L);