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)
