OLD | NEW |
---|---|
(Empty) | |
1 #!/usr/bin/python | |
2 # Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
3 # Use of this source code is governed by a BSD-style license that can be | |
4 # found in the LICENSE file. | |
5 | |
6 import copy | |
7 import ctypes | |
8 from distutils import version | |
9 import fnmatch | |
10 import glob | |
11 import hashlib | |
12 import os | |
13 import platform | |
14 import random | |
15 import subprocess | |
16 import sys | |
17 import tarfile | |
18 import time | |
19 import urllib2 | |
20 | |
21 import pyauto_functional # Must be imported before pyauto. | |
22 import pyauto | |
23 import pyauto_utils | |
24 | |
25 class NaClSDKTest(pyauto.PyUITest): | |
26 """Tests for the NaCl SDK.""" | |
27 _EXTRACTED_NACL_SDK_DIR = 'extracted_nacl_sdk' | |
28 | |
29 def testNaClSDK(self): | |
30 """Verify that NaCl SDK is working properly.""" | |
31 if self._HasAllSystemRequirements() == False: | |
dennis_jeffrey
2011/08/08 18:00:53
if not self._HasAllSystemRequirements():
dennis_jeffrey
2011/08/08 18:00:53
This check is performed before all 3 of the tests
chrisphan
2011/08/08 23:14:39
As for Python 2.6, I don't see a way to skip a tes
chrisphan
2011/08/08 23:14:39
Done.
| |
32 return | |
dennis_jeffrey
2011/08/08 18:00:53
Before returning, maybe we can log a message indic
chrisphan
2011/08/08 22:46:44
Done.
| |
33 | |
34 self._VerifyNaClPlugin() | |
35 | |
36 self._VerifyDownloadLinks() | |
37 | |
38 self._VerifyNaClSDKInstaller() | |
39 | |
40 self._VerifyBuildStubProject() | |
41 | |
42 self._LaunchServerAndVerifyExamples() | |
43 | |
44 self._VerifyRebuildExamples() | |
45 | |
46 self._VerifySelldrAndNcval() | |
47 | |
48 self._RemoveDownloadedTestFile() | |
dennis_jeffrey
2011/08/08 18:00:53
I recommend removing the blank lines in-between ea
chrisphan
2011/08/08 22:46:44
Done.
| |
49 | |
50 def testVerifyNaClSDKChecksum(self): | |
51 """Verify NaCl SDK Checksum.""" | |
52 if self._HasAllSystemRequirements() == False: | |
53 return | |
54 | |
55 settings = self._GetTestSetting() | |
56 | |
57 self._DownloadNaClSDK() | |
58 | |
59 download_dir = os.path.join(self.DataDir(), 'downloads') | |
60 | |
61 if pyauto.PyUITest.IsWin(): | |
62 expected_shasum = settings['release_win_expected_shasum'] | |
63 file_path = os.path.join(download_dir, 'naclsdk_win.exe') | |
64 elif pyauto.PyUITest.IsMac(): | |
65 expected_shasum = settings['release_mac_expected_shasum'] | |
66 file_path = os.path.join(download_dir, 'naclsdk_mac.tgz') | |
67 elif pyauto.PyUITest.IsLinux(): | |
68 expected_shasum = settings['release_lin_expected_shasum'] | |
69 file_path = os.path.join(download_dir, 'naclsdk_linux.tgz') | |
dennis_jeffrey
2011/08/08 18:00:53
maybe add an "else:" in which we fail (just to be
chrisphan
2011/08/08 22:46:44
Done.
| |
70 | |
71 sha = hashlib.sha1() | |
72 try: | |
73 f = open(file_path, 'rb') | |
74 sha.update(f.read()) | |
75 shasum = sha.hexdigest() | |
76 self.assertEqual(expected_shasum, shasum, | |
77 msg='Unexpected checksum. Expected: %s, get: %s' | |
dennis_jeffrey
2011/08/08 18:00:53
nit: 'get' --> 'got'
chrisphan
2011/08/08 22:46:44
Done.
| |
78 % (expected_shasum, shasum)) | |
79 except IOError: | |
80 self.fail(msg='Cannot open %s.' % file_path) | |
81 finally: | |
82 f.close() | |
83 | |
84 def testVerifyPrereleaseGallery(self): | |
85 """Verify Pre-release gallery examples.""" | |
86 if self._HasAllSystemRequirements() == False: | |
87 return | |
88 settings = self._GetTestSetting() | |
89 examples = settings['prerelease_gallery'] | |
90 self._OpenExamplesAndStartTest(examples) | |
dennis_jeffrey
2011/08/08 18:00:53
We could combine the above 3 lines:
self._OpenExa
chrisphan
2011/08/08 22:46:44
Done.
| |
91 | |
92 def _VerifyNaClPlugin(self): | |
93 """Verify NaCl Plugin.""" | |
94 settings = self._GetTestSetting() | |
95 examples = settings['gallery_examples'] | |
96 self._OpenExamplesAndStartTest(examples) | |
dennis_jeffrey
2011/08/08 18:00:53
We could combine the above 3 lines like we did in
chrisphan
2011/08/08 22:46:44
Done.
| |
97 | |
98 def _VerifyDownloadLinks(self): | |
99 """Verify the download links.""" | |
100 settings = self._GetTestSetting() | |
101 sdk_download_url = settings['post_sdk_download_url'] | |
102 self.NavigateToURL(sdk_download_url) | |
dennis_jeffrey
2011/08/08 18:00:53
Combine the above 2 lines:
self.NavigateToURL(set
chrisphan
2011/08/08 22:46:44
Done.
| |
103 html = self.GetTabContents() | |
104 | |
105 # Make sure the correct URL is under the correct label. | |
106 if pyauto.PyUITest.IsWin(): | |
107 win_sdk_url = settings['post_win_sdk_url'] | |
108 win_url_index = html.find(win_sdk_url) | |
109 self.assertTrue(win_url_index > -1, | |
110 msg='Missing SDK download URL: %s' % win_sdk_url) | |
111 win_keyword_index = html.rfind('Windows', 0, win_url_index) | |
112 self.assertTrue(win_keyword_index > -1, | |
113 msg='Misplace download link: %s' % win_sdk_url) | |
dennis_jeffrey
2011/08/08 18:00:53
nit: 'Misplace' --> 'Misplaced". Same at lines 12
chrisphan
2011/08/08 22:46:44
Done.
| |
114 elif pyauto.PyUITest.IsMac(): | |
115 mac_sdk_url = settings['post_mac_sdk_url'] | |
116 mac_url_index = html.find(mac_sdk_url) | |
117 self.assertTrue(mac_url_index > -1, | |
118 msg='Missing SDK download URL: %s' % mac_sdk_url) | |
119 mac_keyword_index = html.rfind('Macintosh', 0, mac_url_index) | |
120 self.assertTrue(mac_keyword_index > -1, | |
121 msg='Misplace download link: %s' % mac_sdk_url) | |
122 elif pyauto.PyUITest.IsLinux(): | |
123 lin_sdk_url = settings['post_lin_sdk_url'] | |
124 lin_url_index = html.find(lin_sdk_url) | |
125 self.assertTrue(lin_url_index > -1, | |
126 msg='Missing SDK download URL: %s' % lin_sdk_url) | |
127 lin_keyword_index = html.rfind('Linux', 0, lin_url_index) | |
128 self.assertTrue(lin_keyword_index > -1, | |
129 msg='Misplace download link: %s' % lin_sdk_url) | |
dennis_jeffrey
2011/08/08 18:00:53
Maybe add an "else:" case in which we fail (just t
chrisphan
2011/08/08 22:46:44
Done.
| |
130 | |
131 def _VerifyNaClSDKInstaller(self): | |
132 """Verify NaCl SDK installer.""" | |
133 search_list = [ | |
134 'build.scons', | |
135 'favicon.ico', | |
136 'geturl/', | |
137 'hello_world/', | |
138 'hello_world_c/', | |
139 'httpd.py', | |
140 'index.html', | |
141 'nacl_sdk_scons/', | |
142 'pi_generator/', | |
143 'scons', | |
144 'sine_synth/' | |
145 ] | |
146 | |
147 mac_lin_additional_search_items = [ | |
148 'sel_ldr_x86_32', | |
149 'sel_ldr_x86_64', | |
150 'ncval_x86_32', | |
151 'ncval_x86_64' | |
152 ] | |
153 | |
154 win_additional_search_items = [ | |
155 'httpd.cmd', | |
156 'sel_ldr_x86_32.exe', | |
157 'sel_ldr_x86_64.exe', | |
158 'ncval_x86_32.exe', | |
159 'ncval_x86_64.exe' | |
160 ] | |
161 | |
162 download_dir = os.path.join(self.DataDir(), 'downloads') | |
163 | |
164 self._DownloadNaClSDK() | |
165 | |
166 if pyauto.PyUITest.IsWin(): | |
167 source_file = os.path.join(download_dir, 'naclsdk_win.exe') | |
168 self._SearchNaClSDKFileWindows( | |
169 search_list + win_additional_search_items, source_file) | |
170 elif pyauto.PyUITest.IsMac(): | |
171 source_file = os.path.join(download_dir, 'naclsdk_mac.tgz') | |
172 self._SearchNaClSDKTarFile(search_list + mac_lin_additional_search_items, | |
173 source_file) | |
174 elif pyauto.PyUITest.IsLinux(): | |
175 source_file = os.path.join(download_dir, 'naclsdk_linux.tgz') | |
176 self._SearchNaClSDKTarFile(search_list + mac_lin_additional_search_items, | |
177 source_file) | |
178 else: | |
179 self.fail(msg='NaCl SDK does not support this OS.') | |
180 | |
181 self._ExtractNaClSDK() | |
182 | |
183 def _VerifyBuildStubProject(self): | |
184 """Build stub project.""" | |
185 stub_project_files = [ | |
186 'build.scons', | |
187 'scons' | |
188 ] | |
189 | |
190 download_dir = os.path.join(self.DataDir(), 'downloads') | |
191 sdk_path = os.path.join(download_dir, self._EXTRACTED_NACL_SDK_DIR) | |
192 project_template_path = self._GetDirectoryPath('project_templates', | |
193 sdk_path) | |
194 examples_path = self._GetDirectoryPath('examples', sdk_path) | |
195 init_project_path = os.path.join(project_template_path, 'init_project.py') | |
196 | |
197 # Build a c project. | |
dennis_jeffrey
2011/08/08 18:00:53
nit: 'c' --> 'C'
chrisphan
2011/08/08 22:46:44
Done.
| |
198 proc = subprocess.Popen( | |
199 ['python', init_project_path, '-n', 'hello_c', '-c', '-d', | |
200 examples_path], stdout=subprocess.PIPE) | |
201 proc.communicate() | |
202 | |
203 hello_c_path = os.path.join(examples_path, 'hello_c') | |
204 for file in stub_project_files: | |
205 self.assertTrue(self._HasFile(file, hello_c_path), | |
206 msg='Cannot build hello_c stub project.') | |
207 | |
208 # Build a cc project. | |
dennis_jeffrey
2011/08/08 18:00:53
nit: 'cc' --> 'C++'
chrisphan
2011/08/08 22:46:44
Done.
| |
209 proc = subprocess.Popen( | |
210 ['python', init_project_path, '-n', 'hello_cc', '-c', '-d', | |
211 examples_path], stdout=subprocess.PIPE) | |
212 proc.communicate() | |
213 | |
214 hello_cc_path = os.path.join(examples_path, 'hello_cc') | |
215 for file in stub_project_files: | |
216 self.assertTrue(self._HasFile(file, hello_cc_path), | |
217 msg='Cannot build hello_cc stub project.') | |
dennis_jeffrey
2011/08/08 18:00:53
Lines 197-206 and lines 208-217 are almost identic
chrisphan
2011/08/08 22:46:44
Done.
| |
218 | |
219 def _LaunchServerAndVerifyExamples(self): | |
220 """Start local HTTP server and verify examples.""" | |
221 download_dir = os.path.join(self.DataDir(), 'downloads') | |
222 sdk_path = os.path.join(download_dir, self._EXTRACTED_NACL_SDK_DIR) | |
223 examples_path = self._GetDirectoryPath('examples', sdk_path) | |
224 | |
225 # Make sure server is not open. | |
226 if not self._IsURLAlive('http://localhost:5103'): | |
dennis_jeffrey
2011/08/08 18:00:53
Shouldn't we remove the "not" here?
chrisphan
2011/08/08 22:46:44
Done.
| |
227 self._CloseHTTPServer() | |
228 | |
229 # Start HTTP server. | |
230 if pyauto.PyUITest.IsWin(): | |
231 http_path = os.path.join(examples_path, 'httpd.cmd') | |
232 proc = subprocess.Popen([http_path], cwd=examples_path) | |
233 else: | |
234 http_path = os.path.join(examples_path, 'httpd.py') | |
235 proc = subprocess.Popen(['python', http_path], cwd=examples_path) | |
236 | |
237 self.NavigateToURL('http://localhost:5103') | |
dennis_jeffrey
2011/08/08 18:00:53
Are we sure the server has started by the time we
dennis_jeffrey
2011/08/08 18:00:53
Also, should we gracefully handle the case when po
chrisphan
2011/08/08 22:46:44
Done.
chrisphan
2011/08/08 22:46:44
Done.
| |
238 self.assertTrue(self._IsURLAlive('http://localhost:5103'), | |
239 msg='Cannot open HTTP server. %s' % | |
240 self.GetActiveTabTitle()) | |
241 | |
242 examples = { | |
243 'hello_world_c': 'http://localhost:5103/hello_world_c/' | |
244 'hello_world.html', | |
245 'hello_world': 'http://localhost:5103/hello_world/hello_world.html', | |
246 'geturl': 'http://localhost:5103/geturl/geturl.html', | |
247 'pi_generator': 'http://localhost:5103/pi_generator/pi_generator.html', | |
248 'sine_synth': 'http://localhost:5103/sine_synth/sine_synth.html', | |
249 } | |
250 try: | |
251 self._OpenExamplesAndStartTest(examples) | |
252 finally: | |
253 self._CloseHTTPServer(proc) | |
254 | |
255 def _VerifyRebuildExamples(self): | |
256 """Re-build the examples.""" | |
dennis_jeffrey
2011/08/08 18:00:53
'Re-build the examples and verify they are as expe
chrisphan
2011/08/08 22:46:44
Done.
| |
257 download_dir = os.path.join(self.DataDir(), 'downloads') | |
258 sdk_path = os.path.join(download_dir, self._EXTRACTED_NACL_SDK_DIR) | |
259 examples_path = self._GetDirectoryPath('examples', sdk_path) | |
260 scons_path = os.path.join(examples_path, 'scons -c') | |
261 | |
262 example_dirs = [ | |
263 'geturl', | |
264 'hello_world', | |
265 'hello_world_c', | |
266 'pi_generator', | |
267 'sine_synth' | |
268 ] | |
269 | |
270 proc = subprocess.Popen([scons_path], cwd=examples_path, shell=True) | |
271 proc.communicate() | |
272 for x in example_dirs: | |
273 ex_path = os.path.join(examples_path, x) | |
274 if self._HasFile('*.nmf', ex_path): | |
275 self.fail(msg='Failed running scons -c.') | |
276 | |
277 scons_path = os.path.join(examples_path, 'scons') | |
278 proc = subprocess.Popen([scons_path], cwd=examples_path, | |
279 stdout=subprocess.PIPE, shell=True) | |
280 proc.communicate() | |
281 | |
282 # Verify each example directory contains .nmf file. | |
283 for dir in example_dirs: | |
284 dir = os.path.join(examples_path, dir) | |
285 if not self._HasFile('*.nmf', dir): | |
286 self.fail(msg='Failed running scons.') | |
287 | |
288 self._LaunchServerAndVerifyExamples() | |
289 | |
290 def _VerifySelldrAndNcval(self): | |
291 """Verify sel_ldr and ncval.""" | |
292 download_dir = os.path.join(self.DataDir(), 'downloads') | |
293 sdk_path = os.path.join(download_dir, self._EXTRACTED_NACL_SDK_DIR) | |
294 examples_path = self._GetDirectoryPath('examples', sdk_path) | |
dennis_jeffrey
2011/08/08 18:00:53
A lot of the helper functions in this file start w
chrisphan
2011/08/08 23:14:39
Done.
| |
295 | |
296 architecture = self._GetPlatformArchitecture() | |
297 scons_arg = None | |
298 if pyauto.PyUITest.IsWin(): | |
299 if architecture == '64bit': | |
300 scons_arg = 'test64' | |
301 else: | |
302 scons_arg = 'test32' | |
303 elif pyauto.PyUITest.IsMac(): | |
304 scons_arg = 'test64' | |
305 elif pyauto.PyUITest.IsLinux(): | |
306 scons_arg = 'test64' | |
307 scons_path = os.path.join(examples_path, 'scons ' + scons_arg) | |
308 | |
309 # Build and run the unit test. | |
310 proc = subprocess.Popen([scons_path], stdout=subprocess.PIPE, | |
311 shell=True, cwd=examples_path) | |
312 stdout = proc.communicate()[0] | |
313 lines = stdout.splitlines() | |
314 test_ran = False | |
315 for line in lines: | |
316 if 'Check:' in line: | |
317 self.assertTrue('passed' in line, | |
318 msg='Nacl-sel_ldr unit test failed.') | |
319 test_ran = True | |
320 self.assertTrue(test_ran, | |
321 msg='Failed to build and run nacl-sel_ldr unit test.') | |
322 | |
323 if architecture == '64bit': | |
324 if not self._HasFileInTree('hello_world_x86_64.nexe', examples_path): | |
325 self.fail(msg='Missing file: hello_world_x86_64.nexe.') | |
326 else: | |
327 if not self._HasFileInTree('hello_world_x86_32.nexe', examples_path): | |
328 self.fail(msg='Missing file: hello_world_x86_32.nexe.') | |
329 | |
330 # Verify that a mismatch of sel_ldr and .nexe produces an error. | |
331 toolchain_path = self._GetDirectoryPath('toolchain', sdk_path) | |
332 bin_path = self._GetDirectoryPath('bin', toolchain_path) | |
333 hello_world_path = self._GetDirectoryPath('hello_world', examples_path) | |
334 sel_32_path = os.path.join(bin_path, 'sel_ldr_x86_32') | |
335 sel_64_path = os.path.join(bin_path, 'sel_ldr_x86_64') | |
336 nexe_32_path = os.path.join(hello_world_path, 'hello_world_x86_32.nexe') | |
337 nexe_64_path = os.path.join(hello_world_path, 'hello_world_x86_64.nexe') | |
338 | |
339 success = False | |
dennis_jeffrey
2011/08/08 18:00:53
We might be able to remove this line. I think the
chrisphan
2011/08/08 22:46:44
Done.
| |
340 if architecture == '64bit': | |
341 success = self._RunProcessAndCheckOutput( | |
342 [sel_64_path, nexe_32_path], 'Error while loading') | |
343 else: | |
344 success = self._RunProcessAndCheckOutput( | |
345 [sel_32_path, nexe_64_path], 'Error while loading') | |
346 self.assertTrue(success, | |
347 msg='Failed to verify sel_ldr and .nexe mismatch.') | |
348 | |
349 # Run the appropriate ncval for your platform on the matching .nexe. | |
dennis_jeffrey
2011/08/08 18:00:53
'your platform' --> 'the current platform'
chrisphan
2011/08/08 22:46:44
Done.
| |
350 ncval_32_path = os.path.join(bin_path, 'ncval_x86_32') | |
351 ncval_64_path = os.path.join(bin_path, 'ncval_x86_64') | |
352 success = False | |
dennis_jeffrey
2011/08/08 18:00:53
Same comment as line 339 above.
chrisphan
2011/08/08 22:46:44
Done.
| |
353 if architecture == '64bit': | |
354 success = self._RunProcessAndCheckOutput( | |
355 [ncval_64_path, nexe_64_path], 'is safe') | |
356 else: | |
357 success = self._RunProcessAndCheckOutput( | |
358 [ncval_32_path, nexe_32_path], 'is safe') | |
359 self.assertTrue(success, msg='Failed to verify ncval.') | |
360 | |
361 # Verify that a mismatch of ncval and .nexe produces an error. | |
362 success = False | |
dennis_jeffrey
2011/08/08 18:00:53
Same comment as line 339 above.
chrisphan
2011/08/08 22:46:44
Done.
| |
363 if architecture == '64bit': | |
364 success = self._RunProcessAndCheckOutput( | |
365 [ncval_64_path, nexe_32_path], 'is safe', is_in=False) | |
366 else: | |
367 success = self._RunProcessAndCheckOutput( | |
368 [ncval_32_path, nexe_64_path], 'is safe', is_in=False) | |
369 self.assertTrue(success, msg='Failed to verify ncval and .nexe mismatch.') | |
370 | |
371 def _RemoveDownloadedTestFile(self): | |
372 """Delete downloaded files and dirs from downloads directory.""" | |
373 download_dir = os.path.join(self.DataDir(), 'downloads') | |
374 extracted_sdk_path = os.path.join(download_dir, self._EXTRACTED_NACL_SDK_DIR ) | |
dennis_jeffrey
2011/08/08 18:00:53
This line is just a little bit too long.
chrisphan
2011/08/08 22:46:44
Done.
| |
375 pyauto_utils.RemovePath(extracted_sdk_path) | |
376 | |
377 sdk_path = os.path.join(download_dir, 'naclsdk_win.exe') | |
378 pyauto_utils.RemovePath(sdk_path) | |
379 sdk_path = os.path.join(download_dir, 'naclsdk_mac.tgz') | |
380 pyauto_utils.RemovePath(sdk_path) | |
381 sdk_path = os.path.join(download_dir, 'naclsdk_linux.tgz') | |
382 pyauto_utils.RemovePath(sdk_path) | |
dennis_jeffrey
2011/08/08 18:00:53
We can shorten the above 6 lines:
for sdk_path in
chrisphan
2011/08/08 22:46:44
Done.
| |
383 | |
384 def _RunProcessAndCheckOutput(self, args, look_for, is_in=True): | |
385 """Run process and look for string in output. | |
386 | |
387 Args: | |
388 args: Argument strings to pass to subprocess. | |
389 look_for: The string to search in output. | |
390 is_in: True if checking if param look_for is in output. | |
391 False if checking if param look_out is not in output. | |
dennis_jeffrey
2011/08/08 18:00:53
'look_out' --> 'look_for'
chrisphan
2011/08/08 22:46:44
Done.
| |
392 | |
393 Returns: | |
394 True, if output contains parameter look_for, or | |
dennis_jeffrey
2011/08/08 18:00:53
'True, if output contains parameter look_for,' -->
chrisphan
2011/08/08 22:46:44
Done.
| |
395 False otherwise. | |
396 """ | |
397 proc = subprocess.Popen(args, stdout=subprocess.PIPE, | |
398 stderr=subprocess.PIPE) | |
399 (stdout, stderr) = proc.communicate() | |
400 lines = stdout.splitlines() | |
401 for line in lines: | |
402 if look_for in line: | |
403 if is_in: | |
404 return True | |
405 else: | |
406 return False | |
dennis_jeffrey
2011/08/08 18:00:53
Change the above 4 lines to this:
return is_in
chrisphan
2011/08/08 22:46:44
Done.
| |
407 | |
408 lines = stderr.splitlines() | |
409 for line in lines: | |
410 if look_for in line: | |
411 if is_in: | |
412 return True | |
413 else: | |
414 return False | |
dennis_jeffrey
2011/08/08 18:00:53
Lines 400-406 and 408-414 are also very similar.
| |
415 if is_in: | |
416 return False | |
417 else: | |
418 return True | |
dennis_jeffrey
2011/08/08 18:00:53
Change the above 4 lines to this:
return not is_i
chrisphan
2011/08/08 22:46:44
Done.
| |
419 | |
420 def _OpenExamplesAndStartTest(self, examples): | |
421 """Open each example and verify that it's working. | |
422 | |
423 Args: | |
424 examples: A dict of name to url of examples. | |
425 """ | |
426 self._EnableNaClPlugin() | |
427 | |
428 # Open all examples. | |
429 for name, url in examples.items(): | |
430 self.AppendTab(pyauto.GURL(url)) | |
431 self._CheckForCrashes() | |
432 | |
433 # Verify all examples are working. | |
434 for name, url in examples.items(): | |
435 self._VerifyAnExample(name, url) | |
436 self._CheckForCrashes() | |
437 | |
438 # Reload all examples. | |
439 for _ in range(1): | |
dennis_jeffrey
2011/08/08 18:00:53
Doesn't this just execute once? Why is this loop
chrisphan
2011/08/08 22:46:44
Done.
| |
440 for tab_index in range(self.GetTabCount()): | |
441 self.GetBrowserWindow(0).GetTab(tab_index).Reload() | |
442 self._CheckForCrashes() | |
443 | |
444 # Verify all examples are working. | |
445 for name, url in examples.items(): | |
446 self._VerifyAnExample(name, url) | |
447 self._CheckForCrashes() | |
448 | |
449 # Randomly close a tab, check for crashes and verify all open | |
450 # examples operate correctly. | |
dennis_jeffrey
2011/08/08 18:00:53
Rather than randomly closing tabs, it might be bet
chrisphan
2011/08/08 22:46:44
Done.
| |
451 tab_count = self.GetTabCount() | |
452 for index in xrange(tab_count - 1, 0, -1): | |
453 tab_index = None | |
454 if index > 1: | |
455 tab_index = random.randint(1, index) | |
456 else: | |
457 tab_index = 1 | |
458 | |
459 self.GetBrowserWindow(0).GetTab(tab_index).Close(True) | |
460 self._CheckForCrashes() | |
461 | |
462 tabs = self.GetBrowserInfo()['windows'][0]['tabs'] | |
463 for tab in tabs: | |
464 if tab['index'] > 0: | |
465 for name, url in examples.items(): | |
466 if url == tab['url']: | |
467 self._VerifyAnExample(name, url) | |
468 break | |
469 | |
470 def _VerifyAnExample(self, name, url): | |
471 """Verify NaCl Example is working. | |
dennis_jeffrey
2011/08/08 18:00:53
nit: 'Example' --> 'example'
chrisphan
2011/08/08 22:46:44
Done.
| |
472 | |
473 Args: | |
474 name: A string name of the example. | |
475 url: A string url of the example. | |
476 """ | |
477 available_example_tests = { | |
478 'hello_world_c': self._VerifyHelloWorldExample, | |
479 'hello_world': self._VerifyHelloWorldExample, | |
480 'pi_generator': self._VerifyPiGeneratorExample, | |
481 'sine_synth': self._VerifySineSynthExample, | |
482 'geturl': self._VerifyGetURLExample, | |
483 'life': self._VerifyConwaysLifeExample | |
484 } | |
485 | |
486 if not name in available_example_tests: | |
487 self.fail(msg='No test available for %s.' % name) | |
488 | |
489 info = self.GetBrowserInfo() | |
490 tabs = info['windows'][0]['tabs'] | |
491 tab_index = None | |
492 for tab in tabs: | |
493 if url == tab['url']: | |
494 self.ActivateTab(tab['index']) | |
495 tab_index = tab['index'] | |
496 break | |
497 | |
498 if tab_index != None: | |
dennis_jeffrey
2011/08/08 18:00:53
I think this should work too:
if tab_index:
chrisphan
2011/08/08 23:14:39
Done.
| |
499 available_example_tests[name](tab_index, name, url) | |
500 | |
501 def _VerifyHelloWorldExample(self, tab_index, name, url): | |
502 """Verify Hello World Example. | |
503 | |
504 Args: | |
505 tab_index: Tab index integer that the example is on. | |
506 name: A string name of the example. | |
507 url: A string url of the example. | |
508 """ | |
509 wait_for = 60 | |
dennis_jeffrey
2011/08/08 18:00:53
Maybe no need to define this as a separate variabl
chrisphan
2011/08/08 22:46:44
Done.
| |
510 success = self.WaitUntil( | |
511 lambda: self.GetDOMValue( | |
512 'document.getElementById("statusField").innerHTML', | |
513 0, tab_index), | |
514 wait_for, expect_retval='SUCCESS') | |
dennis_jeffrey
2011/08/08 18:00:53
timeout=wait_for
chrisphan
2011/08/08 22:46:44
Done.
| |
515 self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url)) | |
516 | |
517 js_code = """ | |
518 window.alert = function(e) { | |
519 window.domAutomationController.send(String(e)); | |
520 } | |
521 window.domAutomationController.send("done"); | |
522 """ | |
523 self.ExecuteJavascript(js_code, 0, tab_index) | |
524 | |
525 result = self.ExecuteJavascript('document.helloForm.elements[1].click();', | |
526 0, tab_index) | |
527 self.assertEqual(result, '42', | |
528 msg='Example %s failed. URL: %s' % (name, url)) | |
529 | |
530 result = self.ExecuteJavascript('document.helloForm.elements[2].click();', | |
531 0, tab_index) | |
532 self.assertEqual(result, 'dlrow olleH', | |
533 msg='Example %s failed. URL: %s' % (name, url)) | |
534 | |
535 def _VerifyPiGeneratorExample(self, tab_index, name, url): | |
536 """Verify Pi Generator Example. | |
537 | |
538 Args: | |
539 tab_index: Tab index integer that the example is on. | |
540 name: A string name of the example. | |
541 url: A string url of the example. | |
542 """ | |
543 wait_for = 120 | |
544 success = self.WaitUntil( | |
545 lambda: self.GetDOMValue('document.form.pi.value', 0, tab_index)[0:3], | |
546 wait_for, expect_retval='3.1') | |
547 self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url)) | |
548 | |
549 # Get top corner of pi image. | |
dennis_jeffrey
2011/08/08 18:00:53
nit: 'pi' --> 'Pi' to make it consistent with line
chrisphan
2011/08/08 22:46:44
Done.
| |
550 js_code = """ | |
551 var obj = document.getElementById('piGenerator'); | |
552 var curleft = curtop = 0; | |
553 do { | |
554 curleft += obj.offsetLeft; | |
555 curtop += obj.offsetTop; | |
556 } while (obj = obj.offsetParent); | |
557 window.domAutomationController.send(curleft + "," + curtop); | |
558 """ | |
559 result = self.ExecuteJavascript(js_code, 0, tab_index) | |
560 result_split = result.split(",") | |
561 x = int(result_split[0]) | |
562 y = int(result_split[1]) | |
563 window = self.GetBrowserInfo()['windows'][0] | |
564 window_to_content_x = 2 | |
565 window_to_content_y = 80 | |
566 pi_image_x = x + window['x'] + window_to_content_x | |
567 pi_image_y = y + window['y'] + window_to_content_y | |
568 | |
569 if self._IsGetPixelSupported(): | |
570 is_animating = self._IsColorChanging(pi_image_x, pi_image_y, 50, 50) | |
571 self.assertTrue(is_animating, | |
572 msg='Example %s failed. URL: %s' % (name, url)) | |
573 | |
574 def _VerifySineSynthExample(self, tab_index, name, url): | |
575 """Verify Sine Wave Synthesizer Example. | |
576 | |
577 Args: | |
578 tab_index: Tab index integer that the example is on. | |
579 name: A string name of the example. | |
580 url: A string url of the example. | |
581 """ | |
582 success = self.WaitUntil( | |
583 lambda: self.GetDOMValue( | |
584 'document.getElementById("frequency_field").value', | |
585 0, tab_index), | |
586 timeout=30, expect_retval='440') | |
587 self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url)) | |
588 | |
589 self.ExecuteJavascript( | |
590 'document.body.getElementsByTagName("button")[0].click();' | |
591 'window.domAutomationController.send("done")', | |
592 0, tab_index) | |
593 | |
594 # TODO(chrisphan): Verify sound. | |
595 | |
596 def _VerifyGetURLExample(self, tab_index, name, url): | |
597 """Verify GetURL Example. | |
598 | |
599 Args: | |
600 tab_index: Tab index integer that the example is on. | |
601 name: A string name of the example. | |
602 url: A string url of the example. | |
603 """ | |
604 success = self.WaitUntil( | |
605 lambda: self.GetDOMValue( | |
606 'document.getElementById("status_field").innerHTML', | |
607 0, tab_index), | |
608 timeout=60, expect_retval='SUCCESS') | |
609 self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url)) | |
610 | |
611 self.ExecuteJavascript( | |
612 'document.geturl_form.elements[0].click();' | |
613 'window.domAutomationController.send("done")', | |
614 0, tab_index) | |
615 | |
616 js_code = """ | |
617 var output = document.getElementById("general_output").innerHTML; | |
618 var result; | |
619 if (output.indexOf("test passed") != -1) | |
620 result = "pass"; | |
621 else | |
622 result = "fail"; | |
623 window.domAutomationController.send(result); | |
624 """ | |
625 success = self.WaitUntil( | |
626 lambda: self.ExecuteJavascript(js_code, 0, tab_index), | |
627 timeout=30, expect_retval='pass') | |
628 self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url)) | |
629 | |
630 def _VerifyConwaysLifeExample(self, tab_index, name, url): | |
631 """Verify Conway's Life Example. | |
632 | |
633 Args: | |
634 tab_index: Tab index integer that the example is on. | |
635 name: A string name of the example. | |
636 url: A string url of the example. | |
637 """ | |
638 window = self.GetBrowserInfo()['windows'][0] | |
639 window_to_content_x = 2 | |
640 window_to_content_y = 80 | |
641 x = window['x'] + window_to_content_x | |
642 y = window['y'] + window_to_content_y | |
643 offset_pixel = 100 | |
644 if self._IsGetPixelSupported(): | |
645 wait_for = 30 | |
dennis_jeffrey
2011/08/08 18:00:53
Similar comment as line 509 above.
chrisphan
2011/08/08 22:46:44
Done.
| |
646 success = self.WaitUntil( | |
647 lambda: self._GetPixel(x + offset_pixel, y + offset_pixel), | |
648 wait_for, expect_retval=16777215) | |
dennis_jeffrey
2011/08/08 18:00:53
timeout=wait_for
chrisphan
2011/08/08 22:46:44
Done.
| |
649 self.assertTrue(success, msg='Example %s failed. URL: %s' % (name, url)) | |
650 | |
651 def _IsColorChanging(self, x, y, width, height, tries=3, retry_sleep=2): | |
652 """Check screen for anything that is moving. | |
653 | |
654 Args: | |
655 x: X coordinate on the screen. | |
656 y: Y coordinate on the screen. | |
657 width: Width of the area to look scan. | |
dennis_jeffrey
2011/08/08 18:00:53
Remove 'look'? If you do so, make the same change
chrisphan
2011/08/08 22:46:44
Done.
| |
658 height: Height of the area to look scan. | |
659 tries: Number of tries. | |
660 retry_sleep: Sleep time in between each tries. | |
dennis_jeffrey
2011/08/08 18:00:53
nits:
'in between' --> 'in-between'
'tries' --> 't
chrisphan
2011/08/08 22:46:44
Done.
| |
661 | |
662 Returns: | |
663 True, if pixel color in area is changing, or | |
664 False otherwise. | |
665 """ | |
666 color_a = self._GetAreaPixelColor(x, y, width, height) | |
667 for _ in xrange(tries): | |
668 time.sleep(retry_sleep) | |
669 color_b = self._GetAreaPixelColor(x, y, width, height) | |
670 if color_a != color_b: | |
671 return True | |
672 return False | |
673 | |
674 def _IsGetPixelSupported(self): | |
675 """Check if get pixel is supported. | |
676 | |
677 Returns: | |
678 True, if get pixel is supported, or | |
679 False otherwise. | |
680 """ | |
681 if pyauto.PyUITest.IsWin(): | |
682 return True | |
683 else: | |
684 return False | |
dennis_jeffrey
2011/08/08 18:00:53
Change the above 4 lines to this:
return pyauto.P
chrisphan
2011/08/08 22:46:44
Done.
| |
685 | |
686 def _GetAreaPixelColor(self, x, y, width, height): | |
687 """Get an area of pixel color and return a list. | |
dennis_jeffrey
2011/08/08 18:00:53
'a list' --> 'a list of color code values'
chrisphan
2011/08/08 23:14:39
Done.
| |
688 | |
689 Args: | |
690 x: X coordinate on the screen. | |
691 y: Y coordinate on the screen. | |
692 width: Width of the area to look scan. | |
693 height: Height of the area to look scan. | |
694 | |
695 Returns: | |
696 A list containing color codes. | |
697 """ | |
698 if pyauto.PyUITest.IsMac(): | |
699 pass # TODO(chrisphan): Do Mac. | |
700 elif pyauto.PyUITest.IsWin(): | |
701 return self._GetAreaPixelColorWin(x, y, width, height) | |
dennis_jeffrey
2011/08/08 18:00:53
As you implement these helpers for the other platf
chrisphan
2011/08/08 22:46:44
Done.
| |
702 elif pyauto.PyUITest.IsLinux(): | |
703 pass # TODO(chrisphan): Do Linux. | |
704 return None | |
705 | |
706 def _GetAreaPixelColorWin(self, x, y, width, height): | |
707 """Get an area of pixel color for Windows and return a list. | |
708 | |
709 Args: | |
710 x: X coordinate on the screen. | |
711 y: Y coordinate on the screen. | |
712 width: Width of the area to look scan. | |
713 height: Height of the area to look scan. | |
714 | |
715 Returns: | |
716 A list containing color codes. | |
717 """ | |
718 colors = [] | |
719 hdc = ctypes.windll.user32.GetDC(0) | |
720 for x_pos in xrange(x, x + width, 1): | |
721 for y_pos in xrange(y, y + height, 1): | |
722 color = ctypes.windll.gdi32.GetPixel(hdc, x_pos, y_pos) | |
723 colors.append(color) | |
724 return colors | |
725 | |
726 def _GetPixel(self, x, y): | |
727 """Get pixel color at coordinate x and y. | |
728 | |
729 Args: | |
730 x: X coordinate on the screen. | |
731 y: Y coordinate on the screen. | |
732 | |
733 Returns: | |
734 An integer color code. | |
735 """ | |
736 if pyauto.PyUITest.IsMac(): | |
737 pass # TODO(chrisphan): Do Mac. | |
738 elif pyauto.PyUITest.IsWin(): | |
739 return self._GetPixelWin(x, y) | |
dennis_jeffrey
2011/08/08 18:00:53
Similar comment as line 701 above.
chrisphan
2011/08/08 22:46:44
Done.
| |
740 elif pyauto.PyUITest.IsLinux(): | |
741 pass # TODO(chrisphan): Do Linux. | |
742 return None | |
743 | |
744 def _GetPixelWin(self, x, y): | |
745 """Get pixel color at coordinate x and y for Windows | |
746 | |
747 Args: | |
748 x: X coordinate on the screen. | |
749 y: Y coordinate on the screen. | |
750 | |
751 Returns: | |
752 An integer color code. | |
753 """ | |
754 hdc = ctypes.windll.user32.GetDC(0) | |
755 color = ctypes.windll.gdi32.GetPixel(hdc, x, y) | |
756 return color | |
757 | |
758 def _CheckForCrashes(self, last_action=None, last_action_param=None): | |
759 """Check for any browser/tab crashes and hangs. | |
760 | |
761 Args: | |
762 last_action: Specify action taken before checking for crashes. | |
763 last_action_param: Parameter for last action. | |
764 """ | |
765 self.assertTrue(self.GetBrowserWindowCount(), | |
766 msg='Browser crashed, no window is open.') | |
767 | |
768 info = self.GetBrowserInfo() | |
769 breakpad_folder = info['properties']['DIR_CRASH_DUMPS'] | |
770 old_dmp_files = glob.glob(os.path.join(breakpad_folder, '*.dmp')) | |
771 | |
772 # Verify there're no crash dump files | |
dennis_jeffrey
2011/08/08 18:00:53
nit: add period at end of sentence.
chrisphan
2011/08/08 22:46:44
Done.
| |
773 for dmp_file in glob.glob(os.path.join(breakpad_folder, '*.dmp')): | |
774 self.assertTrue(dmp_file in old_dmp_files, | |
775 msg='Crash dump %s found' % dmp_file) | |
dennis_jeffrey
2011/08/08 18:00:53
indent this line so that the parameter lines up un
chrisphan
2011/08/08 22:46:44
Done.
| |
776 | |
777 # Check for any crashed tabs. | |
778 tabs = info['windows'][0]['tabs'] | |
779 for tab in tabs: | |
780 if tab['url'] != 'about:blank': | |
781 if self.GetDOMValue('document.body.innerHTML', 0, tab['index']) == '': | |
dennis_jeffrey
2011/08/08 18:00:53
if not self.GetDOMValue('document.body.innerHTML',
chrisphan
2011/08/08 22:46:44
Done.
| |
782 self.fail(msg='Tab crashed on %s' % tab['url']) | |
783 | |
784 # TODO(chrisphan): check for tab hangs and browser hangs | |
785 # TODO(chrisphan): handle specific action: close browser, close tab | |
dennis_jeffrey
2011/08/08 18:00:53
nit: add periods at the end of the above 2 lines.
chrisphan
2011/08/08 22:46:44
Done.
| |
786 if last_action == 'close tab': | |
787 pass | |
788 elif last_action == 'close browser': | |
789 pass | |
dennis_jeffrey
2011/08/08 18:00:53
Maybe add an "else" case that gracefully handles a
chrisphan
2011/08/08 22:46:44
Done.
| |
790 | |
791 def _GetPlatformArchitecture(self): | |
792 """Get platform architecture. | |
793 | |
794 Args: | |
795 last_action: Specify action taken before checking for crashes. | |
dennis_jeffrey
2011/08/08 18:00:53
'Specify' --> 'Last'
chrisphan
2011/08/08 22:46:44
Done.
| |
796 last_action_param: Parameter for last action. | |
797 | |
798 Returns: | |
799 A string representing the platform architecture. | |
800 """ | |
801 if pyauto.PyUITest.IsWin(): | |
802 if os.environ['PROGRAMFILES'] == 'C:\\Program Files (x86)': | |
803 return '64bit' | |
804 else: | |
805 return '32bit' | |
806 elif pyauto.PyUITest.IsMac() or pyauto.PyUITest.IsLinux(): | |
807 if platform.machine() == 'x86_64': | |
808 return '64bit' | |
809 else: | |
810 return '32bit' | |
811 return '32bit' | |
812 | |
813 def _HasFile(self, pattern, root=os.curdir): | |
814 """Check if file exist in directory. | |
dennis_jeffrey
2011/08/08 18:00:53
'Check if a file matching the specified pattern ex
chrisphan
2011/08/08 22:46:44
Done.
| |
815 | |
816 Args: | |
817 pattern: Pattern of file name. | |
818 root: Directory to start looking. | |
819 | |
820 Returns: | |
821 True, if root contains the file name pattern, or | |
822 False otherwise. | |
823 """ | |
824 if len(glob.glob(os.path.join(root, pattern))) > 0: | |
825 return True | |
826 return False | |
dennis_jeffrey
2011/08/08 18:00:53
Change the above 3 lines to this:
return len(glob
chrisphan
2011/08/08 22:46:44
Done.
| |
827 | |
828 def _HasFileInTree(self, pattern, root=os.curdir): | |
829 """Check if file exist in directory. | |
830 | |
831 Args: | |
832 pattern: Pattern of file name. | |
833 root: Directory to start looking. | |
834 | |
835 Returns: | |
836 True, if root contains the file name pattern, or | |
837 False otherwise. | |
838 """ | |
dennis_jeffrey
2011/08/08 18:00:53
This docstring looks the same as for the function
chrisphan
2011/08/08 22:46:44
Done.
| |
839 for path, dirs, files in os.walk(os.path.abspath(root)): | |
840 if len(fnmatch.filter(files, pattern)) > 0: | |
841 return True | |
842 return False | |
843 | |
844 def _HasDir(self, pattern, root=os.curdir): | |
845 """Check if directory exist in another directory. | |
846 | |
847 Args: | |
848 pattern: Pattern of directory name. | |
849 root: Directory to start looking. | |
850 | |
851 Returns: | |
852 True, if root contains the directory name pattern, or | |
853 False otherwise. | |
854 """ | |
855 for path, dirs, files in os.walk(os.path.abspath(root)): | |
856 if len(fnmatch.filter(dirs, pattern)) > 0: | |
857 return True | |
858 return False | |
dennis_jeffrey
2011/08/08 18:00:53
This function is almost exactly the same as for th
chrisphan
2011/08/08 22:46:44
Done.
| |
859 | |
860 def _GetDirectoryPath(self, pattern, root=os.curdir): | |
861 """Get the path of a directory in another directory. | |
862 | |
863 Args: | |
864 pattern: Pattern of directory name. | |
865 root: Directory to start looking. | |
866 | |
867 Returns: | |
868 A string of the path. | |
869 """ | |
870 for path, dirs, files in os.walk(os.path.abspath(root)): | |
871 result = fnmatch.filter(dirs, pattern) | |
872 if len(result) > 0: | |
873 return os.path.join(path, result[0]) | |
874 return None | |
875 | |
876 def _HasAllSystemRequirements(self): | |
877 """Verify NaCl SDK Installation system requirements. | |
dennis_jeffrey
2011/08/08 18:00:53
nit: 'Installation' --> 'installation'
chrisphan
2011/08/08 22:46:44
Done.
| |
878 | |
879 Returns: | |
880 True, if system passed requirements, or | |
881 False otherwise. | |
882 """ | |
883 # Check python version. | |
884 if sys.version_info[0:2] < (2, 5): | |
885 return False | |
886 | |
887 # Check OS requirements. | |
888 if pyauto.PyUITest.IsMac(): | |
889 mac_min_version = version.StrictVersion('10.6') | |
890 mac_version = version.StrictVersion(platform.mac_ver()[0]) | |
891 if mac_version < mac_min_version: | |
892 return False | |
893 elif pyauto.PyUITest.IsWin(): | |
894 if not (self.IsWin7() or self.IsWinVista() or self.IsWinXP()): | |
895 return False | |
896 elif pyauto.PyUITest.IsLinux(): | |
897 pass # TODO(chrisphan): Check Lin requirements. | |
898 else: | |
899 return False | |
900 | |
901 # Check for Chrome version compatibility. | |
902 # Nacl supports Chrome 10 and higher builds. | |
dennis_jeffrey
2011/08/08 18:00:53
nit: 'Nacl' --> 'NaCl'
chrisphan
2011/08/08 22:46:44
Done.
| |
903 settings = self._GetTestSetting() | |
904 min_required_chrome_build = settings['min_required_chrome_build'] | |
905 browser_info = self.GetBrowserInfo() | |
906 chrome_version = browser_info['properties']['ChromeVersion'] | |
907 chrome_build = int(chrome_version.split('.')[0]) | |
908 if chrome_build < min_required_chrome_build: | |
909 return False | |
910 | |
911 return True | |
dennis_jeffrey
2011/08/08 18:00:53
Change lines 908-911 to this:
return chrome_build
| |
912 | |
913 def _DownloadNaClSDK(self): | |
914 """Download NaCl SDK.""" | |
915 settings = self._GetTestSetting() | |
916 win_sdk_url = settings['release_win_sdk_url'] | |
917 mac_sdk_url = settings['release_mac_sdk_url'] | |
918 lin_sdk_url = settings['release_lin_sdk_url'] | |
dennis_jeffrey
2011/08/08 18:00:53
Remove the above 3 lines, and just inline the valu
chrisphan
2011/08/08 22:46:44
Done.
| |
919 | |
920 download_dir = os.path.join(self.DataDir(), 'downloads') | |
921 | |
922 if pyauto.PyUITest.IsWin(): | |
923 dl_file = urllib2.urlopen(win_sdk_url) | |
924 file_path = os.path.join(download_dir, 'naclsdk_win.exe') | |
925 elif pyauto.PyUITest.IsMac(): | |
926 dl_file = urllib2.urlopen(mac_sdk_url) | |
927 file_path = os.path.join(download_dir, 'naclsdk_mac.tgz') | |
928 elif pyauto.PyUITest.IsLinux(): | |
929 dl_file = urllib2.urlopen(lin_sdk_url) | |
930 file_path = os.path.join(download_dir, 'naclsdk_linux.tgz') | |
931 else: | |
932 self.fail(msg='NaCl SDK does not support this OS.') | |
933 | |
934 try: | |
935 f = open(file_path, 'wb') | |
936 f.write(dl_file.read()) | |
937 except IOError: | |
938 self.fail(msg='Cannot open %s.' % file_path) | |
939 finally: | |
940 f.close() | |
941 | |
942 | |
dennis_jeffrey
2011/08/08 18:00:53
Remove 1 of these blank lines.
chrisphan
2011/08/08 22:46:44
Done.
| |
943 def _ExtractNaClSDK(self): | |
944 """Extract NaCl SDK.""" | |
945 download_dir = os.path.join(self.DataDir(), 'downloads') | |
946 destination = os.path.join(download_dir, self._EXTRACTED_NACL_SDK_DIR) | |
947 | |
948 pyauto_utils.RemovePath(destination) | |
949 if not os.path.exists(destination): | |
dennis_jeffrey
2011/08/08 18:00:53
If we remove the directory in line 948 above, why
chrisphan
2011/08/08 22:46:44
Done.
| |
950 os.makedirs(destination) | |
951 | |
952 if pyauto.PyUITest.IsWin(): | |
953 # Requires 7-Zip to extract self-install archive. | |
954 seven_z_file_path = self._Get7ZipPath() | |
955 self.assertNotEqual(seven_z_file_path, None, | |
956 msg='7-Zip is required but could not be found.') | |
dennis_jeffrey
2011/08/08 18:00:53
indent this line by 4 more spaces
chrisphan
2011/08/08 22:46:44
Done.
| |
957 | |
958 source_file = os.path.join(download_dir, 'naclsdk_win.exe') | |
959 proc = subprocess.Popen( | |
960 [seven_z_file_path, 'x', source_file, '-o' + destination], | |
961 stdout=subprocess.PIPE) | |
dennis_jeffrey
2011/08/08 18:00:53
Indent these 2 lines by only 4 spaces as compared
| |
962 proc.communicate() | |
963 elif pyauto.PyUITest.IsMac(): | |
964 source_file = os.path.join(download_dir, 'naclsdk_mac.tgz') | |
965 tar = tarfile.open(source_file, 'r') | |
966 tar.extractall(destination) | |
967 elif pyauto.PyUITest.IsLinux(): | |
968 source_file = os.path.join(download_dir, 'naclsdk_linux.tgz') | |
969 tar = tarfile.open(source_file, 'r') | |
970 tar.extractall(destination) | |
971 else: | |
972 self.fail(msg='NaCl SDK does not support this OS.') | |
973 | |
974 def _Get7ZipPath(self): | |
dennis_jeffrey
2011/08/08 18:00:53
If this function only applies to Windows, maybe we
chrisphan
2011/08/08 22:46:44
Done.
| |
975 """Check if 7-Zip is installed on Windows. | |
976 | |
977 Returns: | |
978 String path of 7-Zip. | |
979 None if not exist. | |
dennis_jeffrey
2011/08/08 18:00:53
'String path to the 7-Zip executable, or None if i
chrisphan
2011/08/08 22:46:44
Done.
| |
980 """ | |
981 current_dir = os.path.dirname(__file__) | |
982 seven_zip_path = os.path.join( | |
983 current_dir, os.pardir, os.pardir, | |
984 os.pardir, 'third_party', '7-Zip', '7z.exe') | |
985 | |
986 if os.path.isfile(seven_zip_path): | |
987 return seven_zip_path | |
988 | |
989 # Attempt to find 7-Zip in Program Files. | |
990 program_files_path = os.getenv('ProgramFiles') | |
991 if program_files_path != None: | |
992 seven_zip_path = os.path.join(program_files_path, '7-Zip', '7z.exe') | |
993 if os.path.isfile(seven_zip_path): | |
994 return seven_zip_path | |
995 program_files_path = os.getenv('ProgramW6432') | |
996 if program_files_path != None: | |
997 seven_zip_path = os.path.join(program_files_path, '7-Zip', '7z.exe') | |
998 if os.path.isfile(seven_zip_path): | |
999 return seven_zip_path | |
1000 | |
1001 return None | |
1002 | |
1003 def _IsURLAlive(self, url): | |
1004 """Test if URL is alive.""" | |
1005 try: | |
1006 urllib2.urlopen(url) | |
1007 except: | |
1008 return False | |
1009 return True | |
1010 | |
1011 def _CloseHTTPServer(self, proc=None): | |
1012 """Close HTTP server. | |
1013 | |
1014 Args: | |
1015 proc: Process that opened the HTTP server. | |
1016 """ | |
1017 if not self._IsURLAlive('http://localhost:5103'): | |
1018 return | |
1019 urllib2.urlopen('http://localhost:5103?quit=1') | |
1020 wait_for = 30 | |
1021 success = self.WaitUntil( | |
1022 lambda: self._IsURLAlive('http://localhost:5103'), | |
1023 timeout=wait_for, expect_retval=False) | |
1024 if not success: | |
1025 if proc == None: | |
1026 self.fail(msg='Fail to close HTTP server') | |
1027 else: | |
1028 if proc.poll() == None: | |
1029 try: | |
1030 proc.kill() | |
1031 except: | |
1032 pass | |
dennis_jeffrey
2011/08/08 18:00:53
If we get to this 'except' clause, does that mean
chrisphan
2011/08/08 22:46:44
Done.
| |
1033 | |
1034 def _SearchNaClSDKTarFile(self, search_list, source_file): | |
1035 """Search NaCl SDK tar file for example files and directories. | |
1036 | |
1037 Args: | |
1038 search_list: A list of strings, representing file and | |
1039 directory names for which to search. | |
1040 source_file: The string name of an NaCl SDK tar file. | |
1041 """ | |
1042 tar = tarfile.open(source_file, 'r') | |
1043 | |
1044 # Look for files and directories in examples. | |
1045 files = copy.deepcopy(search_list) | |
1046 for tar_info in tar: | |
1047 file_name = tar_info.name | |
1048 if tar_info.isdir() and not file_name.endswith('/'): | |
1049 file_name = file_name + '/' | |
1050 | |
1051 for name in files: | |
1052 if file_name.find('examples/' + name): | |
1053 files.remove(name) | |
1054 if len(files) == 0: | |
1055 break | |
1056 | |
1057 tar.close() | |
1058 | |
1059 self.assertEqual(len(files), 0, | |
1060 msg='Missing files or directories: %s' % | |
1061 ', '.join(map(str, files))) | |
1062 | |
1063 def _SearchNaClSDKFileWindows(self, search_list, source_file): | |
1064 """Search NaCl SDK file for example files and directories. | |
dennis_jeffrey
2011/08/08 18:00:53
add to this docstring: 'in Windows'
chrisphan
2011/08/08 22:46:44
Done.
| |
1065 | |
1066 Args: | |
1067 search_list: A list of strings, representing file and | |
1068 directory names for which to search. | |
1069 source_file: The string name of an NaCl SDK Windows | |
1070 self-install archive file. | |
1071 """ | |
1072 files = [] | |
1073 for i in xrange(len(search_list)): | |
1074 files.append(search_list[i].replace(r'/', '\\')) | |
1075 | |
1076 # Requires 7-Zip to look at self-install archive. | |
1077 seven_z_file_path = self._Get7ZipPath() | |
1078 self.assertNotEqual(seven_z_file_path, None, | |
1079 msg='7-Zip is required but could not be found.') | |
dennis_jeffrey
2011/08/08 18:00:53
indent this line more so that it lines up undernea
chrisphan
2011/08/08 23:14:39
Done.
| |
1080 | |
1081 proc = subprocess.Popen([seven_z_file_path, 'l', source_file], | |
1082 stdout=subprocess.PIPE) | |
1083 stdout = proc.communicate()[0] | |
1084 lines = stdout.splitlines() | |
1085 for x in lines: | |
1086 item_name = x.split(' ')[-1] | |
1087 for name in files: | |
1088 if item_name.find(name) > 0: | |
1089 files.remove(name) | |
1090 if len(files) == 0: | |
1091 break | |
1092 | |
1093 self.assertEqual(len(files), 0, | |
1094 msg='Missing files or directories: %s' % | |
1095 ', '.join(map(str, files))) | |
1096 | |
1097 def _EnableNaClPlugin(self): | |
1098 """"Enable NaCl Plugin.""" | |
dennis_jeffrey
2011/08/08 18:00:53
nit: 'Plugin' --> 'plugin'
chrisphan
2011/08/08 22:46:44
Done.
| |
1099 nacl_plugin = self.GetPluginsInfo().PluginForName('Chrome NaCl') | |
1100 if len(nacl_plugin) == 0: | |
dennis_jeffrey
2011/08/08 18:00:53
if not nacl_plugin:
| |
1101 nacl_plugin = self.GetPluginsInfo().PluginForName('Native Client') | |
1102 if len(nacl_plugin) == 0: | |
dennis_jeffrey
2011/08/08 18:00:53
if not nacl_plugin:
| |
1103 self.fail(msg='No NaCl Plugin found.') | |
dennis_jeffrey
2011/08/08 18:00:53
nit: 'Plugin' --> 'plugin'
chrisphan
2011/08/08 22:46:44
Done.
| |
1104 self.EnablePlugin(nacl_plugin[0]['path']) | |
1105 | |
1106 self.NavigateToURL('about:flags') | |
1107 | |
1108 js_code = """ | |
1109 chrome.send('enableFlagsExperiment', ['enable-nacl', 'true']); | |
1110 requestFlagsExperimentsData(); | |
1111 window.domAutomationController.send('done'); | |
1112 """ | |
1113 self.ExecuteJavascript(js_code) | |
1114 self.RestartBrowser(False) | |
1115 | |
1116 def _GetTestSetting(self): | |
1117 """Read the given data file and return a dictionary of items. | |
1118 | |
1119 Returns: | |
1120 A dict mapping of keys/values in the NaCl SDK setting file. | |
1121 """ | |
1122 data_file = os.path.join(self.DataDir(), 'nacl_sdk', 'nacl_sdk_setting') | |
1123 | |
1124 try: | |
1125 f = open(data_file, 'r') | |
1126 except IOError: | |
1127 self.fail(msg='Cannot open %s.' % data_file) | |
1128 | |
1129 try: | |
1130 dictionary = eval(f.read(), {'__builtins__': None}, None) | |
1131 except SyntaxError: | |
1132 self.fail(msg='%s is an invalid setting file.' % data_file) | |
1133 finally: | |
1134 f.close() | |
1135 | |
1136 return dictionary | |
1137 | |
1138 | |
1139 if __name__ == '__main__': | |
1140 pyauto_functional.Main() | |
OLD | NEW |