import os
import shutil

from fastapi.testclient import TestClient


def test_main():
    if os.path.isdir("./static"):  # pragma: nocover
        shutil.rmtree("./static")
    if os.path.isdir("./templates"):  # pragma: nocover
        shutil.rmtree("./templates")
    shutil.copytree("./docs_src/templates/templates/", "./templates")
    shutil.copytree("./docs_src/templates/static/", "./static")

        # removed or not.
    else:
        dist_path = site_path / lang
        shutil.rmtree(dist_path, ignore_errors=True)
    current_dir = os.getcwd()
    os.chdir(lang_path)
    shutil.rmtree(build_site_dist_path, ignore_errors=True)
    subprocess.run(["mkdocs", "build", "--site-dir", build_site_dist_path], check=True)

package jcifs;

/**
 * Opaque reference to a SMB tree
 *
 * @author mbechler
 *
 */
public interface SmbTree extends AutoCloseable {

    /**
     * Unwraps the tree instance to the specified type
     * @param <T> the target tree type
     * @param type the class of the tree type to unwrap to
     * @return tree instance with the given type
     */
    <T extends SmbTree> T unwrap(Class<T> type);

    /**

    }

    @Override
    public boolean equals(final Object obj) {
        if (obj instanceof final SmbTree tree) {
            return matches(tree.share, tree.service);
        }
        return false;
    }

    void send(final ServerMessageBlock request, final ServerMessageBlock response) throws SmbException {

    }

    @Test
    void testEquals() {
        SmbTree tree1 = new SmbTree(session, "testShare", "testService");
        SmbTree tree2 = new SmbTree(session, "testShare", "testService");
        SmbTree tree3 = new SmbTree(session, "otherShare", "testService");
        SmbTree tree4 = new SmbTree(session, "testShare", "otherService");

        // Test equal trees (same share and service)

        verify(smbTree).unwrap(CustomSmbTree.class);
    }

    /**
     * Test for unwrap() method with the same type.
     * Verifies that unwrap can return itself when the type matches.
     */
    @Test
    void testUnwrap_withSameType() {
        when(smbTree.unwrap(SmbTree.class)).thenReturn(smbTree);

        SmbTree result = smbTree.unwrap(SmbTree.class);

        assertNotNull(result, "Unwrapped tree should not be null");

        when(tree.unwrap(SmbTreeInternal.class)).thenReturn(tree);

        // Act
        SmbTree unwrapped = tree.unwrap(SmbTreeInternal.class);
        tree.close();

        // Assert
        assertNotNull(unwrapped);
        verify(tree).unwrap(SmbTreeInternal.class);
        verify(tree).close();
    }

 */
package jcifs.smb;

import jcifs.CIFSContext;
import jcifs.CIFSException;
import jcifs.SmbTree;
import jcifs.internal.CommonServerMessageBlockResponse;
import jcifs.internal.Request;

/**
 * Internal SMB tree connection interface providing extended tree management capabilities.
 * Defines methods for internal tree operations and resource management.
 *
 * @author mbechler
 *

    }

    // Happy path: getSmbTree returns a tree and receives correct args
    @Test
    @DisplayName("getSmbTree returns tree and captures arguments")
    void getSmbTree_happy() {
        when(session.getSmbTree(anyString(), anyString())).thenReturn(tree);

        String share = "share";
        String svc = "A:"; // service string example (e.g., DISK, IPC, or other)
        SmbTree result = session.getSmbTree(share, svc);

        smbSession = mock(SmbSession.class);
        smbSession.transport = smbTransport;

        // Mock the tree layer
        smbTree = mock(SmbTree.class);
        smbTree.session = smbSession;
        smbTree.tree_num = 1;

        // Mock the SmbFile
        smbFile = mock(SmbFile.class);
        smbFile.tree = smbTree;
        smbFile.fid = 1;
        smbFile.tree_num = 1;

        when(smbFile.isFile()).thenReturn(true);