assertTrue(response instanceof TreeConnectResponse);

        // Test all interface methods
        assertNull(response.getService()); // getService returns null in SMB2
        assertEquals(response.getTreeId(), response.getTid());
    }

    /**
     * Helper method to create a valid response buffer
     */
    private byte[] createValidResponseBuffer(int shareType, int shareFlags, int capabilities, int maxAccess) {

        if (message instanceof ServerMessageBlock2Request) {
            ServerMessageBlock2Request smb2Request = (ServerMessageBlock2Request) message;
            affinityKey = smb2Request.getTreeId();
        }

        if (affinityKey != 0) {
            // Select channel based on affinity key
            List<ChannelInfo> list = new ArrayList<>(channels);

        }

        // Test channel selection with mock request when no channels available
        ServerMessageBlock2Request mockRequest = mock(ServerMessageBlock2Request.class);
        when(mockRequest.getTreeId()).thenReturn(123);

        // Should throw NoAvailableChannelException when no healthy channels
        assertThrows(ChannelLoadBalancer.NoAvailableChannelException.class, () -> loadBalancer.selectChannel(mockRequest),

        when(mockTree.getMaximumBufferSize()).thenReturn(0x10000);
        when(mockTree.getConfig()).thenReturn(mockConfig);
        when(mockTree.isConnected()).thenReturn(true);
        when(mockTree.getTreeId()).thenReturn(1L);

        when(mockFile.getType()).thenReturn(SmbConstants.TYPE_FILESYSTEM);
    }

    @Nested
    @DisplayName("Happy path")
    class HappyPath {

        @Test

        @Test
        @DisplayName("setTid should set tree ID")
        void testSetTid() {
            int tid = 12345;
            testRequest.setTid(tid);
            assertEquals(tid, testRequest.getTreeId());
        }
    }

    @Nested
    @DisplayName("Async Operation Tests")
    class AsyncOperationTests {

        @Test
        @DisplayName("isResponseAsync should return false when asyncId is 0")

     */
    void setNonPooled(final boolean np) {
        this.nonPooled = np;
    }

    /**
     * @return the currently connected tid
     */
    @SuppressWarnings("resource")
    public long getTreeId() {
        final SmbTreeImpl t = getTreeInternal();
        if (t == null) {
            return -1;
        }
        return t.getTreeNum();
    }

    /**
     *

        this.command = command;
    }

    /**
     * Gets the tree identifier for this message.
     *
     * @return the treeId
     */
    public final int getTreeId() {
        return this.treeId;
    }

    /**
     * Sets the tree identifier for this message.
     *
     * @param treeId
     *            the treeId to set
     */

        @BeforeEach
        void setUp() {
            response = new Smb2TreeConnectResponse(mockConfig);
        }

        @Test
        @DisplayName("Should return tree ID from getTreeId")
        void testGetTid() {
            // Given
            response.setTreeId(54321);

            // When
            int tid = response.getTid();

            // Then