def test_merger(self):
link_ids = list()
link_ids2 = list()
proc_ids = list()
proc_ids2 = list()
for i in range(3):
link_ids.append(i)
for i in range(3, 6):
link_ids2.append(i)
for i in range(9):
proc_ids.append(i)
for i in range(9, 18):
proc_ids2.append(i)
key = 1
mask = 1
a_multicast = MulticastRoutingEntry(
key, mask, proc_ids, link_ids, True)
b_multicast = MulticastRoutingEntry(
key, mask, proc_ids2, link_ids2, True)
result_multicast = a_multicast.merge(b_multicast)
comparison_link_ids = list()
comparison_proc_ids = list()
for i in range(6):
comparison_link_ids.append(i)
self.assertEqual(link_ids + link_ids2, comparison_link_ids)
for i in range(18):
comparison_proc_ids.append(i)
self.assertEqual(proc_ids + proc_ids2, comparison_proc_ids)
self.assertEqual(result_multicast.routing_entry_key, key)
self.assertEqual(result_multicast.link_ids, set(comparison_link_ids))
self.assertEqual(result_multicast.mask, mask)
self.assertEqual(result_multicast.processor_ids,
set(comparison_proc_ids))
def test_merger_with_invalid_parameter_mask(self):
link_ids = list()
link_ids2 = list()
proc_ids = list()
proc_ids2 = list()
for i in range(3):
link_ids.append(i)
for i in range(3, 6):
link_ids2.append(i)
for i in range(9):
proc_ids.append(i)
for i in range(9, 18):
proc_ids2.append(i)
key = 1
mask = 1
a_multicast = MulticastRoutingEntry(
key, mask, proc_ids, link_ids, True)
b_multicast = MulticastRoutingEntry(
key + 1, mask + 1, proc_ids2, link_ids2, True)
with self.assertRaises(SpinnMachineInvalidParameterException):
a_multicast.merge(b_multicast)
def test_merger_with_invalid_parameter_key(self):
link_ids = list()
link_ids2 = list()
proc_ids = list()
proc_ids2 = list()
for i in range(3):
link_ids.append(i)
for i in range(3, 6):
link_ids2.append(i)
for i in range(9):
proc_ids.append(i)
for i in range(9, 18):
proc_ids2.append(i)
key_combo = 1
mask = 1
a_multicast = MulticastRoutingEntry(
key_combo, mask, proc_ids, link_ids, True)
b_multicast = MulticastRoutingEntry(
key_combo + 1, mask + 1, proc_ids2, link_ids2, True)
with self.assertRaises(SpinnMachineInvalidParameterException) as e:
a_multicast.merge(b_multicast)
self.assertEqual(e.exception.parameter, "other_entry.key")
self.assertEqual(e.exception.value, "0x2")
self.assertEqual(e.exception.problem, "The key does not match 0x1")