def test_luid(self, mocked_init): s = Stage() s._p_pipeline = {'uid': 'pipe.0000', 'name': None} s._uid = 'stage.0000' s._name = None self.assertEqual(s.luid, 'pipe.0000.stage.0000') s = Stage() s._p_pipeline = {'uid': 'pipe.0000', 'name': 'test_pipe'} s._uid = 'stage.0000' s._name = None self.assertEqual(s.luid, 'test_pipe.stage.0000') s = Stage() s._p_pipeline = {'uid': 'pipe.0000', 'name': None} s._uid = 'stage.0000' s._name = 'test_stage' self.assertEqual(s.luid, 'pipe.0000.test_stage') s = Stage() s._p_pipeline = {'uid': 'pipe.0000', 'name': 'test_pipe'} s._uid = 'stage.0000' s._name = 'test_stage' self.assertEqual(s.luid, 'test_pipe.test_stage')
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_from_dict(self, mocked_init): d = { 'uid': 're.Stage.0000', 'name': 's1', 'state': states.DONE, 'state_history': [states.INITIAL, states.DONE], 'parent_pipeline': { 'uid': 'p1', 'name': 'pipe1' } } s = Stage() s._uid = None s._name = None s._state = None s._state_history = None s._p_pipeline = None s.from_dict(d) self.assertEqual(s._uid, d['uid']) self.assertEqual(s._name, d['name']) self.assertEqual(s._state, d['state']) self.assertEqual(s._state_history, d['state_history']) self.assertEqual(s._p_pipeline, d['parent_pipeline'])
def test_stage_to_dict(self, mocked_init): s = Stage() s._uid = 'stage.0000' s._name = 'test_stage' s._state = states.INITIAL s._state_history = [states.INITIAL] s._p_pipeline = {'uid': 'pipeline.0000', 'name': 'parent'} self.assertEqual( s.to_dict(), { 'uid': 'stage.0000', 'name': 'test_stage', 'state': states.INITIAL, 'state_history': [states.INITIAL], 'parent_pipeline': { 'uid': 'pipeline.0000', 'name': 'parent' } })