}

    /**
     * Sets the context handle for the heartbeat.
     *
     * @param contextHandle the context handle from registration
     */
    public void setContextHandle(byte[] contextHandle) {
        this.contextHandle = contextHandle != null ? contextHandle.clone() : null;
    }

    /**
     * Gets the context handle.
     *
     * @return the context handle
     */
    public byte[] getContextHandle() {

        assertEquals(Smb2LeaseState.SMB2_LEASE_FULL, fullLease);
    }

    @Test
    @DisplayName("Should handle custom state combinations")
    void testCustomStateCombinations() {
        int customState = 0x06; // Write + Handle but no Read

        assertFalse(Smb2LeaseState.hasReadCaching(customState));
        assertTrue(Smb2LeaseState.hasWriteCaching(customState));

            fail("setObject should handle parameter validation gracefully: " + e.getMessage());
        }

        // getObject should handle various inputs
        try {
            WebApiUtil.getObject("validKey");
            WebApiUtil.getObject("");
            WebApiUtil.getObject(null);
        } catch (Exception e) {
            fail("getObject should handle parameter validation gracefully: " + e.getMessage());

            throw new IOException("Witness async notify RPC failed", e);
        }
    }

    /**
     * Generates a registration ID from context handle and share name.
     *
     * @param contextHandle the context handle
     * @param shareName the share name
     * @return the registration ID
     */
    private String generateRegistrationId(byte[] contextHandle, String shareName) {

     * @param handler
     *            The SAX {@link DefaultHandler} to use. Must not be {@literal null}.
     */
    public static void parse(final SAXParser parser, final InputSource inputSource, final DefaultHandler handler) {
        assertArgumentNotNull("parser", parser);
        assertArgumentNotNull("inputSource", inputSource);
        assertArgumentNotNull("handler", handler);

        try {

        // When
        request.setSessionBinding(false);

        // Then
        assertFalse((boolean) sessionBindingField.get(request));
    }

    @Test
    @DisplayName("Should handle chain operation correctly")
    void testChain() {
        // Given
        ServerMessageBlock2 nextMessage = mock(ServerMessageBlock2.class);

        // When
        boolean result = request.chain(nextMessage);

         *
         * @param handle the policy handle to close
         */
        public LsarClose(final rpc.policy_handle handle) {
            this.handle = handle;
        }

        @Override
        public void encode_in(final NdrBuffer _dst) throws NdrException {
            handle.encode(_dst);
        }

        @Override

                }
                policyHandle = new LsaPolicyHandle(handle, "\\\\" + server, 0x00000800);
                SID.resolveSids(handle, policyHandle, sids);
            } finally {
                if (handle != null) {
                    if (policyHandle != null) {
                        policyHandle.close();
                    }
                    handle.close();
                }
            }
        }
    }

            SIDCacheImpl cache = new SIDCacheImpl(mock(CIFSContext.class));
            DcerpcHandle handle = mock(DcerpcHandle.class);
            LsaPolicyHandle policy = mock(LsaPolicyHandle.class);
            jcifs.SID[] none = new jcifs.SID[0];

            cache.resolveSids(handle, policy, none);
            verify(handle, times(1)).sendrecv(any(jcifs.dcerpc.msrpc.MsrpcLookupSids.class));
        }
    }

        // Assert
        assertEquals(level1, queryPolicy1.level);
        assertEquals(level2, queryPolicy2.level);
        assertSame(mockPolicyHandle, queryPolicy1.handle);
        assertSame(mockPolicyHandle2, queryPolicy2.handle);
        assertSame(mockNdrObject, queryPolicy1.info);
        assertSame(mockNdrObject2, queryPolicy2.info);
    }

    @Test
    void constructor_shouldInitializeRetvalToZero() {