registration.setState(WitnessRegistrationState.REGISTERED);
        assertEquals(WitnessRegistrationState.REGISTERED, registration.getState());

        registration.setState(WitnessRegistrationState.UNREGISTERING);
        assertEquals(WitnessRegistrationState.UNREGISTERING, registration.getState());

        registration.setState(WitnessRegistrationState.FAILED);

        assertEquals(ChannelState.ESTABLISHED, channelInfo.getState());
        assertTrue(channelInfo.isHealthy());

        channelInfo.setState(ChannelState.ACTIVE);
        assertTrue(channelInfo.isHealthy());

        channelInfo.setState(ChannelState.FAILED);
        assertFalse(channelInfo.isHealthy());
    }

    @Test
    void testActivityTracking() {

        channel.setState(ChannelState.ESTABLISHED);

        // Simulate failure
        IOException failure = new IOException("Connection lost");
        failover.handleFailure(channel, failure);

        // Verify channel state changed
        assertFalse(channel.isHealthy());
        assertEquals(ChannelState.FAILED, channel.getState());
    }

    @Test

        failedChannel.setState(ChannelState.ESTABLISHED);

        when(mockChannelManager.getLoadBalancer()).thenReturn(mockLoadBalancer);
    }

    @Test
    void testHandleFailure() {
        IOException error = new IOException("Connection failed");

        failover.handleFailure(failedChannel, error);

        assertEquals(ChannelState.FAILED, failedChannel.getState());

    while (!isThreadBlockedOnInstanceOf(t, blocker)) {
      if (t.getState() == Thread.State.TERMINATED) {
        throw new RuntimeException("Thread " + t + " exited unexpectedly");
      }
      Thread.sleep(1);
    }
  }

  private static boolean isThreadBlockedOnInstanceOf(Thread t, Class<?> blocker) {
    return t.getState() == Thread.State.WAITING && blocker.isInstance(LockSupport.getBlocker(t));
  }

     *
     * @return the current registration state
     */
    public WitnessRegistrationState getState() {
        return state;
    }

    /**
     * Sets the registration state.
     *
     * @param state the new registration state
     */
    public void setState(WitnessRegistrationState state) {
        this.state = state;
    }

    /**
     * Gets the registration flags.

    assertEquals(THREAD_PRIORITY, thread.getPriority());
    assertEquals(THREAD_DAEMON, thread.isDaemon());
    assertSame(UNCAUGHT_EXCEPTION_HANDLER, thread.getUncaughtExceptionHandler());
    assertThat(thread.getState()).isEqualTo(Thread.State.NEW);

    assertFalse(completed);
    thread.start();
    thread.join();
    assertTrue(completed);
  }

  public void testNulls() {

    assertEquals(THREAD_PRIORITY, thread.getPriority());
    assertEquals(THREAD_DAEMON, thread.isDaemon());
    assertSame(UNCAUGHT_EXCEPTION_HANDLER, thread.getUncaughtExceptionHandler());
    assertThat(thread.getState()).isEqualTo(Thread.State.NEW);

    assertFalse(completed);
    thread.start();
    thread.join();
    assertTrue(completed);
  }

  public void testNulls() {

        return receiveCredits.get();
    }

    /**
     * Get connection state
     *
     * @return current connection state
     */
    public RdmaConnectionState getState() {
        return state;
    }

    /**
     * Get remote address
     *
     * @return remote socket address
     */
    public InetSocketAddress getRemoteAddress() {

            for (int state : disconnectedStates) {
                transport.setState(state);
                assertTrue(transport.isDisconnected(), "State " + state + " should be disconnected");
            }

            for (int state : connectedStates) {
                transport.setState(state);
                assertFalse(transport.isDisconnected(), "State " + state + " should not be disconnected");