public void testServiceCleanup() throws CIFSException {
        String sessionId1 = "cleanup-test-1";
        String sessionId2 = "cleanup-test-2";
        byte[] salt = preauthService.generatePreauthSalt();

        preauthService.initializeSession(sessionId1, salt, PreauthIntegrityService.HASH_ALGO_SHA512);
        preauthService.initializeSession(sessionId2, salt, PreauthIntegrityService.HASH_ALGO_SHA512);

        // Generate unique session ID for key management
        this.sessionId = String.format("smb-enc-%d-%d", System.currentTimeMillis(), secureRandom.nextLong());

        if (keyManager != null) {
            // Store keys securely
            String encKeyId = sessionId + "-enc";
            String decKeyId = sessionId + "-dec";
            keyManager.storeSessionKey(encKeyId, encryptionKey, "AES");

    }

    /**
     * @return this session's UID
     */
    public int getUid() {
        return this.uid;
    }

    /**
     * @return this session's expiration time
     */
    public Long getExpiration() {
        return this.expiration > 0 ? this.expiration : null;
    }

    /**
     * @return this session's credentials
     */

     */
    public static class ChannelGroup {
        private final String sessionId;
        private final List<ChannelInfo> channels = new CopyOnWriteArrayList<>();
        private final AtomicInteger roundRobinIndex = new AtomicInteger(0);
        private volatile ChannelInfo primaryChannel;

        public ChannelGroup(String sessionId) {
            this.sessionId = sessionId;
        }

        public String getSessionId() {

        sessionContexts.put(sessionId, context);

        log.debug("Initialized preauth integrity context for session {}", sessionId);
        return context;
    }

    /**
     * Updates the preauth integrity hash with a new message.
     *
     * @param sessionId the session identifier
     * @param messageData the message data to include in hash

            this.password = null;
        }
        // Also remove from secure key manager if we have a session
        if (this.sessionId != null) {
            keyManager.removeSessionKey(this.sessionId);
            this.sessionId = null;
        }
    }

    /**
     * Check if the authentication has expired based on TTL
     *
     * @return true if expired, false otherwise

     *
     * @return the sessionId
     */
    public long getSessionId() {
        return this.sessionId;
    }

    /**
     * @param sessionId
     *            the sessionId to set
     */
    @Override
    public final void setSessionId(final long sessionId) {
        this.sessionId = sessionId;
        if (this.next != null) {

{* ../../docs_src/app_testing/tutorial002.py hl[27:31] *}

/// note

For more details, check Starlette's documentation for <a href="https://www.starlette.io/testclient/#testing-websocket-sessions" class="external-link" target="_blank">testing WebSockets</a>.

        // Simulate session already in use by adding a session
        Field sessionsField = SmbTransportImpl.class.getDeclaredField("sessions");
        sessionsField.setAccessible(true);
        @SuppressWarnings("unchecked")
        List<Object> sessions = (List<Object>) sessionsField.get(first);
        sessions.add(new Object()); // Add one session to reach the limit

        // When: Request another connection

                testEncryptionKey, testEncryptionKey); // Use same key for test

        byte[] plaintext = "Hello, SMB3 Encryption!".getBytes();
        long sessionId = 0x123456789ABCDEF0L;

        // When - Encrypt
        byte[] encrypted = context.encryptMessage(plaintext, sessionId);

        // Then - Verify encrypted message structure
        assertNotNull(encrypted, "Encrypted message should not be null");