if (response != null && response.isSuccess()) {
                registration.updateHeartbeat();
            } else {
                log.warn("Witness heartbeat failed for: {}", registration.getRegistrationId());
                registration.incrementHeartbeatFailures();
            }

        } catch (Exception e) {

        }
    }

    /**
     * Sends a witness heartbeat.
     *
     * @param request the heartbeat request
     * @return the heartbeat response
     * @throws IOException if the RPC call fails
     */
    public WitnessHeartbeatResponse heartbeat(WitnessHeartbeatRequest request) throws IOException {
        if (!connected) {

        String regId = mockService.registerWitness("\\\\server\\share", "192.168.1.100", 1);

        // Test successful heartbeat
        assertTrue(mockService.processHeartbeat(regId, 1));
        assertTrue(mockService.processHeartbeat(regId, 2));

        // Test heartbeat for non-existent registration
        assertFalse(mockService.processHeartbeat("non-existent", 1));

        // Clean up

            WitnessHeartbeatResponse response = rpcClient.heartbeat(request);
            
            if (response.isSuccess()) {
                registration.updateHeartbeat();
            } else {
                log.warn("Witness heartbeat failed for: {}", registration.getRegistrationId());
            }
            
        } catch (Exception e) {
            log.debug("Heartbeat error for registration: {}", 

    /**
     * @return timeout for persistent handles in milliseconds
     */
    long getPersistentHandleTimeout();

    /**
     * Gets the witness heartbeat timeout in milliseconds.
     *
     * @return the heartbeat timeout
     */
    long getWitnessHeartbeatTimeout();

    /**
     * Gets the witness registration timeout in milliseconds.
     *
     * @return the registration timeout

    /**
     * Flag indicating whether SMB Witness protocol should be used for failover
     */
    protected boolean useWitness = false; // Disabled by default
    /**
     * Timeout in milliseconds for witness heartbeat messages
     */
    protected long witnessHeartbeatTimeout = 120000; // 2 minutes
    /**
     * Timeout in milliseconds for witness registration
     */