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)