}

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

    /**
     * Sets the registration ID.
     *
     * @param registrationId the registration ID
     */
    public void setRegistrationId(String registrationId) {
        this.registrationId = registrationId;
    }

    /**

        assertNotNull(registration.getRegistrationId());
        assertEquals("\\\\server\\share", registration.getShareName());
        assertEquals(serverAddress, registration.getServerAddress());
        assertEquals(WitnessServiceType.FILE_SERVER_WITNESS, registration.getServiceType());
        assertEquals(WitnessVersion.VERSION_2, registration.getVersion());
        assertEquals(WitnessRegistrationState.REGISTERING, registration.getState());

        }
    }

    /**
     * Performs witness registration.
     *
     * @param request the registration request
     * @return the registration response
     * @throws IOException if the RPC call fails
     */
    public WitnessRegisterResponse register(WitnessRegisterRequest request) throws IOException {

    }

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

     * Enumeration of possible witness registration states.
     */
    public enum WitnessRegistrationState {
        /** Registration is in progress */
        REGISTERING,
        /** Registration is active and confirmed */
        REGISTERED,
        /** Unregistration is in progress */
        UNREGISTERING,
        /** Registration has failed */
        FAILED,
        /** Registration has expired due to timeout */
        EXPIRED

        WitnessRegistration registration = future.get(5, TimeUnit.SECONDS);

        assertNotNull(registration);
        assertEquals(WitnessRegistrationState.REGISTERED, registration.getState());
        assertEquals("\\\\server\\share", registration.getShareName());
        assertEquals(serverAddress, registration.getServerAddress());

    }

    /**
     * Test the complete workflow of witness registration, notification, and cleanup
     */
    @Test
    void testCompleteWitnessWorkflow() throws Exception {
        // Simulate complete witness workflow

        // 1. Service discovery and registration
        assertEquals(0, mockService.getRegistrationCount());

        String regId =

    public void setServerAddress(String serverAddress) {
        this.serverAddress = serverAddress;
    }

    /**
     * Gets the registration flags.
     *
     * @return the flags
     */
    public int getFlags() {
        return flags;
    }

    /**
     * Sets the registration flags.
     *
     * @param flags the flags
     */
    public void setFlags(int flags) {
        this.flags = flags;
    }

     * @param contextHandle the context handle from registration
     */
    public WitnessUnregisterMessage(byte[] contextHandle) {
        super(WITNESS_UNREGISTER);
        this.contextHandle = contextHandle != null ? contextHandle.clone() : new byte[20];
    }

    /**
     * Sets the context handle for unregistration.
     *
     * @param contextHandle the context handle from registration
     */

     */
    RDMA_WRITE,

    /**
     * Traditional send/receive with RDMA
     */
    RDMA_SEND_RECEIVE,

    /**
     * Dynamic memory registration
     */
    MEMORY_REGISTRATION,

    /**
     * Fast memory region registration
     */
    FAST_REGISTRATION