SmbNegotiationResponse nego = mock(SmbNegotiationResponse.class);
        when(transport.getNegotiateResponse()).thenReturn(nego);
        when(nego.getSendBufferSize()).thenReturn(1111);
        when(nego.getReceiveBufferSize()).thenReturn(2222);
        when(nego.getTransactionBufferSize()).thenReturn(3333);
        when(nego.isSigningNegotiated()).thenReturn(true);

        // Arrange
        SmbNegotiationResponse nego = mock(SmbNegotiationResponse.class);
        setField(transport, "negotiated", nego);
        when(nego.haveCapabilitiy(SmbConstants.CAP_DFS)).thenReturn(true);

        // Act & Assert
        assertTrue(transport.hasCapability(SmbConstants.CAP_DFS));
        verify(nego, times(1)).haveCapabilitiy(SmbConstants.CAP_DFS);
    }

    @Test

        }
        final ValidateNegotiateInfoResponse out = resp.getOutputData(ValidateNegotiateInfoResponse.class);

        if (nego.getSecurityMode() != out.getSecurityMode() || nego.getCapabilities() != out.getCapabilities()
                || nego.getDialectRevision() != out.getDialect() || !Arrays.equals(nego.getServerGuid(), out.getServerGuid())) {
            log.debug("Secure negotiation failure");

            final SmbNegotiationResponse nego = transport.getNegotiateResponse();
            if (nego instanceof SmbComNegotiateResponse) {
                return ((SmbComNegotiateResponse) nego).getServerData().serverTimeZone * 1000 * 60L;
            }
            return 0;
        }
    }

    /**
     * {@inheritDoc}

        jcifs.internal.SmbNegotiationResponse nego = mock(jcifs.internal.SmbNegotiationResponse.class);
        when(transport.getNegotiateResponse()).thenReturn(nego);

        when(nego.isSigningNegotiated()).thenReturn(true);
        assertTrue(session.isSignatureSetupRequired());
        when(nego.isSigningNegotiated()).thenReturn(false);
        assertFalse(session.isSignatureSetupRequired());

import org.mockito.junit.jupiter.MockitoSettings;
import org.mockito.quality.Strictness;

import jcifs.Address;
import jcifs.CIFSContext;
import jcifs.Configuration;
import jcifs.DialectVersion;
import jcifs.internal.smb2.nego.Smb2NegotiateResponse;

/**
 * Test Pre-Authentication Integrity improvements
 */
@ExtendWith(MockitoExtension.class)
@MockitoSettings(strictness = Strictness.LENIENT)
public class PreauthIntegrityTest {

package jcifs.internal.smb2.nego;

import static org.junit.jupiter.api.Assertions.assertArrayEquals;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertTrue;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Nested;

import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import jcifs.CIFSException;
import jcifs.internal.smb2.nego.PreauthIntegrityNegotiateContext;

/**
 * Enhanced Pre-Authentication Integrity Service for SMB 3.1.1.
 *
 * Provides comprehensive pre-authentication integrity protection against

package jcifs.internal.smb2.nego;

import static org.junit.jupiter.api.Assertions.assertArrayEquals;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertFalse;
import static org.junit.jupiter.api.Assertions.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertNull;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;

package jcifs.internal.smb2.nego;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.junit.jupiter.api.Assertions.assertTrue;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;

import jcifs.Configuration;
import jcifs.internal.SMBProtocolDecodingException;
import jcifs.internal.smb2.Smb2Constants;