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");

        // Mock negotiate response for SMB1
        SmbComNegotiateResponse nego = mock(SmbComNegotiateResponse.class);
        ServerData serverData = new ServerData();
        when(nego.getServerData()).thenReturn(serverData);
        when(transport.getNegotiateResponse()).thenReturn(nego);

        // Mock config methods needed for SMB1
        when(config.getPid()).thenReturn(1234);

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

    /**
     * {@inheritDoc}

 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package jcifs.internal.smb2.nego;

import jcifs.Configuration;
import jcifs.internal.SMBProtocolDecodingException;
import jcifs.internal.util.SMBUtil;

/**
 * SMB2 Pre-authentication Integrity Negotiate Context.
 *

 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package jcifs.internal.smb2.nego;

import jcifs.Configuration;
import jcifs.internal.SMBProtocolDecodingException;
import jcifs.internal.util.SMBUtil;

/**
 * SMB2 Encryption Negotiate Context.
 *

    while (true) {
      if (RANDOM_SOURCE.nextBoolean()) {
        return randomNonNegativeBigInteger(numBits);
      }
      BigInteger neg = randomNonNegativeBigInteger(numBits).negate();
      if (neg.signum() != 0) {
        return neg;
      }
    }
  }

  /**
   * Generates a number in [0, 2^numBits) with an exponential distribution. The floor of the log2 of

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

 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package jcifs.internal.smb2.nego;

import jcifs.Encodable;

/**
 * SMB2 Negotiate Context request data structure.
 *
 * This class represents negotiate contexts used in SMB 3.1.1
 * for capability negotiation and security enhancements.
 *