def testPSScriptSuccessCodes(self, cache, run):
cache.return_value = SCRIPT_PATH
ps = powershell.PSScript([SCRIPT, ARGS, [1337, 1338]], self.bi)
run.return_value = 0
self.assertRaises(powershell.ActionError, ps.Run)
run.return_value = 1337
ps.Run()
def testPSScriptValidateType(self, script, ps_args, success_codes,
reboot_codes, retry_on_restart, shell, log):
ps = powershell.PSScript([
script, ps_args, success_codes, reboot_codes, retry_on_restart,
shell, log
], None)
self.assertRaises(powershell.ValidationError, ps.Validate)
def testPSScriptLog(self, cache, ps_lib):
cache.return_value = SCRIPT_PATH
ps = powershell.PSScript([SCRIPT, ARGS, [0], [], False, True, False],
self.bi)
ps_lib.return_value.RunLocal.return_value = 0
ps.Run()
ps_lib.assert_called_with(True, False)
def testPSScriptRebootRetry(self, cache, run): cache.return_value = SCRIPT_PATH ps = powershell.PSScript([SCRIPT, ARGS, [0], [1337, 1338], True], self.bi) run.return_value = 1337 self.assertRaises(powershell.RestartEvent, ps.Run) run.assert_called_with(mock.ANY, SCRIPT_PATH, ARGS, [0, 1337, 1338]) cache.assert_called_with(mock.ANY, SCRIPT, self.bi)
def testPSScriptRebootNoRetry(self, cache, run):
cache.return_value = SCRIPT_PATH
run.return_value = 1337
with self.assertRaises(powershell.RestartEvent) as cm:
powershell.PSScript([SCRIPT, ARGS, [0], [1337, 1338]], self.bi).Run()
self.assertEqual(cm.exception.retry_on_restart, False)
run.assert_called_with(mock.ANY, SCRIPT_PATH, ARGS, [0, 1337, 1338])
def testPSScript(self, cache, run):
cache.return_value = SCRIPT_PATH
run.return_value = 0
ps = powershell.PSScript([SCRIPT, ARGS], self.bi)
ps.Run()
run.assert_called_with(mock.ANY, SCRIPT_PATH, ARGS, [0])
run.side_effect = powershell.powershell.PowerShellError
self.assertRaises(powershell.ActionError, ps.Run)
# Cache error
run.side_effect = None
cache.side_effect = powershell.cache.CacheError
self.assertRaises(powershell.ActionError, ps.Run)
def testPSScript(self, cache, run):
bi = buildinfo.BuildInfo()
cache.return_value = r'C:\Cache\Some-Script.ps1'
ps = powershell.PSScript(['#Some-Script.ps1', ['-Flag1']], bi)
ps.Run()
cache.assert_called_with(mock.ANY, '#Some-Script.ps1', bi)
run.assert_called_with(
mock.ANY, r'C:\Cache\Some-Script.ps1', args=['-Flag1'])
run.side_effect = powershell.powershell.PowerShellError
self.assertRaises(powershell.ActionError, ps.Run)
# Cache error
run.side_effect = None
cache.side_effect = powershell.cache.CacheError
self.assertRaises(powershell.ActionError, ps.Run)
def testPSScriptValidate(self):
ps = powershell.PSScript(30, None)
self.assertRaises(powershell.ValidationError, ps.Validate)
ps = powershell.PSScript([], None)
self.assertRaises(powershell.ValidationError, ps.Validate)
ps = powershell.PSScript([30, 40], None)
self.assertRaises(powershell.ValidationError, ps.Validate)
ps = powershell.PSScript(['#Some-Script.ps1'], None)
ps.Validate()
ps = powershell.PSScript(['#Some-Script.ps1', '-Flags'], None)
self.assertRaises(powershell.ValidationError, ps.Validate)
ps = powershell.PSScript(['#Some-Script.ps1', ['-Flags']], None)
ps.Validate()
def testPSScriptNoShell(self, cache, ps_lib):
"""Assert Shell=False by default for mutation testing."""
cache.return_value = SCRIPT_PATH
ps_lib.return_value.RunLocal.return_value = 0
powershell.PSScript([SCRIPT], self.bi).Run()
ps_lib.assert_called_with(False, True)
def testPSScriptValidate(self):
ps = powershell.PSScript(
[SCRIPT, ARGS, [0], [1337, 1338], True, False, True], None)
ps.Validate()
def testPSScriptValidateLen(self):
ps = powershell.PSScript([], None)
self.assertRaises(powershell.ValidationError, ps.Validate)
ps = powershell.PSScript([1, 2, 3, 4, 5, 6], None)
self.assertRaises(powershell.ValidationError, ps.Validate)
