[test] Adding initialization tests
[minor] Added missing `heartbeat timeout`
[test] Adding tests for heartbeats