@Test
    @DisplayName("Test with different buffer sizes")
    void testWithDifferentBufferSizes() {
        // Test with buffers of different sizes
        byte[] smallBuffer = new byte[] { 0x01 };
        byte[] mediumBuffer = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05 };

        SmbNegotiation negotiationDifferentSizes = new SmbNegotiation(mockRequest, mockResponse, smallBuffer, mediumBuffer);

    void constructorSetsCorrectCommand() {
        SmbComLogoffAndX msg = new SmbComLogoffAndX(null);
        assertNotNull(msg, "Message should not be null");
        // SMB_COM_LOGOFF_ANDX = 0x74
        assertEquals((byte) 0x74, msg.command, "Command should be SMB_COM_LOGOFF_ANDX");
    }

    @Test
    @DisplayName("constructor with mock andx properly sets andx field")
    void constructorWithMockAndx() {

    class ToStringTests {

        @ParameterizedTest
        @DisplayName("Should format command name correctly")
        @ValueSource(shorts = { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10, 0x11,
                0x12 })
        void testToStringCommandNames(short command) {
            testMessage.setCommand(command);
            String result = testMessage.toString();

        when(this.mechContext.getFlags()).thenReturn(0x1234);
        when(this.mechContext.initSecContext(any(byte[].class), eq(0), eq(0))).thenReturn(new byte[] { 0x01, 0x02 });

        // Act: len==0 triggers initial token construction
        byte[] out = ctx.initSecContext(null, 0, 0);

        // Assert: returns a SPNEGO token (opaque here but non-null/non-empty)

    /**
     * No flags set in SMB header.
     */
    int FLAGS_NONE = 0x00;
    /**
     * Obsolete lock and read/write and unlock flag.
     */
    int FLAGS_LOCK_AND_READ_WRITE_AND_UNLOCK = 0x01;
    /**
     * Receive buffer has been posted flag.
     */
    int FLAGS_RECEIVE_BUFFER_POSTED = 0x02;
    /**
     * Path names are case-insensitive flag.
     */
    int FLAGS_PATH_NAMES_CASELESS = 0x08;

        }

        @ParameterizedTest
        @DisplayName("Should read different oplock levels correctly")
        @ValueSource(bytes = { 0x00, 0x01, 0x02, 0x08, (byte) 0xFF })
        void testReadDifferentOplockLevels(byte oplockLevel) throws Exception {
            byte[] buffer = createValidOplockBreakBuffer(oplockLevel, createTestData(16));

    void read_oneByteVariants() throws Exception {
        SmbRandomAccessFile raf = spy(newInstance("r", false, false, false));
        // For each 1-byte read, fill provided buffer with 0xFF/0x01
        doAnswer(inv -> {
            byte[] b = inv.getArgument(0);
            int off = inv.getArgument(1);
            b[off] = (byte) 0xFF;
            return 1;

    }

    @Test
    @DisplayName("Should define SMB flags constants")
    void testSmbFlags() {
        assertEquals(0x00, SmbConstants.FLAGS_NONE);
        assertEquals(0x01, SmbConstants.FLAGS_LOCK_AND_READ_WRITE_AND_UNLOCK);
        assertEquals(0x02, SmbConstants.FLAGS_RECEIVE_BUFFER_POSTED);
        assertEquals(0x08, SmbConstants.FLAGS_PATH_NAMES_CASELESS);

                // Wrong TOK_ID
                Arguments.of("wrong TOK_ID", new byte[] { 0x00, 0x02, 0x00, 0x01 }, IllegalArgumentException.class),

                // OID length claims more than available
                Arguments.of("oid length exceeds buffer", new byte[] { 0x04, 0x01, 0x00, 0x10, 0x06 }, IllegalArgumentException.class),

            context = new PreauthIntegrityNegotiateContext();
            byte[] buffer = new byte[] { (byte) 0xFF, (byte) 0xFF, (byte) 0xFF, // Padding
                    0x01, 0x00, // 1 hash algo
                    0x02, 0x00, // 2 bytes salt
                    0x01, 0x00, // SHA512
                    (byte) 0xAA, (byte) 0xBB // Salt
            };

            // Act