15c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# Copyright (C) 2010 Google Inc. All rights reserved. 25c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# Copyright (C) 2009 Daniel Bates (dbates@intudata.com). All rights reserved. 35c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# 45c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# Redistribution and use in source and binary forms, with or without 55c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# modification, are permitted provided that the following conditions are 65c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# met: 75c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# 85c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# * Redistributions of source code must retain the above copyright 95c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# notice, this list of conditions and the following disclaimer. 105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# * Redistributions in binary form must reproduce the above 115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# copyright notice, this list of conditions and the following disclaimer 125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# in the documentation and/or other materials provided with the 135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# distribution. 145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# * Neither the name of Google Inc. nor the names of its 155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# contributors may be used to endorse or promote products derived from 165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# this software without specific prior written permission. 175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# 185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 215c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 225c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 235c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 245c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 255c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 265c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 275c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 285c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 305c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import os 315c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import errno 325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import signal 335c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import subprocess 345c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import sys 355c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)import time 36e38fbeeb576b5094e34e038ab88d9d6a5c5c2214Torne (Richard Coles)import unittest 375c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 385c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)# Since we execute this script directly as part of the unit tests, we need to ensure 397757ec2eadfa2dd8ac2aeed0a4399e9b07ec38cbBen Murdoch# that Tools/Scripts and Tools/Scripts/thirdparty are in sys.path for the next imports to work correctly. 405c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)script_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))) 415c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)if script_dir not in sys.path: 425c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) sys.path.append(script_dir) 437757ec2eadfa2dd8ac2aeed0a4399e9b07ec38cbBen Murdochthird_party_py = os.path.join(script_dir, "webkitpy", "thirdparty") 44926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles)if third_party_py not in sys.path: 45926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) sys.path.append(third_party_py) 46926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) 475c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 485c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)from webkitpy.common.system.executive import Executive, ScriptError 495c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)from webkitpy.common.system.filesystem_mock import MockFileSystem 505c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 515c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 525c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)class ScriptErrorTest(unittest.TestCase): 535c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def test_message_with_output(self): 545c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) error = ScriptError('My custom message!', '', -1) 55926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertEqual(error.message_with_output(), 'My custom message!') 565c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) error = ScriptError('My custom message!', '', -1, 'My output.') 57e69819bd8e388ea4ad1636a19aa6b2eed4952191Ben Murdoch self.assertEqual(error.message_with_output(), 'My custom message!\n\noutput: My output.') 585c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) error = ScriptError('', 'my_command!', -1, 'My output.', '/Users/username/blah') 59e69819bd8e388ea4ad1636a19aa6b2eed4952191Ben Murdoch self.assertEqual(error.message_with_output(), 'Failed to run "\'my_command!\'" exit_code: -1 cwd: /Users/username/blah\n\noutput: My output.') 605c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) error = ScriptError('', 'my_command!', -1, 'ab' + '1' * 499) 61e69819bd8e388ea4ad1636a19aa6b2eed4952191Ben Murdoch self.assertEqual(error.message_with_output(), 'Failed to run "\'my_command!\'" exit_code: -1\n\noutput: Last 500 characters of output:\nb' + '1' * 499) 62926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) 63926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) def test_message_with_tuple(self): 64926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) error = ScriptError('', ('my', 'command'), -1, 'My output.', '/Users/username/blah') 65e69819bd8e388ea4ad1636a19aa6b2eed4952191Ben Murdoch self.assertEqual(error.message_with_output(), 'Failed to run "(\'my\', \'command\')" exit_code: -1 cwd: /Users/username/blah\n\noutput: My output.') 665c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 675c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)def never_ending_command(): 685c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) """Arguments for a command that will never end (useful for testing process 695c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) killing). It should be a process that is unlikely to already be running 705c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) because all instances will be killed.""" 715c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if sys.platform == 'win32': 725c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return ['wmic'] 735c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return ['yes'] 745c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 755c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 765c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)def command_line(cmd, *args): 775c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return [sys.executable, __file__, '--' + cmd] + list(args) 785c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 795c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 805c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)class ExecutiveTest(unittest.TestCase): 815c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def assert_interpreter_for_content(self, intepreter, content): 825c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) fs = MockFileSystem() 835c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 845c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) tempfile, temp_name = fs.open_binary_tempfile('') 855c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) tempfile.write(content) 865c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) tempfile.close() 875c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) file_interpreter = Executive.interpreter_for_script(temp_name, fs) 885c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 895c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assertEqual(file_interpreter, intepreter) 905c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 915c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def test_interpreter_for_script(self): 925c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content(None, '') 935c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content(None, 'abcd\nefgh\nijklm') 945c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content(None, '##/usr/bin/perl') 955c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content('perl', '#!/usr/bin/env perl') 965c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content('perl', '#!/usr/bin/env perl\nfirst\nsecond') 975c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content('perl', '#!/usr/bin/perl') 985c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content('perl', '#!/usr/bin/perl -w') 995c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content(sys.executable, '#!/usr/bin/env python') 1005c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content(sys.executable, '#!/usr/bin/env python\nfirst\nsecond') 1015c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content(sys.executable, '#!/usr/bin/python') 1025c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content('ruby', '#!/usr/bin/env ruby') 1035c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content('ruby', '#!/usr/bin/env ruby\nfirst\nsecond') 1045c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assert_interpreter_for_content('ruby', '#!/usr/bin/ruby') 1055c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1065c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def test_run_command_with_bad_command(self): 1075c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def run_bad_command(): 1085c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) Executive().run_command(["foo_bar_command_blah"], error_handler=Executive.ignore_error, return_exit_code=True) 109926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertRaises(OSError, run_bad_command) 1105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def test_run_command_args_type(self): 1125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) executive = Executive() 1135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assertRaises(AssertionError, executive.run_command, "echo") 1145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assertRaises(AssertionError, executive.run_command, u"echo") 1155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) executive.run_command(command_line('echo', 'foo')) 1165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) executive.run_command(tuple(command_line('echo', 'foo'))) 1175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 118926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) def test_auto_stringify_args(self): 119926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) executive = Executive() 120926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) executive.run_command(command_line('echo', 1)) 121926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) executive.popen(command_line('echo', 1), stdout=executive.PIPE).wait() 122926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertEqual('echo 1', executive.command_for_printing(['echo', 1])) 123926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) 124926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) def test_popen_args(self): 125926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) executive = Executive() 126926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) # Explicitly naming the 'args' argument should not thow an exception. 127926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) executive.popen(args=command_line('echo', 1), stdout=executive.PIPE).wait() 128926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) 1295c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def test_run_command_with_unicode(self): 1305c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) """Validate that it is safe to pass unicode() objects 1315c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) to Executive.run* methods, and they will return unicode() 1325c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) objects by default unless decode_output=False""" 1335c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unicode_tor_input = u"WebKit \u2661 Tor Arne Vestb\u00F8!" 1345c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if sys.platform == 'win32': 1355c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) encoding = 'mbcs' 1365c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) else: 1375c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) encoding = 'utf-8' 1385c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) encoded_tor = unicode_tor_input.encode(encoding) 1395c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) # On Windows, we expect the unicode->mbcs->unicode roundtrip to be 1405c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) # lossy. On other platforms, we expect a lossless roundtrip. 1415c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if sys.platform == 'win32': 1425c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unicode_tor_output = encoded_tor.decode(encoding) 1435c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) else: 1445c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) unicode_tor_output = unicode_tor_input 1455c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1465c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) executive = Executive() 1475c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1485c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) output = executive.run_command(command_line('cat'), input=unicode_tor_input) 149926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertEqual(output, unicode_tor_output) 1505c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1515c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) output = executive.run_command(command_line('echo', unicode_tor_input)) 152926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertEqual(output, unicode_tor_output) 1535c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1545c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) output = executive.run_command(command_line('echo', unicode_tor_input), decode_output=False) 155926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertEqual(output, encoded_tor) 1565c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1575c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) # Make sure that str() input also works. 1585c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) output = executive.run_command(command_line('cat'), input=encoded_tor, decode_output=False) 159926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertEqual(output, encoded_tor) 1605c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1615c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) # FIXME: We should only have one run* method to test 1625c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) output = executive.run_and_throw_if_fail(command_line('echo', unicode_tor_input), quiet=True) 163926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertEqual(output, unicode_tor_output) 1645c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1655c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) output = executive.run_and_throw_if_fail(command_line('echo', unicode_tor_input), quiet=True, decode_output=False) 166926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertEqual(output, encoded_tor) 1675c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 168e38fbeeb576b5094e34e038ab88d9d6a5c5c2214Torne (Richard Coles) def test_kill_process(self): 1695c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) executive = Executive() 1705c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) process = subprocess.Popen(never_ending_command(), stdout=subprocess.PIPE) 1715c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assertEqual(process.poll(), None) # Process is running 1725c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) executive.kill_process(process.pid) 1735c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1745c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) # Killing again should fail silently. 1755c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) executive.kill_process(process.pid) 1765c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1775c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def _assert_windows_image_name(self, name, expected_windows_name): 1785c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) executive = Executive() 1795c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) windows_name = executive._windows_image_name(name) 1805c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assertEqual(windows_name, expected_windows_name) 1815c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 1825c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def test_windows_image_name(self): 1835c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self._assert_windows_image_name("foo", "foo.exe") 1845c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self._assert_windows_image_name("foo.exe", "foo.exe") 1855c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self._assert_windows_image_name("foo.com", "foo.com") 1865c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) # If the name looks like an extension, even if it isn't 1875c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) # supposed to, we have no choice but to return the original name. 1885c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self._assert_windows_image_name("foo.baz", "foo.baz") 1895c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self._assert_windows_image_name("foo.baz.exe", "foo.baz.exe") 1905c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 191e38fbeeb576b5094e34e038ab88d9d6a5c5c2214Torne (Richard Coles) def test_check_running_pid(self): 1925c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) executive = Executive() 1935c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assertTrue(executive.check_running_pid(os.getpid())) 1945c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) # Maximum pid number on Linux is 32768 by default 1955c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assertFalse(executive.check_running_pid(100000)) 1965c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 197e38fbeeb576b5094e34e038ab88d9d6a5c5c2214Torne (Richard Coles) def test_running_pids(self): 1985c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if sys.platform in ("win32", "cygwin"): 1995c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return # This function isn't implemented on Windows yet. 2005c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2015c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) executive = Executive() 2025c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) pids = executive.running_pids() 203926b001d589ce2f10facb93dd4b87578ea35a855Torne (Richard Coles) self.assertIn(os.getpid(), pids) 2045c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2055c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) def test_run_in_parallel_assert_nonempty(self): 2065c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) self.assertRaises(AssertionError, Executive().run_in_parallel, []) 2075c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2085c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2095c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)def main(platform, stdin, stdout, cmd, args): 2105c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if platform == 'win32' and hasattr(stdout, 'fileno'): 2115c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) import msvcrt 2125c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) msvcrt.setmode(stdout.fileno(), os.O_BINARY) 2135c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) if cmd == '--cat': 2145c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) stdout.write(stdin.read()) 2155c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) elif cmd == '--echo': 2165c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) stdout.write(' '.join(args)) 2175c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) return 0 2185c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) 2195c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles)if __name__ == '__main__' and len(sys.argv) > 1 and sys.argv[1] in ('--cat', '--echo'): 2205c87bf8b86a7c82ef50fb7a89697d8e02e2553beTorne (Richard Coles) sys.exit(main(sys.platform, sys.stdin, sys.stdout, sys.argv[1], sys.argv[2:])) 221