def test_set_stage_first(self, gv, sv): gv.return_value = None stage.set_stage(1) sv.assert_has_calls([ mock.call('Start', mock.ANY, 'HKLM', stage.STAGES_ROOT + r'\1'), mock.call('_Active', '1', 'HKLM', stage.STAGES_ROOT) ])
def Run(self): try: stage.set_stage(int(self._args[0])) # Terminal stages exit immediately; the build should be complete. if len(self._args) > 1 and self._args[1]: stage.exit_stage(int(self._args[0])) except stage.Error as e: raise ActionError(str(e))
def test_set_stage_first(self, reg, get_active): get_active.return_value = None stage.set_stage(1) reg.assert_called_with('HKLM') reg.return_value.SetKeyValue.assert_has_calls([ mock.call(key_path=stage.STAGES_ROOT + r'\1', key_name='Start', key_value=mock.ANY, key_type='REG_SZ', use_64bit=True), mock.call(key_path=stage.STAGES_ROOT, key_name='_Active', key_value='1', key_type='REG_SZ', use_64bit=True) ])
def test_set_stage_next(self, unused_reg, get_active, exit_stage): get_active.return_value = '1' stage.set_stage(2) exit_stage.assert_called_with('1')
def test_set_stage_next(self, reg, get_active, exit_stage): get_active.return_value = 1 stage.set_stage(2) exit_stage.assert_called_with(1) reg.assert_called_with('_Active', '2', 'HKLM', stage.STAGES_ROOT)
def Run(self): try: stage.set_stage(int(self._args[0])) except stage.Error as e: raise ActionError(str(e))