protected void doConnect() throws IOException {
    // ... existing connection logic ...
    
    if (context.getConfig().isUseLeases() && tree.getSession().supports(SMB3_0)) {
        // Request lease when opening file
        leaseManager = tree.getSession().getLeaseManager();
        int requestedState = isDirectory() ? 
            Smb2LeaseState.SMB2_LEASE_READ_HANDLE :
            Smb2LeaseState.SMB2_LEASE_FULL;

        verify(c, never()).connectHost(any(), anyString());
    }

    @Test
    @DisplayName("getSession returns null when no tree is present")
    void getSession_nullWhenNoTree() {
        SmbTreeConnection c = newConn();
        assertNull(c.getSession());
    }

    public HtmlResponse search(final SearchForm form) {
        if (viewHelper.isUseSession()) {
            LaRequestUtil.getOptionalRequest().ifPresent(request -> {
                final HttpSession session = request.getSession(false);
                if (session != null && form.num != null) {
                    session.setAttribute(Constants.RESULTS_PER_PAGE, form.num);
                }
            });
        }

                return;
            }
            final HttpSession ssn = request.getSession();
            ssn.setAttribute("NtlmHttpAuth", ntlm);
            ssn.setAttribute("ntlmdomain", ntlm.getUserDomain());
            ssn.setAttribute("ntlmuser", ntlm.getUsername());
        } else {
            final HttpSession ssn = request.getSession(false);
            if (ssn == null || ssn.getAttribute("NtlmHttpAuth") == null) {

        this.delegate = delegate;
        this.eventType = eventType;
    }

    @Override
    public EventType getType() {
        return eventType;
    }

    @Override
    public Session getSession() {
        return session;
    }

    @Override
    public Optional<Project> getProject() {
        return Optional.ofNullable(session.getProject(delegate.getProject()));
    }

    @Override

        ArtifactInstallerRequest request =
                ArtifactInstallerRequest.builder().session(session).trace(trace).build();

        assertEquals(trace, request.getTrace());
        assertEquals(session, request.getSession());
    }

    @Test
    void testDependencyResolverRequestEquality() {
        Session session = mock(Session.class);

    public Iterator<ProjectSegment> iterator() {
        return items.iterator();
    }

    public void closeAll() {
        for (ProjectSegment item : items) {
            MavenSession sessionForThisModule = item.getSession();
            sessionForThisModule.setCurrentProject(null);
        }
    }

    public int size() {
        return items.size();
    }

    public ProjectSegment get(int index) {

        SmbSessionImpl session = mock(SmbSessionImpl.class);
        when(tree.getSession()).thenReturn(session);
        when(session.getSessionKey()).thenReturn(new byte[] { 1, 2, 3 });

        byte[] key = target.getSessionKey();
        assertArrayEquals(new byte[] { 1, 2, 3 }, key);
        verify(tree, times(1)).getSession();
        verify(session, times(1)).getSessionKey();
    }

    @Test

@Immutable
public interface Request<S extends ProtoSession> {

    /**
     * Returns the session associated with this request.
     *
     * @return the session instance, never {@code null}
     */
    @Nonnull
    S getSession();

    /**
     * Returns the trace information associated with this request, if any.
     * The trace provides context about the request's position in the operation

        InstallRequest request = new InstallRequest();

        request.setTrace(RequestTrace.newChild(null, legacySupport.getSession().getCurrentProject()));

        org.eclipse.aether.artifact.Artifact mainArtifact = RepositoryUtils.toArtifact(artifact);
        mainArtifact = mainArtifact.setFile(source);
        request.addArtifact(mainArtifact);