}

    @Test
    void testAffinityBasedStrategy() throws Exception {
        loadBalancer.setStrategy(LoadBalancingStrategy.AFFINITY_BASED);

        Smb2ReadRequest readRequest = mock(Smb2ReadRequest.class);
        when(readRequest.getTreeId()).thenReturn(123); // Set consistent tree ID for affinity

        when(mockChannelManager.getHealthyChannels()).thenReturn(Arrays.asList(channel1, channel2));

        NodeStatusRequest readRequest = new NodeStatusRequest(mockConfig, new Name(mockConfig));

        // Act - Read
        int readLength = readRequest.readWireFormat(buffer, 0);

        // Assert read
        assertEquals(0x1234, readRequest.nameTrnId);
        assertEquals(NameServicePacket.NBSTAT, readRequest.questionType);
        assertFalse(readRequest.isRecurDesired);

        void testConstructor() {
            byte[] buffer = new byte[1024];
            Smb2ReadRequest readRequest = new Smb2ReadRequest(mockConfig, testFileId, buffer, 100);
            assertNotNull(readRequest);
            assertTrue(readRequest instanceof ServerMessageBlock2Request);
            assertTrue(readRequest instanceof RequestWithFileId);
        }

        @Test

        RequestWithFileId setInfoRequest = new Smb2SetInfoRequest(mockConfig, testFileId);
        RequestWithFileId queryDirRequest = new Smb2QueryDirectoryRequest(mockConfig, testFileId);
        RequestWithFileId readRequest = new Smb2ReadRequest(mockConfig, testFileId, testOutputBuffer, 0);
        RequestWithFileId writeRequest = new Smb2WriteRequest(mockConfig, testFileId);

    private fun processOneRequest(socket: MockWebServerSocket): Boolean {
      if (socket.source.exhausted()) {
        return false // No more requests on this socket.
      }

      val request =
        readRequest(
          socket = socket,
          connectionIndex = connectionIndex,
          exchangeIndex = nextExchangeIndex++,
        )
      atomicRequestCount.incrementAndGet()
      requestQueue.add(request)

        .build(),
    )
    server.enqueue(
      MockResponse(body = "abc"),
    )
    val headRequest =
      Request
        .Builder()
        .url(server.url("/"))
        .head()
        .build()
    val response = client.newCall(headRequest).execute()
    assertThat(response.code).isEqualTo(200)
    assertArrayEquals(ByteArray(0), response.body.bytes())