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

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

            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");

    }

    public void setOutputDir(File outputDir) {
        this.outputDir.set(outputDir);
    }

    public void copy(String resource) {
        if (getState().getExecuted() || getState().getExecuting()) {
            throw new GradleException(
                "buildResources can't be configured after the task ran. " + "Make sure task is not used after configuration time"
            );
        }

		var throughputHighestResults []SpeedTestResult

		sendResult := func() {
			var result madmin.SpeedTestResult

			durationSecs := opts.duration.Seconds()

			result.GETStats.ThroughputPerSec = throughputHighestGet / uint64(durationSecs)
			result.GETStats.ObjectsPerSec = throughputHighestGet / uint64(opts.objectSize) / uint64(durationSecs)
			result.PUTStats.ThroughputPerSec = throughputHighestPut / uint64(durationSecs)

        channel1 = new ChannelInfo("channel1", mockTransport1, local, remote1);
        channel1.setState(ChannelState.ESTABLISHED);

        channel2 = new ChannelInfo("channel2", mockTransport2, local, remote2);
        channel2.setState(ChannelState.ESTABLISHED);
    }

    @Test
    void testSingleChannelSelection() {

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