def test_on_shutdown_already_inactive(self, inlet1, outlet1):
type(inlet1).active = mock.PropertyMock(return_value=False)
type(outlet1).active = mock.PropertyMock(return_value=False)
link = Link([inlet1], [outlet1],
timedelta(seconds=1),
tags='test_on_shutdown_already_inactive')
link.on_shutdown()
inlet1.on_shutdown.assert_not_called()
outlet1.on_shutdown.assert_not_called()
def test_on_shutdown(self, inlet1, outlet1):
type(inlet1).active = mock.PropertyMock(return_value=True)
type(outlet1).active = mock.PropertyMock(return_value=True)
link = Link([inlet1], [outlet1],
timedelta(seconds=1),
tags='test_on_shutdown')
link.on_shutdown()
inlet1.try_shutdown.assert_called()
outlet1.try_shutdown.assert_called()
def test_on_shutdown_already_inactive(self):
inlet1 = DummyInlet()
outlet1 = DummyOutlet()
inlet1.on_shutdown = mock.Mock()
outlet1.on_shutdown = mock.Mock()
link = Link([inlet1], [outlet1], timedelta(seconds=1))
link.on_shutdown()
inlet1.on_shutdown.assert_not_called()
outlet1.on_shutdown.assert_not_called()
def test_on_shutdown(self):
inlet1 = DummyInlet()
outlet1 = DummyOutlet()
inlet1._active = True
outlet1._active = True
link = Link(inlet1, outlet1, timedelta(seconds=1))
self.assertTrue(inlet1.active)
self.assertTrue(outlet1.active)
link.on_shutdown()
self.assertFalse(inlet1.active)
self.assertFalse(outlet1.active)
def test_on_shutdown_inlet_exception_catch(self, inlet1, outlet1):
inlet1.try_shutdown.side_effect = lambda: exec('raise(RuntimeError())')
link = Link([inlet1], [outlet1],
timedelta(seconds=1),
tags='test_on_shutdown',
ignore_exceptions=True)
with self.assertLogs(logging.getLogger('databay.Link'),
level='ERROR') as cm:
link.on_shutdown()
self.assertTrue(
'on_shutdown inlet exception: "" for inlet:'
in ';'.join(cm.output), cm.output)
inlet1.try_shutdown.assert_called()
outlet1.try_shutdown.assert_called()
