def test_remove_outlet(self, outlet1, outlet2):
link = Link([], [], timedelta(seconds=1), tags='test_remove_outlet')
link.add_outlets([outlet1, outlet2])
link.remove_outlets(outlet2)
self.assertEqual(link.outlets, [outlet1])
def test_add_outlet_same(self, outlet1):
link = Link([], [], timedelta(seconds=1), tags='test_add_outlet_same')
link.add_outlets(outlet1)
self.assertRaises(InvalidNodeError, link.add_outlets, outlet1)
self.assertEqual(link.outlets, [outlet1])
def test_add_outlet_multiple(self, outlet1, outlet2):
link = Link([], [],
timedelta(seconds=1),
tags='test_add_outlet_multiple')
link.add_outlets([outlet1, outlet2])
self.assertEqual(link.outlets, [outlet1, outlet2])
def test_remove_outlet_invalid(self, outlet1, outlet2):
link = Link([], [],
timedelta(seconds=1),
tags='test_remove_outlet_invalid')
link.add_outlets([outlet1])
self.assertRaises(InvalidNodeError, link.remove_outlets, outlet2)
self.assertEqual(link.outlets, [outlet1])
def test_add_outlet(self, outlet1):
link = Link([], [], timedelta(seconds=1), tags='test_add_outlet')
link.add_outlets(outlet1)
self.assertEqual(link.outlets, [outlet1])
