}

        @Test
        @DisplayName("Should get and set tree ID")
        void testTreeIdProperty() {
            assertEquals(0, testMessage.getTreeId());
            testMessage.setTreeId(123);
            assertEquals(123, testMessage.getTreeId());
        }

        @Test
        @DisplayName("Should propagate tree ID to chained message")
        void testTreeIdPropagation() {

        assertTrue(h.isValid());

        // Tree id change makes it invalid
        when(tree.getTreeId()).thenReturn(11L);
        assertFalse(h.isValid());

        // Connected=false makes it invalid
        when(tree.getTreeId()).thenReturn(10L);
        when(tree.isConnected()).thenReturn(false);
        assertFalse(h.isValid());

        // Mark closed makes it invalid

     * {@inheritDoc}
     *
     * @see jcifs.internal.TreeConnectResponse#getTid()
     */
    @Override
    public final int getTid() {
        return getTreeId();
    }

    @Override
    public boolean isValidTid() {
        return getTreeId() != -1;
    }

    /**
     * {@inheritDoc}
     *
     * @see jcifs.internal.TreeConnectResponse#getService()
     */
    @Override

        Smb2CancelRequest request = new Smb2CancelRequest(mockConfig, 1L, 0L);
        int treeId = 42;

        // When
        request.setTid(treeId);

        // Then
        assertEquals(treeId, request.getTreeId(), "Tree ID should be set correctly");
    }

    @Test
    @DisplayName("Test isCancel returns true")
    void testIsCancel() {
        // Given

    }

    @Test
    @DisplayName("getTreeId delegates")
    void getTreeIdDelegates() {
        // Ensures tree id is forwarded from connection
        when(treeConnection.getTreeId()).thenReturn(42L);
        assertEquals(42L, handle.getTreeId());
        verify(treeConnection).getTreeId();
    }

    @Test

    @Override
    public Configuration getConfig() {
        return this.treeConnection.getConfig();
    }

    /**
     * @return the currently connected tree id
     */
    public long getTreeId() {
        return this.treeConnection.getTreeId();
    }

    /**
     *
     * @param req
     * @param params
     * @return response
     * @throws CIFSException
     */

        this.unc = unc;
        this.flags = flags;
        this.access = access;
        this.attrs = attrs;
        this.options = options;
        this.tree = tree.acquire();
        this.tree_num = tree.getTreeId();

        if (cfg.isTraceResourceUsage()) {
            this.creationBacktrace = Thread.currentThread().getStackTrace();
        } else {
            this.creationBacktrace = null;
        }
    }

    /**

    }

    @Test
    @DisplayName("getTreeId(): no tree returns -1")
    void getTreeId_noTree_returnsMinusOne() {
        // Arrange
        CIFSContext ctx = mock(CIFSContext.class);
        SmbTreeConnectionTrace trace = new SmbTreeConnectionTrace(ctx);

        // Act & Assert
        assertEquals(-1L, trace.getTreeId());
    }

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

        setTree(c, tree);
        assertTrue(c.isConnected());
    }

    @Test
    @DisplayName("getTreeId returns -1 when no tree")
    void getTreeId_noTree() {
        SmbTreeConnection c = newConn();
        assertEquals(-1, c.getTreeId());
    }

    @Test
    @DisplayName("getTreeType and getConnectedShare delegate to tree")
    void getters_delegateToTree() {