}

    // Happy path: hasCapability returns based on provided capability codes
    @ParameterizedTest
    @DisplayName("hasCapability returns expected values for various caps")
    @ValueSource(ints = { -1, 0, 1, 1024 })
    void hasCapability_variousCaps(int cap) throws SmbException {
        when(transport.hasCapability(anyInt())).thenAnswer(inv -> ((int) inv.getArgument(0)) >= 0);

    }

    @Test
    @DisplayName("hasCapability delegates and propagates exceptions")
    void hasCapabilityDelegates() throws Exception {
        // Happy path delegates; invalid case propagates SmbException
        when(treeConnection.hasCapability(123)).thenReturn(true);
        assertTrue(handle.hasCapability(123));
        verify(treeConnection).hasCapability(123);

        // Exception path

        when(handle.getTree()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(false);
        when(tree.getConfig()).thenReturn(mock(Configuration.class));
        when(tree.hasCapability(SmbConstants.CAP_NT_SMBS)).thenReturn(true);
        when(handle.getFid()).thenReturn(fid);
        when(tree.send(any(CommonServerMessageBlockRequest.class), any(), any(), any())).thenReturn(resp);
    }

    public void ensureDFSResolved() throws CIFSException {
        this.treeConnection.ensureDFSResolved(this.resourceLoc);
    }

    @Override
    public boolean hasCapability(final int cap) throws SmbException {
        return this.treeConnection.hasCapability(cap);
    }

    /**
     * {@inheritDoc}
     *
     * @see jcifs.SmbTreeHandle#isConnected()
     */
    @Override
    public boolean isConnected() {

        setTree(c2, other);
        assertFalse(c1.isSame(c2));
    }

    @Test
    @DisplayName("hasCapability throws when not connected")
    void hasCapability_notConnected_throws() {
        SmbTreeConnection c = newConn();
        SmbException ex = assertThrows(SmbException.class, () -> c.hasCapability(1));
        assertTrue(ex.getMessage().contains("Not connected"));
    }

    @Test

     * @throws SmbException
     */
    public boolean hasCapability(final int cap) throws SmbException {
        try (SmbSessionImpl s = getSession()) {
            if (s != null) {
                try (SmbTransportImpl transport = s.getTransport()) {
                    return transport.hasCapability(cap);
                }
            }
            throw new SmbException("Not connected");

                        port == 445 && this.port == 139)
                && (localAddr == this.localAddr || localAddr != null && localAddr.equals(this.localAddr)) && localPort == this.localPort;
    }

    boolean hasCapability(final int cap) throws SmbException {
        try {
            connect(RESPONSE_TIMEOUT);
        } catch (final IOException ioe) {
            throw new SmbException(ioe.getMessage(), ioe);
        }

            }

            try (SmbFileHandle h = ensureOpen()) {}
            this.readSize = th.getReceiveBufferSize() - 70;
            this.writeSize = th.getSendBufferSize() - 70;

            if (th.hasCapability(SmbConstants.CAP_LARGE_READX)) {
                this.largeReadX = true;
                this.readSize =

                    req.setFileInformation(new FileBasicInfo(ctime, atime, mtime, 0L, attrs));
                    dh.send(req);
                } else if (dh.hasCapability(SmbConstants.CAP_NT_SMBS)) {
                    // use the open file descriptor
                    dh.send(new Trans2SetFileInformation(dh.getConfig(), dfd.getFid(), attrs, ctime, mtime, atime),

        when(pipe.ensureTreeConnected()).thenReturn(tree);
        when(tree.acquire()).thenReturn(tree);

        // Setup for ensureOpen
        when(tree.isSMB2()).thenReturn(false);
        when(tree.hasCapability(SmbConstants.CAP_NT_SMBS)).thenReturn(true);
        when(pipe.openUnshared(anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenReturn(fh);
        when(fh.acquire()).thenReturn(fh);