def test_disconnecting_ports_via_shadow(self):
shadow = ShadowPortArray([self.array_a, self.array_b])
shadow.connect_port('in_both', lambda: 'both')
shadow.connect_port('a_only', lambda: 'a')
shadow.connect_port('b_only', lambda: 'b')
shadow.disconnect_port('in_both')
shadow.disconnect_port('b_only')
with self.assertRaisesRegexp(DisconnectedPort, 'Port "in_both" has not been connected'):
self.array_a.in_both()
with self.assertRaisesRegexp(DisconnectedPort, 'Port "in_both" has not been connected'):
self.array_b.in_both()
with self.assertRaisesRegexp(DisconnectedPort, 'Port "b_only" has not been connected'):
self.array_b.b_only()
self.assertEqual('a', self.array_a.a_only()) # this should remain connected
def test_raise_UnknownPort_when_disconnecting_an_unknown_port(self):
shadow = ShadowPortArray([self.array_a, self.array_b])
with self.assertRaisesRegexp(UnknownPort, '"hello" is not a valid port'):
shadow.disconnect_port('hello')