def test_stage_check_complete(): s = Stage() t1 = Task() t1.executable = ['/bin/date'] t2 = Task() t2.executable = ['/bin/date'] s.add_tasks([t1, t2]) assert s._check_stage_complete() == False s._set_tasks_state(states.DONE) assert s._check_stage_complete() == True
def test_stage_check_complete(): s = Stage() t1 = Task() t1.executable = '/bin/date' t2 = Task() t2.executable = '/bin/date' s.add_tasks([t1, t2]) assert s._check_stage_complete() is False s._set_tasks_state(states.DONE) assert s._check_stage_complete() is True
def test_stage_set_tasks_state(self, mocked_init): s = Stage() s._uid = 'stage.0000' t1 = mock.MagicMock(spec=Task) t2 = mock.MagicMock(spec=Task) s._tasks = set([t1, t2]) with self.assertRaises(ValueError): s._set_tasks_state(2) s._set_tasks_state(states.DONE) self.assertEqual(t1.state, states.DONE) self.assertEqual(t2.state, states.DONE)
def test_stage_set_tasks_state(): s = Stage() t1 = Task() t1.executable = ['/bin/date'] t2 = Task() t2.executable = ['/bin/date'] s.add_tasks([t1, t2]) with pytest.raises(ValueError): s._set_tasks_state(2) s._set_tasks_state(states.DONE) assert t1.state == states.DONE assert t2.state == states.DONE
def test_stage_set_tasks_state(): s = Stage() t1 = Task() t1.executable = '/bin/date' t2 = Task() t2.executable = '/bin/date' s.add_tasks([t1, t2]) with pytest.raises(ValueError): s._set_tasks_state(2) s._set_tasks_state(states.DONE) assert t1.state == states.DONE assert t2.state == states.DONE