}

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

     */
    RDMA_WRITE,

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

    /**
     * Dynamic memory registration
     */
    MEMORY_REGISTRATION,

    /**
     * Fast memory region registration
     */
    FAST_REGISTRATION

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

    }

    /**
     * Safely registers a class with Kryo, logging any registration failures at WARN level.
     * <p>
     * This method catches exceptions for individual class registrations to ensure
     * that a failure to register one class doesn't prevent other classes from being registered.
     * Registration failures are logged at WARN level since they may cause serialization errors later.
     * </p>
     *

==============================================================================*/

#ifndef TENSORFLOW_C_EAGER_CUSTOM_DEVICE_TESTUTIL_H_
#define TENSORFLOW_C_EAGER_CUSTOM_DEVICE_TESTUTIL_H_

// A simple logging device to test custom device registration.
#include <memory>

#include "tensorflow/c/c_api.h"
#include "tensorflow/c/eager/c_api.h"
#include "tensorflow/c/eager/c_api_experimental.h"
#include "tensorflow/c/tf_status.h"

        return clientComputerName;
    }

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

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

├── WitnessRegistration.java     - Client registration with witness
├── WitnessNotification.java     - Notification handling
├── ResourceChange.java          - Resource change events
└── WitnessTransport.java       - Witness-specific RPC transport
```

#### 6.2 Implementation Tasks
- [ ] Implement Witness RPC protocol
- [ ] Create witness service discovery
- [ ] Implement client registration
- [ ] Add notification callback mechanism

        this.heartbeatInterval = 0;
    }

    /**
     * Creates a new WitnessHeartbeat RPC message with the specified parameters.
     *
     * @param contextHandle the context handle from registration
     * @param sequenceNumber the sequence number for this heartbeat
     */
    public WitnessHeartbeatMessage(byte[] contextHandle, long sequenceNumber) {
        super(WITNESS_HEARTBEAT);

 * memory regions for high-performance RDMA operations.
 *
 * Note: This is a skeleton implementation. A real implementation would
 * require proper DiSNI integration with actual memory registration.
 */
public class DisniMemoryRegion extends RdmaMemoryRegion {

    private static final Logger log = LoggerFactory.getLogger(DisniMemoryRegion.class);