def testPSCommandRebootRetry(self, run): ps = powershell.PSCommand([COMMAND, [0], [1337, 1338], True], self.bi) run.return_value = 1337 with self.assertRaises(powershell.RestartEvent) as cm: ps.Run() exception = cm.exception self.assertEqual(exception.retry_on_restart, True) run.assert_called_with(mock.ANY, TOKENIZED_COMMAND, [0, 1337, 1338])
def testPSCommandCache(self, cache, run): cache.return_value = SCRIPT_PATH ps = powershell.PSCommand([SCRIPT + ' -confirm:$false'], self.bi) run.return_value = 0 ps.Run() run.assert_called_with( mock.ANY, [SCRIPT_PATH, '-confirm:$false', '; exit $LASTEXITCODE'], [0]) cache.assert_called_with(mock.ANY, SCRIPT, self.bi)
def testPSCommandValidate(self): ps = powershell.PSCommand( [COMMAND, [0], [1337, 1338], True, False, True], None) ps.Validate()
def testPSCommandValidateTooMany(self): ps = powershell.PSCommand( [COMMAND, [0], [1337, 1338], True, True, True, True], None) self.assertRaises(powershell.ValidationError, ps.Validate)
def testPSCommandValidateNotEnough(self): ps = powershell.PSCommand([], None) self.assertRaises(powershell.ValidationError, ps.Validate)
def testPSCommandValidateType(self, command, success_codes, reboot_codes, retry_on_restart, shell, log): ps = powershell.PSCommand([ command, success_codes, reboot_codes, retry_on_restart, shell, log ], None) self.assertRaises(powershell.ValidationError, ps.Validate)
def testPSCommandLog(self, ps_lib): ps = powershell.PSCommand([COMMAND, [0], [], False, True, False], self.bi) ps_lib.return_value.RunCommand.return_value = 0 ps.Run() ps_lib.assert_called_with(True, False)
def testPSCommandCacheError(self, cache, run): ps = powershell.PSCommand([SCRIPT + ' -confirm:$false'], self.bi) run.side_effect = None cache.side_effect = powershell.cache.CacheError with self.assertRaises(powershell.ActionError): ps.Run()
def testPSCommandSuccessError(self, run): ps = powershell.PSCommand([COMMAND, [0]], self.bi) run.return_value = 1337 self.assertRaises(powershell.ActionError, ps.Run)
def testPSCommandError(self, run): ps = powershell.PSCommand([COMMAND, [1337]], None) run.side_effect = powershell.powershell.PowerShellError self.assertRaises(powershell.ActionError, ps.Run)
def testPSCommandNoShell(self, ps_lib): """Assert Shell=False by default for mutation testing.""" ps_lib.return_value.RunCommand.return_value = 0 powershell.PSCommand([COMMAND], self.bi).Run() ps_lib.assert_called_with(False, True)
def testPSCommand(self, run): ps = powershell.PSCommand([COMMAND, [1337]], self.bi) run.return_value = 1337 ps.Run() run.assert_called_with(mock.ANY, TOKENIZED_COMMAND, [1337])
def testPSCommandRebootNoRetry(self, run): ps = powershell.PSCommand([COMMAND, [0], [1337, 1338]], self.bi) run.return_value = 1337 self.assertRaises(powershell.RestartEvent, ps.Run) run.assert_called_with(mock.ANY, TOKENIZED_COMMAND, [0, 1337, 1338])