this.lastNotification = notification;
            notificationLatch.countDown();
        }

        @Override
        public void onRegistrationFailed(WitnessRegistration registration, Exception error) {
            // Test implementation
        }

        @Override
        public void onRegistrationExpired(WitnessRegistration registration) {

    private static final int DEFAULT_WITNESS_PORT = 135;  // RPC endpoint mapper
    
    public interface WitnessNotificationListener {
        void onWitnessNotification(WitnessNotification notification);
        void onRegistrationFailed(WitnessRegistration registration, Exception error);
        void onRegistrationExpired(WitnessRegistration registration);
    }
    
    public WitnessClient(InetAddress witnessServer, CIFSContext context) {

        /**
         * Called when a witness registration fails.
         *
         * @param registration the failed registration
         * @param error the error that occurred
         */
        void onRegistrationFailed(WitnessRegistration registration, Exception error);

        /**
         * Called when a witness registration expires.
         *
         * @param registration the expired registration
         */

        @Override
        public void onWitnessNotification(WitnessNotification notification) {
            handleWitnessEvent(notification);
        }

        @Override
        public void onRegistrationFailed(WitnessRegistration registration, Exception error) {
            log.error("Witness registration failed: {}", registration.getRegistrationId(), error);
        }

        @Override