Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(217)

Side by Side Diff: chrome/test/pylib/common/util.py

Issue 14882007: Android: support glob-style gtest filters with content browser tests (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Use DIR_SOURCE_ROOT. Created 7 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
« no previous file with comments | « chrome/test/pylib/common/unittest_util.py ('k') | no next file » | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 # Use of this source code is governed by a BSD-style license that can be
3 # found in the LICENSE file.
4
5 """Generic utilities for all python scripts."""
6
7 import atexit
8 import httplib
9 import os
10 import signal
11 import stat
12 import subprocess
13 import sys
14 import tempfile
15 import urlparse
16
17
18 def GetPlatformName():
19 """Return a string to be used in paths for the platform."""
20 if IsWindows():
21 return 'win'
22 if IsMac():
23 return 'mac'
24 if IsLinux():
25 return 'linux'
26 raise NotImplementedError('Unknown platform "%s".' % sys.platform)
27
28
29 def IsWindows():
30 return sys.platform == 'cygwin' or sys.platform.startswith('win')
31
32
33 def IsLinux():
34 return sys.platform.startswith('linux')
35
36
37 def IsMac():
38 return sys.platform.startswith('darwin')
39
40
41 def _DeleteDir(path):
42 """Deletes a directory recursively, which must exist."""
43 # Don't use shutil.rmtree because it can't delete read-only files on Win.
44 for root, dirs, files in os.walk(path, topdown=False):
45 for name in files:
46 filename = os.path.join(root, name)
47 os.chmod(filename, stat.S_IWRITE)
48 os.remove(filename)
49 for name in dirs:
50 os.rmdir(os.path.join(root, name))
51 os.rmdir(path)
52
53
54 def Delete(path):
55 """Deletes the given file or directory (recursively), which must exist."""
56 if os.path.isdir(path):
57 _DeleteDir(path)
58 else:
59 os.remove(path)
60
61
62 def MaybeDelete(path):
63 """Deletes the given file or directory (recurisvely), if it exists."""
64 if os.path.exists(path):
65 Delete(path)
66
67
68 def MakeTempDir(parent_dir=None):
69 """Creates a temporary directory and returns an absolute path to it.
70
71 The temporary directory is automatically deleted when the python interpreter
72 exits normally.
73
74 Args:
75 parent_dir: the directory to create the temp dir in. If None, the system
76 temp dir is used.
77
78 Returns:
79 The absolute path to the temporary directory.
80 """
81 path = tempfile.mkdtemp(dir=parent_dir)
82 atexit.register(MaybeDelete, path)
83 return path
84
85
86 def Unzip(zip_path, output_dir):
87 """Unzips the given zip file using a system installed unzip tool.
88
89 Args:
90 zip_path: zip file to unzip.
91 output_dir: directory to unzip the contents of the zip file. The directory
92 must exist.
93
94 Raises:
95 RuntimeError if the unzip operation fails.
96 """
97 if IsWindows():
98 unzip_cmd = ['C:\\Program Files\\7-Zip\\7z.exe', 'x', '-y']
99 else:
100 unzip_cmd = ['unzip', '-o']
101 unzip_cmd += [zip_path]
102 if RunCommand(unzip_cmd, output_dir) != 0:
103 raise RuntimeError('Unable to unzip %s to %s' % (zip_path, output_dir))
104
105
106 def Kill(pid):
107 """Terminate the given pid."""
108 if IsWindows():
109 subprocess.call(['taskkill.exe', '/T', '/F', '/PID', str(pid)])
110 else:
111 os.kill(pid, signal.SIGTERM)
112
113
114 def RunCommand(cmd, cwd=None):
115 """Runs the given command and returns the exit code.
116
117 Args:
118 cmd: list of command arguments.
119 cwd: working directory to execute the command, or None if the current
120 working directory should be used.
121
122 Returns:
123 The exit code of the command.
124 """
125 process = subprocess.Popen(cmd, cwd=cwd)
126 process.wait()
127 return process.returncode
128
129
130 def DoesUrlExist(url):
131 """Determines whether a resource exists at the given URL.
132
133 Args:
134 url: URL to be verified.
135
136 Returns:
137 True if url exists, otherwise False.
138 """
139 parsed = urlparse.urlparse(url)
140 try:
141 conn = httplib.HTTPConnection(parsed.netloc)
142 conn.request('HEAD', parsed.path)
143 response = conn.getresponse()
144 except (socket.gaierror, socket.error):
145 return False
146 finally:
147 conn.close()
148 # Follow both permanent (301) and temporary (302) redirects.
149 if response.status == 302 or response.status == 301:
150 return DoesUrlExist(response.getheader('location'))
151 return response.status == 200
OLDNEW
« no previous file with comments | « chrome/test/pylib/common/unittest_util.py ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698