// Clean up expired registration
                stopAsyncNotificationMonitoring(registration.getRegistrationId());
                registrations.remove(registration.getRegistrationId());
                listeners.remove(registration.getRegistrationId());
            }
        }
    }

    /**
     * Monitors active registrations and sends heartbeats.
     */
    private void monitorRegistrations() {

                    registrations.put(registration.getRegistrationId(), registration);
                    listeners.put(registration.getRegistrationId(), listener);
                    
                    log.info("Successfully registered for witness notifications: {}", 
                        registration.getRegistrationId());
                    
                    return registration;

        // Setup mock RPC response
        WitnessRegisterResponse mockResponse = mock(WitnessRegisterResponse.class);
        lenient().when(mockResponse.isSuccess()).thenReturn(true);
        lenient().when(mockResponse.getRegistrationId()).thenReturn("test-reg-123");

        WitnessRpcClient mockRpc = mock(WitnessRpcClient.class);
        lenient().when(mockRpc.register(any(WitnessRegisterRequest.class))).thenReturn(mockResponse);

                new WitnessRegistration("\\\\server\\share2", InetAddress.getByName("192.168.1.101"), WitnessServiceType.CLUSTER_WITNESS);

        assertNotEquals(registration.getRegistrationId(), registration2.getRegistrationId());
    }

    @Test
    void testFlagsModification() {
        assertEquals(WitnessRegistration.WITNESS_REGISTER_IP_NOTIFICATION, registration.getFlags());

    }

    private String registrationId;
    private byte[] contextHandle;

    /**
     * Gets the registration ID to unregister.
     *
     * @return the registration ID
     */
    public String getRegistrationId() {
        return registrationId;
    }

    /**
     * Sets the registration ID to unregister.
     *
     * @param registrationId the registration ID
     */

    private long sequenceNumber;
    private byte[] contextHandle;

    /**
     * Gets the registration ID.
     *
     * @return the registration ID
     */
    public String getRegistrationId() {
        return registrationId;
    }

    /**
     * Sets the registration ID.
     *
     * @param registrationId the registration ID
     */

     */
    public long getLastHeartbeat() {
        return lastHeartbeat;
    }

    /**
     * Gets the registration ID.
     *
     * @return the registration ID
     */
    public String getRegistrationId() {
        return registrationId;
    }

    /**
     * Gets the share name.
     *
     * @return the share name
     */
    public String getShareName() {
        return shareName;

     */
    public void setError(String error) {
        this.error = error;
    }

    /**
     * Gets the registration ID.
     *
     * @return the registration ID
     */
    public String getRegistrationId() {
        return registrationId;
    }

    /**
     * Sets the registration ID.
     *
     * @param registrationId the registration ID
     */

        if (!connected) {
            throw new IOException("Witness client not connected");
        }

        try {
            log.debug("Unregistering witness: {}", request.getRegistrationId());

            // Create and populate the RPC message
            WitnessUnregisterMessage message = new WitnessUnregisterMessage();
            message.setContextHandle(request.getContextHandle());

            log.error("Witness registration failed: {}", registration.getRegistrationId(), error);
        }

        @Override
        public void onRegistrationExpired(WitnessRegistration registration) {
            log.warn("Witness registration expired: {}", registration.getRegistrationId());
            // Could attempt re-registration here
        }
    }

    /**