// Act
        lookupSids = new MsrpcLookupSids(mockPolicyHandle, testSids);

        // Assert
        assertTrue((lookupSids.getFlags() & DcerpcConstants.DCERPC_FIRST_FRAG) != 0);
        assertTrue((lookupSids.getFlags() & DcerpcConstants.DCERPC_LAST_FRAG) != 0);
    }

    @Test
    void constructor_shouldPassCorrectCountToSuperclass() throws Exception {
        // Arrange

    }

    @Test
    @DisplayName("Test getFlags returns correct value")
    void testGetFlags() {
        ace.flags = 0x00;
        assertEquals(0x00, ace.getFlags());

        ace.flags = 0xFF;
        assertEquals(0xFF, ace.getFlags());

        ace.flags = ACE.FLAGS_OBJECT_INHERIT | ACE.FLAGS_CONTAINER_INHERIT | ACE.FLAGS_INHERIT_ONLY;
        assertEquals(0x0B, ace.getFlags());
    }

    @Test

            message.setIpAddress(request.getServerAddress());
            message.setClientComputerName(context.getConfig().getNetbiosHostname());
            message.setFlags(request.getFlags());
            message.setTimeout((int) (context.getConfig().getWitnessRegistrationTimeout() / 1000));

            // Send the RPC request
            rpcHandle.sendrecv(message);

    }

    @Test
    @DisplayName("getFlags delegates to underlying mechanism context")
    void testGetFlagsDelegates() {
        SpnegoContext ctx = newContext();
        when(this.mechContext.getFlags()).thenReturn(0xCAFE);
        assertEquals(0xCAFE, ctx.getFlags());
        verify(this.mechContext, times(1)).getFlags();
    }

    @Test

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

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

    /**
     * Sets the timeout value in seconds.
     *

    }

    /**
     * Gets the SMB message flags
     * @return the flags
     */
    public final byte getFlags() {
        return this.flags;
    }

    /**
     * Sets the SMB message flags
     * @param flags
     *            the flags to set
     */
    public final void setFlags(final byte flags) {
        this.flags = flags;
    }

    /**
     * Gets the SMB message flags2 field

        @Test
        @DisplayName("Should add flags correctly")
        void testAddFlags() {
            assertEquals(0, testMessage.getFlags());
            testMessage.addFlags(ServerMessageBlock2.SMB2_FLAGS_SIGNED);
            assertEquals(ServerMessageBlock2.SMB2_FLAGS_SIGNED, testMessage.getFlags());
            testMessage.addFlags(ServerMessageBlock2.SMB2_FLAGS_DFS_OPERATIONS);

        assertEquals(encodedSize, decodedSize);
        assertArrayEquals(context.getCompressionAlgorithms(), decodedContext.getCompressionAlgorithms());
        assertEquals(context.getFlags(), decodedContext.getFlags());
    }

    @Test
    @DisplayName("Test invalid buffer size for decoding")
    public void testInvalidBufferSize() {
        byte[] smallBuffer = new byte[4]; // Too small

        final MsrpcSamrConnect4 request = new MsrpcSamrConnect4(server, access, this.policyHandle);

        // Then
        assertEquals(0, request.getPtype());
        assertEquals(DCERPC_FIRST_FRAG | DCERPC_LAST_FRAG, request.getFlags());
    }

        public InetAddress getAddress() {
            return address;
        }

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

        /**
         * Checks if this is an IPv4 address.
         *
         * @return true if IPv4
         */
        public boolean isIPv4() {