private SmbNegotiation negotiation;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        testRequestBuffer = new byte[] { 0x01, 0x02, 0x03, 0x04 };
        testResponseBuffer = new byte[] { 0x05, 0x06, 0x07, 0x08 };
        negotiation = new SmbNegotiation(mockRequest, mockResponse, testRequestBuffer, testResponseBuffer);
    }

    @Test

    private byte[] testMacSigningKey;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        testMacSigningKey = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };
    }

    @Test
    @DisplayName("Test constructor with MAC signing key and bypass flag")
    void testConstructorWithBypass() {

        // Verify results is now set (even if empty)
        assertNotNull(response.getResults());
    }

    @ParameterizedTest
    @CsvSource({ "0x01, FILE_DIRECTORY_INFO", "0x02, FILE_FULL_DIRECTORY_INFO", "0x03, FILE_BOTH_DIRECTORY_INFO" })
    @DisplayName("Test constructor with different info class constants")
    void testConstructorWithDifferentInfoClasses(byte infoClass, String description) {

        // Bit 2: active (1)
        // Bit 1: permanent (1)

        // MAC address
        byte[] testMac = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
        System.arraycopy(testMac, 0, src, srcIndex + 19, 6);

        // Statistics
        byte[] stats = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00 };
        System.arraycopy(stats, 0, src, srcIndex + 25, 6);

        response.readRDataWireFormat(src, srcIndex);

            context = new PreauthIntegrityNegotiateContext();
            byte[] buffer = new byte[] { 0x02, 0x00, // 2 hash algos
                    0x04, 0x00, // 4 bytes salt
                    0x01, 0x00, // SHA512
                    0x02, 0x00, // Another hash algo
                    0x0A, 0x0B, 0x0C, 0x0D // Salt
            };

            // Act

        int bufferIndex = 0;

        // Set up test values
        int namedPipeState = 0x03; // NP_NOWAIT | NP_READMODE_MESSAGE
        int readDataAvailable = 100;
        int numberOfMessages = 2;
        int messageLength = 50;
        byte[] testData = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08 };

        // Write values to buffer
        SMBUtil.writeInt4(namedPipeState, buffer, bufferIndex);

    private Smb2SessionSetupRequest request;

    // Test constants
    private static final int TEST_SECURITY_MODE = 0x01;
    private static final int TEST_CAPABILITIES = 0x00000001;
    private static final long TEST_PREVIOUS_SESSION_ID = 0x1234567890ABCDEFL;
    private static final byte[] TEST_TOKEN = { 0x01, 0x02, 0x03, 0x04, 0x05 };

    @BeforeEach
    void setUp() {
        mockConfig = mock(Configuration.class);

        testBuffer[8] = 0x01; // Revision
        testBuffer[9] = 0x01; // Sub-authority count
        testBuffer[10] = 0x00; // Identifier authority
        testBuffer[11] = 0x00;
        testBuffer[12] = 0x00;
        testBuffer[13] = 0x00;
        testBuffer[14] = 0x00;
        testBuffer[15] = 0x01;
        testBuffer[16] = 0x00; // Sub-authority
        testBuffer[17] = 0x00;
        testBuffer[18] = 0x00;

        void testRpcPtResponse() {
            assertEquals(0x02, DcerpcConstants.RPC_PT_RESPONSE, "RPC_PT_RESPONSE should be 0x02");
        }

        @Test
        @DisplayName("RPC_PT_FAULT should have correct value")
        void testRpcPtFault() {
            assertEquals(0x03, DcerpcConstants.RPC_PT_FAULT, "RPC_PT_FAULT should be 0x03");
        }

        @Test

        originalPairs.add(new AvFlags(0x12345678));
        originalPairs.add(new AvTimestamp(new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08 }));
        originalPairs.add(new AvTargetName("TESTNAME".getBytes()));
        originalPairs.add(new AvSingleHost(new byte[] { 0x11, 0x22, 0x33, 0x44 }));
        originalPairs.add(new AvChannelBindings(new byte[] { 0x55, 0x66 }));