def test_add(self):
hb = NodeHeartBeat(interval=0)
hb.add('spa', '1.1.1.1')
hb.add('cs', None)
assert_that(hb.is_available('cs'), equal_to(False))
def test_get_alive_sp_ip(self):
hb = NodeHeartBeat(interval=0)
hb.add('spa', '1.1.1.1', False)
hb.add('spb', '1.1.1.2', True)
assert_that(hb.get_alive_sp_ip(), equal_to('1.1.1.2'))
def get_test_hb(self):
hb = NodeHeartBeat(interval=0)
hb.add('spa', '1.1.1.1')
hb.add('spb', '1.1.1.2')
return hb
