OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 2 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 """Sets environment variables needed to run a chromium unit test.""" | 6 """Sets environment variables needed to run a chromium unit test.""" |
7 | 7 |
8 import os | 8 import os |
9 import subprocess | 9 import subprocess |
10 import sys | 10 import sys |
11 | 11 |
12 # This is hardcoded to be src/ relative to this script. | 12 # This is hardcoded to be src/ relative to this script. |
13 ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) | 13 ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) |
14 | 14 |
15 | 15 |
| 16 def fix_python_path(cmd): |
| 17 """Returns the fixed command line to call the right python executable.""" |
| 18 out = cmd[:] |
| 19 if out[0] == 'python': |
| 20 out[0] = sys.executable |
| 21 elif out[0].endswith('.py'): |
| 22 out.insert(0, sys.executable) |
| 23 return out |
| 24 |
| 25 |
16 def run_executable(cmd, env): | 26 def run_executable(cmd, env): |
17 """Runs an executable with: | 27 """Runs an executable with: |
18 - environment variable CR_SOURCE_ROOT set to the root directory. | 28 - environment variable CR_SOURCE_ROOT set to the root directory. |
19 - environment variable LANGUAGE to en_US.UTF-8. | 29 - environment variable LANGUAGE to en_US.UTF-8. |
20 - Reuses sys.executable automatically. | 30 - Reuses sys.executable automatically. |
21 """ | 31 """ |
22 # Many tests assume a English interface... | 32 # Many tests assume a English interface... |
23 env['LANGUAGE'] = 'en_US.UTF-8' | 33 env['LANGUAGE'] = 'en_US.UTF-8' |
24 # Used by base/base_paths_linux.cc | 34 # Used by base/base_paths_linux.cc |
25 env['CR_SOURCE_ROOT'] = os.path.abspath(ROOT_DIR).encode('utf-8') | 35 env['CR_SOURCE_ROOT'] = os.path.abspath(ROOT_DIR).encode('utf-8') |
26 # Ensure paths are correctly separated on windows. | 36 # Ensure paths are correctly separated on windows. |
27 cmd[0] = cmd[0].replace('/', os.path.sep) | 37 cmd[0] = cmd[0].replace('/', os.path.sep) |
28 if cmd[0].endswith('.py'): | 38 cmd = fix_python_path(cmd) |
29 cmd.insert(0, sys.executable) | |
30 try: | 39 try: |
31 return subprocess.call(cmd, env=env) | 40 return subprocess.call(cmd, env=env) |
32 except OSError: | 41 except OSError: |
33 print >> sys.stderr, 'Failed to start %s' % cmd | 42 print >> sys.stderr, 'Failed to start %s' % cmd |
34 raise | 43 raise |
35 | 44 |
36 | 45 |
37 def main(): | 46 def main(): |
38 return run_executable(sys.argv[1:], os.environ.copy()) | 47 return run_executable(sys.argv[1:], os.environ.copy()) |
39 | 48 |
40 | 49 |
41 if __name__ == "__main__": | 50 if __name__ == "__main__": |
42 sys.exit(main()) | 51 sys.exit(main()) |
OLD | NEW |