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

     *
     * @param registration the witness registration
     */
    private void startAsyncNotificationMonitoring(WitnessRegistration registration) {
        String registrationId = registration.getRegistrationId();

        CompletableFuture<Void> asyncTask = CompletableFuture.runAsync(() -> {
            byte[] contextHandle = registration.getContextHandle();

        registrations.put(registrationId, registration);

        log.debug("Mock registered witness: {} for share: {}", registrationId, shareName);
        return registrationId;
    }

    /**
     * Simulate a witness unregistration
     *
     * @param registrationId the registration ID to remove

     */
    public String getRegistrationId() {
        return registrationId;
    }

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

    /**
     * Gets the sequence number.
     *
     * @return the sequence number
     */

import jcifs.dcerpc.ndr.NdrBuffer;
import jcifs.dcerpc.ndr.NdrException;

/**
 * WitnessHeartbeat RPC message implementation for MS-SWN specification.
 * This message is used to send periodic heartbeats to maintain witness registrations.
 */
public class WitnessHeartbeatMessage extends WitnessRpcMessage {

    // Input parameters for WitnessHeartbeat
    private byte[] contextHandle;
    private long sequenceNumber;

        // Whitespace
        testRule.setRuleId("  rule with spaces  ");
        assertEquals("  rule with spaces  ", testRule.getRuleId());
    }

    /**
     * Test multiple registrations of same rule
     */
    public void test_multipleRegistrations() {
        testRule.setRuleId("multiRegRule");

        // Register multiple times
        testRule.register(0);
        testRule.register(0);

    }

    /**
     * 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());

     * 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

        }
    }

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