def test_pipeline_decrement_stage(): p = Pipeline() s1 = Stage() t = Task() t.executable = ['/bin/date'] s1.tasks = t s2 = Stage() t = Task() t.executable = ['/bin/date'] s2.tasks = t p.add_stages([s1, s2]) p._increment_stage() p._increment_stage() assert p._stage_count == 2 assert p._cur_stage == 2 assert p._completed_flag.is_set() == True p._decrement_stage() assert p._stage_count == 2 assert p._cur_stage == 1 assert p._completed_flag.is_set() == False p._decrement_stage() assert p._stage_count == 2 assert p._cur_stage == 0 assert p._completed_flag.is_set() == False
def test_pipeline_decrement_stage(): p = Pipeline() s1 = Stage() t = Task() t.executable = '/bin/date' s1.tasks = t s2 = Stage() t = Task() t.executable = '/bin/date' s2.tasks = t p.add_stages([s1, s2]) p._increment_stage() p._increment_stage() assert p._stage_count == 2 assert p._cur_stage == 2 assert p._completed_flag.is_set() == True p._decrement_stage() assert p._stage_count == 2 assert p._cur_stage == 1 assert p._completed_flag.is_set() == False p._decrement_stage() assert p._stage_count == 2 assert p._cur_stage == 0 assert p._completed_flag.is_set() == False
def test_pipeline_decrement_stage(self, mocked_init): p = Pipeline() p._completed_flag = mock.Mock() p._completed_flag.is_set = mock.MagicMock(side_effect=[False, True]) p._cur_stage = 2 p._stage_count = 2 p._decrement_stage() self.assertEqual(p._stage_count, 2) self.assertEqual(p._cur_stage, 1) self.assertFalse(p._completed_flag.is_set(), False) p._decrement_stage() self.assertEqual(p._stage_count, 2) self.assertEqual(p._cur_stage, 0) self.assertFalse(p._completed_flag.is_set(), False)