Add heartbeat test