def test_empty(self):
heartbeatGroups = []
routingTable = {}
candidateSet = []
constructHeartbeatGroups(heartbeatGroups, routingTable, candidateSet)
self.assertEqual(heartbeatGroups, [[]])
def test_three(self):
heartbeatGroups = []
routingTable = {2: [3, 4], 3: [2, 4], 4:[3, 2]}
candidateSet = [2, 3, 4]
constructHeartbeatGroups(heartbeatGroups, routingTable, candidateSet)
self.assertEqual(heartbeatGroups, [[2, 3, 4]])
def test_three_but_three_first(self):
heartbeatGroups = []
routingTable = {2: [3, 4], 3: [2], 4:[2]}
candidateSet = [3, 4, 2]
constructHeartbeatGroups(heartbeatGroups, routingTable, candidateSet)
self.assertEqual(heartbeatGroups, [[3, 2], [4]])
def test_two(self):
heartbeatGroups = []
routingTable = {2: [3], 3: [2]}
candidateSet = [2, 3]
constructHeartbeatGroups(heartbeatGroups, routingTable, candidateSet)
self.assertEqual(heartbeatGroups, [[2, 3]])
