@Mock
    private Credentials credentials;
    @Mock
    private CredentialsInternal credentialsInternal;
    @Mock
    private SmbTransportImpl transport;

    private SmbSessionImpl newSession() {
        return new SmbSessionImpl(cifsContext, "server.example", "EXAMPLE", transport);
    }

    @BeforeEach
    void setup() {
        // Base context configuration - always needed

     * Test that concurrent tree operations are thread-safe with CopyOnWriteArrayList.
     */
    @Test
    public void testConcurrentTreeOperationsThreadSafe() throws Exception {
        // Given
        SmbSessionImpl session = new SmbSessionImpl(mockContext, "testhost", "testdomain", mockTransport);

        int threadCount = 10;
        int opsPerThread = 100;
        ExecutorService executor = Executors.newFixedThreadPool(threadCount);

    public int getSendBufferSize() throws SmbException {
        try (SmbSessionImpl session = this.treeConnection.getSession(); SmbTransportImpl transport = session.getTransport()) {
            return transport.getNegotiateResponse().getSendBufferSize();
        }
    }

    @Override
    public int getReceiveBufferSize() throws SmbException {

     * 2 - connected
     * 3 - disconnecting
     */
    private final AtomicInteger connectionState = new AtomicInteger();

    private final String share;
    private final String service0;
    private final SmbSessionImpl session;

    private volatile int tid = -1;
    private volatile String service = "?????";
    private volatile boolean inDfs, inDomainDfs;
    private volatile long treeNum; // used by SmbFile.isOpen

            log.warn("Tree connection was not properly released " + this);
        }
    }

    synchronized void disconnect(final boolean inError) {
        try (SmbSessionImpl session = getSession()) {
            if (session == null) {
                return;
            }
            try (SmbTransportImpl transport = session.getTransport()) {
                synchronized (transport) {

        // Act: create new session (happy path)
        SmbSessionImpl s1 = transport.getSmbSession(ctx);
        assertNotNull(s1);
        assertEquals(1, transport.getNumSessions());
        s1.close();

        // Act: request again with same context -> reuse existing
        SmbSessionImpl s2 = transport.getSmbSession(ctx);
        assertNotNull(s2);

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

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

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

     *
     * @see jcifs.smb.SmbPipeHandleInternal#getSessionKey()
     */
    @Override
    public byte[] getSessionKey() throws CIFSException {
        try (SmbTreeHandleImpl th = ensureTreeConnected(); SmbSessionImpl sess = th.getSession()) {
            return sess.getSessionKey();
        }
    }

    /**
     * {@inheritDoc}
     *
     * @see jcifs.SmbPipeHandle#isStale()
     */
    @Override