}

    private static MavenProject createProject(String id) {
        MavenProject result = new MavenProject();
        result.setGroupId("org.apache");
        result.setArtifactId(id);
        result.setVersion("1.2");
        return result;
    }

        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

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

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;

        pluginDescriptor.setSource(source);
        pluginDescriptor.setGroupId(extractGroupId(c));
        pluginDescriptor.setArtifactId(extractArtifactId(c));
        pluginDescriptor.setVersion(extractVersion(c));
        pluginDescriptor.setGoalPrefix(extractGoalPrefix(c));

        pluginDescriptor.setName(extractName(c));
        pluginDescriptor.setDescription(extractDescription(c));

                versionRequest.setTrace(trace);
                VersionResult versionResult = versionResolver.resolveVersion(session, versionRequest);

                a = a.setVersion(versionResult.getVersion());

                versionRequest =
                        new VersionRequest(pomArtifact, request.getRepositories(), request.getRequestContext());
                versionRequest.setTrace(trace);

            }
            if (timeoutInMillis != -1) {
                builder.setTimeout(new TimeValue(timeoutInMillis));
            }
            if (version != null) {
                builder.setVersion(version);
            }
            if (seqNoAndPrimaryTerm != null) {
                builder.seqNoAndPrimaryTerm(seqNoAndPrimaryTerm);
            }
            if (terminateAfter > 0) {

            }
            if (timeoutInMillis != -1) {
                builder.setTimeout(new TimeValue(timeoutInMillis));
            }
            if (version != null) {
                builder.setVersion(version);
            }
            if (seqNoAndPrimaryTerm != null) {
                builder.seqNoAndPrimaryTerm(seqNoAndPrimaryTerm);
            }
            if (terminateAfter > 0) {

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