}

        @Test
        @DisplayName("Should handle session key retrieval")
        void testSessionKeyRetrieval() throws CIFSException {
            // When: Getting session key
            byte[] sessionKey = handle.getSessionKey();

            // Then: Should return expected session key
            assertNotNull(sessionKey);

        this.remaining = remaining;
        this.b = b;
        this.off = off;
        dataLength = len;
        digest = null; /* otherwise recycled commands
                        * like writeandx will choke if session
                        * closes in between */
    }

    @Override
    int getBatchLimit(final byte command) {
        if (command == SMB_COM_READ_ANDX) {
            return READ_ANDX_BATCH_LIMIT;
        }

        this.openFlags &= ~(SmbFile.O_CREAT | SmbFile.O_TRUNC); /* in case close and reopen */
        writeSize = file.tree.session.transport.snd_buf_size - 70;

        useNTSmbs = file.tree.session.transport.hasCapability(SmbConstants.CAP_NT_SMBS);
        if (useNTSmbs) {
            reqx = new SmbComWriteAndX();
            rspx = new SmbComWriteAndXResponse();
        } else {

                final HttpSession session = req.getSession(false);
                if (session != null) {
                    session.setAttribute(SCREEN_WIDTH, width);
                }
            } else {
                final HttpSession session = req.getSession(false);
                if (session != null) {
                    final Integer width = (Integer) session.getAttribute(SCREEN_WIDTH);

 */
public class SmbComWriteResponse extends ServerMessageBlock {

    private long count;

    /**
     * Constructs a write response for SMB1 protocol.
     *
     * @param config the configuration for this SMB session
     */
    public SmbComWriteResponse(final Configuration config) {
        super(config);
    }

    /**
     * Gets the number of bytes written.
     *
     * @return the count of bytes written
     */

        this.b = b;
        this.off = off;
        this.count = len;
        this.digest = null; /*
                             * otherwise recycled commands
                             * like writeandx will choke if session
                             * closes in between
                             */
    }

    @Override
    protected int writeParameterWordsWireFormat(final byte[] dst, int dstIndex) {

     */
    @Resource
    protected HttpServletRequest request;

    /**
     * Returns an empty OptionalThing for login manager since API actions
     * use token-based authentication instead of traditional session-based login.
     *
     * @return empty OptionalThing indicating no login manager is used
     */
    @Override
    protected OptionalThing<LoginManager> myLoginManager() {
        return OptionalThing.empty();
    }

package jcifs.internal.smb2.session;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoMoreInteractions;
import static org.mockito.Mockito.when;

                } catch (final IOException ioe) {}
                throw new SmbException("EOF during NetBIOS session request");
            }
            switch (sbuf[0] & 0xFF) {
            case SessionServicePacket.POSITIVE_SESSION_RESPONSE:
                if (LogStream.level >= 4) {
                    log.println("session established ok with " + address);
                }
                return;

    if (msg instanceof Smb2LeaseBreakNotification) {
        Smb2LeaseBreakNotification breakNotif = (Smb2LeaseBreakNotification) msg;
        
        // Get lease manager from session
        LeaseManager leaseManager = session.getLeaseManager();
        
        // Handle the lease break
        leaseManager.handleLeaseBreak(
            breakNotif.getLeaseKey(),
            breakNotif.getNewLeaseState()