assertEquals(expectedFlags, message.getFlags());
            assertArrayEquals(TEST_CHALLENGE, message.getChallenge());
            assertEquals(TEST_TARGET, message.getTarget());
            assertNotNull(message.getTargetInformation());
        }

        @Test
        @DisplayName("Constructor with custom flags should use provided flags")
        void testConstructor_CIFSContext_Flags_Challenge_Target() {

    int FLAGS_OPLOCK_REQUESTED_OR_GRANTED = 0x20;
    /** Notify of modify action flag */
    int FLAGS_NOTIFY_OF_MODIFY_ACTION = 0x40;
    /** Response flag */
    int FLAGS_RESPONSE = 0x80;

    /** No flags2 set */
    int FLAGS2_NONE = 0x0000;
    /** Long filenames supported flag */
    int FLAGS2_LONG_FILENAMES = 0x0001;
    /** Extended attributes supported flag */

     * @param flags
     *            The flags to apply to this message.
     * @param challenge
     *            The challenge from the domain controller/server.
     * @param target
     *            The authentication target.
     */
    public Type2Message(final CIFSContext tc, final int flags, final byte[] challenge, final String target) {
        setFlags(flags);
        setChallenge(challenge);

        // These tests should be removed or replaced with actual constants
    }

    /**
     * Test individual FLAGS2 constants.
     */
    @Test
    @DisplayName("FLAGS2 individual constants are correct")
    void flags2ConstantsTest() {
        // Test individual flag values
        assertEquals(0x0001, SmbConstants.FLAGS2_LONG_FILENAMES);
        assertEquals(0x0002, SmbConstants.FLAGS2_EXTENDED_ATTRIBUTES);

     */
    public ServerData() {
        // Default constructor
    }

    /**
     * Server flags from the SMB header.
     */
    public byte sflags;
    /**
     * Server flags2 field from the SMB header.
     */
    public int sflags2;
    /**
     * Maximum number of outstanding multiplex requests.
     */
    public int smaxMpxCount;
    /**

        }

        @Test
        @DisplayName("Should define getFlags method returning int")
        void shouldDefineGetFlags() {
            ACE ace = mock(ACE.class);
            when(ace.getFlags()).thenReturn(ACE.FLAGS_INHERITED);

            int flags = ace.getFlags();
            assertEquals(ACE.FLAGS_INHERITED, flags);
            verify(ace).getFlags();
        }

        @Test

    }

    /**
     * Gets the registration flags.
     *
     * @return the registration flags
     */
    public int getFlags() {
        return flags;
    }

    /**
     * Sets the registration flags.
     *
     * @param flags the registration flags
     */
    public void setFlags(int flags) {
        this.flags = flags;
    }

    /**
     * Gets the context handle.

     * @return the serverType
     */
    public final int getServerType() {
        return this.serverType;
    }

    /**
     * Gets the referral flags.
     *
     * @return the rflags
     */
    public final int getRFlags() {
        return this.rflags;
    }

    /**
     * Gets the proximity value indicating the distance to the target.
     *
     * @return the proximity
     */

    /**
     * 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.
     */

        }

        @Test
        @DisplayName("Should handle all flags set")
        void testAllFlagsSet() {
            int allFlags = 0xFFFFFFFF;
            testMessage.addFlags(allFlags);
            assertEquals(allFlags, testMessage.getFlags());

            // Should still be able to clear specific flags
            testMessage.clearFlags(ServerMessageBlock2.SMB2_FLAGS_SIGNED);