queryPolicy = new MsrpcQueryInformationPolicy(mockPolicyHandle, level, mockNdrObject);

        // Assert
        assertEquals(expectedFlags, queryPolicy.getFlags());
        // Binary check for individual flags
        assertEquals(0x03, queryPolicy.getFlags()); // 0x01 | 0x02 = 0x03
    }

    @Test
    void getOpnum_shouldReturnCorrectValue() {
        // Arrange
        short level = 1;

    }

    @Test
    @DisplayName("Test context with flags")
    public void testContextWithFlags() {
        CompressionNegotiateContext contextWithFlags =
                new CompressionNegotiateContext(config, new int[] { CompressionNegotiateContext.COMPRESSION_LZ77 }, 0x12345678);

        assertEquals(0x12345678, contextWithFlags.getFlags());
    }

    @Test
    @DisplayName("Test toString representation")

    }

    @Nested
    @DisplayName("Flag Management Tests")
    class FlagManagementTests {

        @Test
        @DisplayName("isFlagSet should correctly identify set flags")
        void testIsFlagSet() {
            // Test when flag is not set
            message.flags = 0;
            assertFalse(message.isFlagSet(DcerpcConstants.RPC_C_PF_BROADCAST));

            // Test when flag is set

    // RPC_C_PF_ flags (packet flags)
    /**
     * Broadcast packet flag
     */
    int RPC_C_PF_BROADCAST = 0x01;
    /**
     * No fragmentation packet flag
     */
    int RPC_C_PF_NO_FRAGMENT = 0x02;
    /**
     * Maybe semantics packet flag
     */
    int RPC_C_PF_MAYBE = 0x04;
    /**
     * Idempotent operation packet flag
     */
    int RPC_C_PF_IDEMPOTENT = 0x08;

        }

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

     */
    @Test
    void testConstructor_CreateAndOpen() {
        int flags = SmbFile.O_CREAT;
        smbComOpenAndX = new SmbComOpenAndX(fileName, access, flags, andx);
        assertEquals(0x11, smbComOpenAndX.openFunction); // OPEN_FN_CREATE | OPEN_FN_OPEN
    }

    /**
     * Test constructor with no special flags.
     */
    @Test
    void testConstructor_OpenOnly() {
        int flags = 0;

     *
     * @param config the configuration
     * @param compressionAlgorithms the supported compression algorithms
     * @param flags compression flags (reserved, should be 0)
     */
    public CompressionNegotiateContext(final Configuration config, final int[] compressionAlgorithms, final int flags) {

    }

    /**
     * Get the flags
     * @return the flags
     */
    public int getFlags() {
        return flags;
    }

    /**
     * Check if this is a persistent handle request
     * @return true if persistent
     */
    public boolean isPersistent() {
        return (flags & Smb2HandleCapabilities.SMB2_DHANDLE_FLAG_PERSISTENT) != 0;
    }

    }

    @Test
    @DisplayName("DCE/RPC flags should be set correctly")
    void testDcerpcFlags() {
        // Test individual flag checking
        assertTrue(dfsRootEnum.isFlagSet(DcerpcConstants.DCERPC_FIRST_FRAG));
        assertTrue(dfsRootEnum.isFlagSet(DcerpcConstants.DCERPC_LAST_FRAG));

        // Combined flags should equal FIRST_FRAG | LAST_FRAG

        String username = "testuser";
        String workstation = "TESTWS";
        int flags = NtlmFlags.NTLMSSP_NEGOTIATE_UNICODE | NtlmFlags.NTLMSSP_NEGOTIATE_NTLM;

        // When
        Type3Message type3 = new Type3Message(createMockContext(), type2, null, password, domain, username, workstation, flags);

        // Then
        assertNotNull(type3);