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)
