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])
