def test_recovery(self):
"""Checks the recovery mechanism."""
pipe = Pipeline(common.create_pitivi_mock())
pipe.set_timeline(GES.Timeline())
pipeline_died_cb = mock.Mock()
pipe.connect("died", pipeline_died_cb)
with mock.patch.object(pipe, "set_state") as set_state:
pipe.pause()
set_state.assert_called_once_with(Gst.State.PAUSED)
self.assertEqual(pipe._attempted_recoveries, 0)
self.assertFalse(pipeline_died_cb.called)
for i in range(MAX_RECOVERIES):
with mock.patch.object(pipe, "set_state") as set_state:
set_state.return_value = Gst.StateChangeReturn.SUCCESS
self.post_fake_error_message(pipe)
set_state.assert_has_calls(
[mock.call(Gst.State.NULL),
mock.call(Gst.State.PAUSED)])
self.assertEqual(set_state.call_count, 2)
self.assertEqual(pipe._attempted_recoveries, i + 1)
self.assertFalse(pipeline_died_cb.called)
with mock.patch.object(pipe, "set_state") as set_state:
self.post_fake_error_message(pipe)
set_state.assert_not_called()
self.assertTrue(pipeline_died_cb.called)
self.assertEqual(pipe._attempted_recoveries, MAX_RECOVERIES)
def test_recovery(self):
"""Checks the recovery mechanism."""
pipe = Pipeline(common.create_pitivi_mock())
pipe.set_timeline(GES.Timeline())
pipeline_died_cb = mock.Mock()
pipe.connect("died", pipeline_died_cb)
with mock.patch.object(pipe, "set_state") as set_state:
pipe.pause()
set_state.assert_called_once_with(Gst.State.PAUSED)
self.assertEqual(pipe._attempted_recoveries, 0)
self.assertFalse(pipeline_died_cb.called)
for i in range(MAX_RECOVERIES):
with mock.patch.object(pipe, "set_state") as set_state:
set_state.return_value = Gst.StateChangeReturn.SUCCESS
self.post_fake_error_message(pipe)
set_state.assert_has_calls([mock.call(Gst.State.NULL),
mock.call(Gst.State.PAUSED)])
self.assertEqual(set_state.call_count, 2)
self.assertEqual(pipe._attempted_recoveries, i + 1)
self.assertFalse(pipeline_died_cb.called)
with mock.patch.object(pipe, "set_state") as set_state:
self.post_fake_error_message(pipe)
set_state.assert_not_called()
self.assertTrue(pipeline_died_cb.called)
self.assertEqual(pipe._attempted_recoveries, MAX_RECOVERIES)
