}

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