import jcifs.internal.smb2.nego.Smb2NegotiateResponse;
import jcifs.internal.smb2.tree.Smb2TreeConnectResponse;

class SmbTreeImplTest {

    @Mock
    private SmbSessionImpl session;

    @Mock
    private Configuration config;

    @Mock
    private CIFSContext context;

    @Mock
    private SmbTransportImpl transport;

    @BeforeEach

    void testGetSessionKey() throws CIFSException {
        when(pipe.ensureTreeConnected()).thenReturn(tree);
        when(tree.acquire()).thenReturn(tree);
        SmbSessionImpl session = mock(SmbSessionImpl.class);
        when(tree.getSession()).thenReturn(session);
        when(session.getSessionKey()).thenReturn(new byte[] { 1, 2, 3 });

        byte[] key = target.getSessionKey();

@MockitoSettings(strictness = Strictness.LENIENT)
class SmbTreeHandleImplTest {

    @Mock
    SmbResourceLocatorImpl resourceLoc;

    @Mock
    SmbTreeConnection treeConnection;

    @Mock
    SmbSessionImpl session;

    @Mock
    SmbTransportImpl transport;

    @Mock
    Configuration config;

    private SmbTreeHandleImpl handle;

    @BeforeEach
    void setUp() {

    void connect_whenAlreadyConnected_returnsHandle() throws Exception {
        SmbTreeConnection c = spy(newConn());
        SmbTreeImpl tree = mock(SmbTreeImpl.class);
        SmbSessionImpl session = mock(SmbSessionImpl.class);
        SmbTransportImpl transport = mock(SmbTransportImpl.class);

        when(tree.isConnected()).thenReturn(true);
        when(tree.getSession()).thenReturn(session);

            }
        }

        final SmbTreeConnection treeConn = SmbTreeConnection.create(tc);
        try (SmbTreeHandleImpl th = treeConn.connectHost(locator, locator.getServerWithDfs());
                SmbSessionImpl session = th.getSession();
                SmbTransportImpl transport = session.getTransport()) {
            try {
                entries = doMsrpcShareEnum(tc, locator, transport.getRemoteAddress());

    @Mock
    private SmbFileHandleImpl fh;

    @Mock
    private Smb2IoctlResponse ioctlResp;

    @Mock
    private Configuration config;

    @Mock
    private SmbSessionImpl session;

    @BeforeEach
    void setUp() {
        lenient().when(tree.getConfig()).thenReturn(config);
    }

    private SmbPipeHandleImpl newHandleWithBasicStubs(int pipeType, String unc) {

        SmbTransportImpl mockTransport = Mockito.mock(SmbTransportImpl.class);
        Mockito.when(mockTransport.acquire()).thenReturn(mockTransport);

        SmbSessionImpl session = new SmbSessionImpl(mockContext, "testhost", "testdomain", mockTransport);

        int threadCount = 10;
        int operationsPerThread = 200;
        ExecutorService executor = Executors.newFixedThreadPool(threadCount);

                req.setShareAccess(sharing);
                req.setFileAttributes(attrs);

                // Add SMB3 features support with error handling for compatibility
                SmbSessionImpl session = h.getSession();

                // Enable lease support if available, with fallback for compatibility
                boolean leasesAdded = false;
                if (config.isUseLeases() && h.isSMB3()) {