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;

package com.google.common.testing;

import com.google.common.annotations.GwtCompatible;
import com.google.errorprone.annotations.DoNotMock;
import org.jspecify.annotations.NullMarked;

/**
 * Any object which can accept registrations of {@link TearDown} instances.
 *
 * @author Kevin Bourrillion
 * @since 10.0
 */
@DoNotMock("Implement with a lambda")
@GwtCompatible
@NullMarked
public interface TearDownAccepter {
  /**

        ScriptEngine retrieved = scriptEngineFactory.getScriptEngine("customName");
        assertNotNull(retrieved);
        assertEquals(customEngine, retrieved);
    }

    // Test multiple registrations of the same engine
    public void test_multipleRegistrations() {
        testScriptEngine.register();
        testScriptEngine.register(); // Register again

        // Should still work without errors

    }

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

    }

    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

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