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

        }

        @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

            assertNull(testBlock.getPath());
            assertEquals(12345, testBlock.getPid());
            assertEquals(SmbConstants.FLAGS_PATH_NAMES_CASELESS | SmbConstants.FLAGS_PATH_NAMES_CANONICALIZED, testBlock.getFlags());
        }

        @Test
        @DisplayName("Test constructor with config and command")
        void testConstructorWithCommand() {
            testBlock = new TestServerMessageBlock(mockConfig, (byte) 0x25);

        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

        assertEquals(0, msrpcSamrConnect2.getPtype(), "getPtype() should return 0");
        assertEquals(DcerpcConstants.DCERPC_FIRST_FRAG | DcerpcConstants.DCERPC_LAST_FRAG, msrpcSamrConnect2.getFlags(),
                "getFlags() should return correct flags");
        assertTrue(msrpcSamrConnect2.isFlagSet(DcerpcConstants.DCERPC_FIRST_FRAG), "DCERPC_FIRST_FRAG should be set");

    public byte[] toByteArray() {
        try {
            final String suppliedDomain = getSuppliedDomain();
            final String suppliedWorkstation = getSuppliedWorkstation();
            int flags = getFlags();
            boolean hostInfo = false;
            byte[] domain = {};
            if (suppliedDomain != null && suppliedDomain.length() != 0) {
                hostInfo = true;

            RequirementMatcher matcher = provides.get(key);

            if (matcher == null) {
                getLog().debug("Toolchain {} is missing required property: {}", this, key);
                return false;
            }
            if (!matcher.matches(requirement.getValue())) {
                getLog().debug("Toolchain {} doesn't match required property: {}", this, key);
                return false;
            }
        }

        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() {

            if ((getFlags() & NTLMSSP_NEGOTIATE_SIGN) == NTLMSSP_NEGOTIATE_SIGN) {
                final HMACT64 hmac = new HMACT64(responseKeyNT);
                hmac.update(ntResponse, 0, 16); // only first 16 bytes of ntResponse
                final byte[] userSessionKey = hmac.digest();

                if ((getFlags() & NTLMSSP_NEGOTIATE_KEY_EXCH) != 0) {

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

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