def test_stage_task_addition(self, mocked_init): s = Stage() s._p_pipeline = {'uid': None, 'name': None} s._uid = 'stage.0000' s._name = None s._tasks = set() t1 = mock.MagicMock(spec=Task) t2 = mock.MagicMock(spec=Task) s.add_tasks(set([t1, t2])) self.assertIsInstance(s.tasks, set) self.assertEqual(s._task_count, 2) self.assertIn(t1, s.tasks) self.assertIn(t2, s.tasks) s = Stage() s._uid = 'stage.0000' s._name = None s._p_pipeline = {'uid': None, 'name': None} s._tasks = set() t1 = mock.MagicMock(spec=Task) t2 = mock.MagicMock(spec=Task) s.add_tasks([t1, t2]) self.assertIsInstance(s.tasks, set) self.assertEqual(s._task_count, 2) self.assertIn(t1, s.tasks) self.assertIn(t2, s.tasks)
def test_stage_check_complete(self, mocked_init): s = Stage() s._uid = 'stage.0000' t1 = mock.MagicMock(spec=Task) t2 = mock.MagicMock(spec=Task) s._tasks = set([t1, t2]) self.assertFalse(s._check_stage_complete()) for t in s._tasks: t.state = states.DONE self.assertTrue(s._check_stage_complete())
def test_stage_validate(self, mocked_init): s = Stage() s._uid = 'stage.0000' s._state = 'test' with self.assertRaises(ValueError): s._validate() s = Stage() s._uid = 'stage.0000' s._state = states.INITIAL s._tasks = None with self.assertRaises(MissingError): s._validate() s = Stage() s._uid = 'stage.0000' t = mock.MagicMock(spec=Stage) t._validate = mock.MagicMock(return_value=True) s._tasks = set([t]) s._state = states.INITIAL s._validate()
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)