OLD | NEW |
1 # Copyright 2015 The Chromium Authors. All rights reserved. | 1 # Copyright 2015 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import logging | 5 import logging |
6 import os | 6 import os |
7 import posixpath | 7 import posixpath |
8 import re | 8 import re |
9 import tempfile | |
10 import time | 9 import time |
11 | 10 |
12 from devil.android import device_errors | 11 from devil.android import device_errors |
13 from devil.android import flag_changer | 12 from devil.android import flag_changer |
14 from devil.android.sdk import shared_prefs | 13 from devil.android.sdk import shared_prefs |
15 from devil.utils import reraiser_thread | 14 from devil.utils import reraiser_thread |
16 from pylib import valgrind_tools | 15 from pylib import valgrind_tools |
17 from pylib.android import logdog_logcat_monitor | 16 from pylib.android import logdog_logcat_monitor |
18 from pylib.constants import host_paths | |
19 from pylib.base import base_test_result | 17 from pylib.base import base_test_result |
20 from pylib.instrumentation import instrumentation_test_instance | 18 from pylib.instrumentation import instrumentation_test_instance |
21 from pylib.local.device import local_device_environment | 19 from pylib.local.device import local_device_environment |
22 from pylib.local.device import local_device_test_run | 20 from pylib.local.device import local_device_test_run |
23 from pylib.utils import google_storage_helper | |
24 from pylib.utils import logdog_helper | 21 from pylib.utils import logdog_helper |
25 from py_trace_event import trace_event | 22 from py_trace_event import trace_event |
26 from py_utils import contextlib_ext | 23 from py_utils import contextlib_ext |
27 from py_utils import tempfile_ext | |
28 import tombstones | 24 import tombstones |
29 | 25 |
30 try: | |
31 from PIL import Image # pylint: disable=import-error | |
32 from PIL import ImageChops # pylint: disable=import-error | |
33 can_compute_diffs = True | |
34 except ImportError: | |
35 can_compute_diffs = False | |
36 | |
37 _TAG = 'test_runner_py' | 26 _TAG = 'test_runner_py' |
38 | 27 |
39 TIMEOUT_ANNOTATIONS = [ | 28 TIMEOUT_ANNOTATIONS = [ |
40 ('Manual', 10 * 60 * 60), | 29 ('Manual', 10 * 60 * 60), |
41 ('IntegrationTest', 30 * 60), | 30 ('IntegrationTest', 30 * 60), |
42 ('External', 10 * 60), | 31 ('External', 10 * 60), |
43 ('EnormousTest', 10 * 60), | 32 ('EnormousTest', 10 * 60), |
44 ('LargeTest', 5 * 60), | 33 ('LargeTest', 5 * 60), |
45 ('MediumTest', 3 * 60), | 34 ('MediumTest', 3 * 60), |
46 ('SmallTest', 1 * 60), | 35 ('SmallTest', 1 * 60), |
47 ] | 36 ] |
48 | 37 |
49 _RE_RENDER_IMAGE_NAME = re.compile( | |
50 r'(?P<test_class>\w+)\.' | |
51 r'(?P<description>\w+)\.' | |
52 r'(?P<device_model>\w+)\.' | |
53 r'(?P<orientation>port|land)\.png') | |
54 | |
55 RENDER_TESTS_RESULTS_DIR = { | |
56 'ChromePublicTest': 'chrome/test/data/android/render_tests' | |
57 } | |
58 | 38 |
59 # TODO(jbudorick): Make this private once the instrumentation test_runner is | 39 # TODO(jbudorick): Make this private once the instrumentation test_runner is |
60 # deprecated. | 40 # deprecated. |
61 def DidPackageCrashOnDevice(package_name, device): | 41 def DidPackageCrashOnDevice(package_name, device): |
62 # Dismiss any error dialogs. Limit the number in case we have an error | 42 # Dismiss any error dialogs. Limit the number in case we have an error |
63 # loop or we are failing to dismiss. | 43 # loop or we are failing to dismiss. |
64 try: | 44 try: |
65 for _ in xrange(10): | 45 for _ in xrange(10): |
66 package = device.DismissCrashDialogIfNeeded() | 46 package = device.DismissCrashDialogIfNeeded() |
67 if not package: | 47 if not package: |
(...skipping 266 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
334 # TODO(jbudorick): Make instrumentation tests output a JSON so this | 314 # TODO(jbudorick): Make instrumentation tests output a JSON so this |
335 # doesn't have to parse the output. | 315 # doesn't have to parse the output. |
336 result_code, result_bundle, statuses = ( | 316 result_code, result_bundle, statuses = ( |
337 self._test_instance.ParseAmInstrumentRawOutput(output)) | 317 self._test_instance.ParseAmInstrumentRawOutput(output)) |
338 results = self._test_instance.GenerateTestResults( | 318 results = self._test_instance.GenerateTestResults( |
339 result_code, result_bundle, statuses, start_ms, duration_ms) | 319 result_code, result_bundle, statuses, start_ms, duration_ms) |
340 for result in results: | 320 for result in results: |
341 if logcat_url: | 321 if logcat_url: |
342 result.SetLink('logcat', logcat_url) | 322 result.SetLink('logcat', logcat_url) |
343 | 323 |
344 self._ProcessRenderTestResults(device, results) | |
345 | |
346 # Update the result name if the test used flags. | 324 # Update the result name if the test used flags. |
347 if flags: | 325 if flags: |
348 for r in results: | 326 for r in results: |
349 if r.GetName() == test_name: | 327 if r.GetName() == test_name: |
350 r.SetName(test_display_name) | 328 r.SetName(test_display_name) |
351 | 329 |
352 # Add UNKNOWN results for any missing tests. | 330 # Add UNKNOWN results for any missing tests. |
353 iterable_test = test if isinstance(test, list) else [test] | 331 iterable_test = test if isinstance(test, list) else [test] |
354 test_names = set(self._GetUniqueTestName(t) for t in iterable_test) | 332 test_names = set(self._GetUniqueTestName(t) for t in iterable_test) |
355 results_names = set(r.GetName() for r in results) | 333 results_names = set(r.GetName() for r in results) |
(...skipping 11 matching lines...) Expand all Loading... |
367 # - optionally taking a screenshot | 345 # - optionally taking a screenshot |
368 # - logging the raw output at INFO level | 346 # - logging the raw output at INFO level |
369 # - clearing the application state while persisting permissions | 347 # - clearing the application state while persisting permissions |
370 if any(r.GetType() not in (base_test_result.ResultType.PASS, | 348 if any(r.GetType() not in (base_test_result.ResultType.PASS, |
371 base_test_result.ResultType.SKIP) | 349 base_test_result.ResultType.SKIP) |
372 for r in results): | 350 for r in results): |
373 if self._test_instance.screenshot_dir: | 351 if self._test_instance.screenshot_dir: |
374 file_name = '%s-%s.png' % ( | 352 file_name = '%s-%s.png' % ( |
375 test_display_name, | 353 test_display_name, |
376 time.strftime('%Y%m%dT%H%M%S', time.localtime())) | 354 time.strftime('%Y%m%dT%H%M%S', time.localtime())) |
377 screenshot_file = device.TakeScreenshot( | 355 saved_dir = device.TakeScreenshot( |
378 os.path.join(self._test_instance.screenshot_dir, file_name)) | 356 os.path.join(self._test_instance.screenshot_dir, file_name)) |
379 logging.info( | 357 logging.info( |
380 'Saved screenshot for %s to %s.', | 358 'Saved screenshot for %s to %s.', |
381 test_display_name, screenshot_file) | 359 test_display_name, saved_dir) |
382 if self._test_instance.should_save_images: | |
383 link = google_storage_helper.upload( | |
384 google_storage_helper.unique_name('screenshot', device=device), | |
385 screenshot_file, | |
386 bucket='chromium-render-tests') | |
387 for result in results: | |
388 result.SetLink('failure_screenshot', link) | |
389 | |
390 logging.info('detected failure in %s. raw output:', test_display_name) | 360 logging.info('detected failure in %s. raw output:', test_display_name) |
391 for l in output: | 361 for l in output: |
392 logging.info(' %s', l) | 362 logging.info(' %s', l) |
393 if (not self._env.skip_clear_data | 363 if (not self._env.skip_clear_data |
394 and self._test_instance.package_info): | 364 and self._test_instance.package_info): |
395 permissions = ( | 365 permissions = ( |
396 self._test_instance.apk_under_test.GetPermissions() | 366 self._test_instance.apk_under_test.GetPermissions() |
397 if self._test_instance.apk_under_test | 367 if self._test_instance.apk_under_test |
398 else None) | 368 else None) |
399 device.ClearApplicationState(self._test_instance.package_info.package, | 369 device.ClearApplicationState(self._test_instance.package_info.package, |
400 permissions=permissions) | 370 permissions=permissions) |
| 371 |
401 else: | 372 else: |
402 logging.debug('raw output from %s:', test_display_name) | 373 logging.debug('raw output from %s:', test_display_name) |
403 for l in output: | 374 for l in output: |
404 logging.debug(' %s', l) | 375 logging.debug(' %s', l) |
405 if self._test_instance.coverage_directory: | 376 if self._test_instance.coverage_directory: |
406 device.PullFile(coverage_directory, | 377 device.PullFile(coverage_directory, |
407 self._test_instance.coverage_directory) | 378 self._test_instance.coverage_directory) |
408 device.RunShellCommand('rm -f %s' % os.path.join(coverage_directory, | 379 device.RunShellCommand('rm -f %s' % os.path.join(coverage_directory, |
409 '*')) | 380 '*')) |
410 if self._test_instance.store_tombstones: | 381 if self._test_instance.store_tombstones: |
411 tombstones_url = None | 382 tombstones_url = None |
412 for result in results: | 383 for result in results: |
413 if result.GetType() == base_test_result.ResultType.CRASH: | 384 if result.GetType() == base_test_result.ResultType.CRASH: |
414 if not tombstones_url: | 385 if not tombstones_url: |
415 resolved_tombstones = tombstones.ResolveTombstones( | 386 resolved_tombstones = tombstones.ResolveTombstones( |
416 device, | 387 device, |
417 resolve_all_tombstones=True, | 388 resolve_all_tombstones=True, |
418 include_stack_symbols=False, | 389 include_stack_symbols=False, |
419 wipe_tombstones=True) | 390 wipe_tombstones=True) |
420 stream_name = 'tombstones_%s_%s' % ( | 391 stream_name = 'tombstones_%s_%s' % ( |
421 time.strftime('%Y%m%dT%H%M%S', time.localtime()), | 392 time.strftime('%Y%m%dT%H%M%S', time.localtime()), |
422 device.serial) | 393 device.serial) |
423 tombstones_url = logdog_helper.text( | 394 tombstones_url = logdog_helper.text( |
424 stream_name, resolved_tombstones) | 395 stream_name, resolved_tombstones) |
425 result.SetLink('tombstones', tombstones_url) | 396 result.SetLink('tombstones', tombstones_url) |
426 return results, None | 397 return results, None |
427 | 398 |
428 def _ProcessRenderTestResults(self, device, results): | |
429 render_results_dir = RENDER_TESTS_RESULTS_DIR.get(self._test_instance.suite) | |
430 if not render_results_dir: | |
431 return | |
432 | |
433 failure_images_device_dir = posixpath.join( | |
434 device.GetExternalStoragePath(), | |
435 'chromium_tests_root', render_results_dir, 'failures') | |
436 if not device.FileExists(failure_images_device_dir): | |
437 return | |
438 | |
439 if self._test_instance.should_save_images: | |
440 with tempfile_ext.NamedTemporaryDirectory() as temp_dir: | |
441 device.PullFile(failure_images_device_dir, temp_dir) | |
442 device.RemovePath(failure_images_device_dir, recursive=True) | |
443 | |
444 for failure_filename in os.listdir( | |
445 os.path.join(temp_dir, 'failures')): | |
446 | |
447 m = _RE_RENDER_IMAGE_NAME.match(failure_filename) | |
448 if not m: | |
449 logging.warning('Unexpected file in render test failures: %s', | |
450 failure_filename) | |
451 continue | |
452 | |
453 failure_filepath = os.path.join( | |
454 temp_dir, 'failures', failure_filename) | |
455 failure_link = google_storage_helper.upload( | |
456 google_storage_helper.unique_name( | |
457 failure_filename, device=device), | |
458 failure_filepath, | |
459 bucket='chromium-render-tests') | |
460 | |
461 golden_filepath = os.path.join( | |
462 host_paths.DIR_SOURCE_ROOT, render_results_dir, | |
463 failure_filename) | |
464 if not os.path.exists(golden_filepath): | |
465 logging.error('Cannot find golden image for %s', failure_filename) | |
466 continue | |
467 golden_link = google_storage_helper.upload( | |
468 google_storage_helper.unique_name( | |
469 failure_filename, device=device), | |
470 golden_filepath, | |
471 bucket='chromium-render-tests') | |
472 | |
473 if can_compute_diffs: | |
474 diff_filename = '_diff'.join( | |
475 os.path.splitext(failure_filename)) | |
476 diff_filepath = os.path.join(temp_dir, diff_filename) | |
477 (ImageChops.difference( | |
478 Image.open(failure_filepath), Image.open(golden_filepath)) | |
479 .convert('L') | |
480 .point(lambda i: 255 if i else 0) | |
481 .save(diff_filepath)) | |
482 diff_link = google_storage_helper.upload( | |
483 google_storage_helper.unique_name( | |
484 diff_filename, device=device), | |
485 diff_filepath, | |
486 bucket='chromium-render-tests') | |
487 else: | |
488 diff_link = '' | |
489 logging.error('Error importing PIL library. Image diffs for ' | |
490 'render test results will not be computed.') | |
491 | |
492 with tempfile.NamedTemporaryFile(suffix='.html') as temp_html: | |
493 temp_html.write(''' | |
494 <html> | |
495 <table> | |
496 <tr> | |
497 <th>Failure</th> | |
498 <th>Golden</th> | |
499 <th>Diff</th> | |
500 </tr> | |
501 <tr> | |
502 <td><img src="%s"/></td> | |
503 <td><img src="%s"/></td> | |
504 <td><img src="%s"/></td> | |
505 </tr> | |
506 </table> | |
507 </html> | |
508 ''' % (failure_link, golden_link, diff_link)) | |
509 html_results_link = google_storage_helper.upload( | |
510 google_storage_helper.unique_name( | |
511 'render_html', device=device), | |
512 temp_html.name, | |
513 bucket='chromium-render-tests') | |
514 for result in results: | |
515 result.SetLink(failure_filename, html_results_link) | |
516 | |
517 #override | 399 #override |
518 def _ShouldRetry(self, test): | 400 def _ShouldRetry(self, test): |
519 if 'RetryOnFailure' in test.get('annotations', {}): | 401 if 'RetryOnFailure' in test.get('annotations', {}): |
520 return True | 402 return True |
521 | 403 |
522 # TODO(jbudorick): Remove this log message once @RetryOnFailure has been | 404 # TODO(jbudorick): Remove this log message once @RetryOnFailure has been |
523 # enabled for a while. See crbug.com/619055 for more details. | 405 # enabled for a while. See crbug.com/619055 for more details. |
524 logging.error('Default retries are being phased out. crbug.com/619055') | 406 logging.error('Default retries are being phased out. crbug.com/619055') |
525 return False | 407 return False |
526 | 408 |
(...skipping 16 matching lines...) Expand all Loading... |
543 timeout = v | 425 timeout = v |
544 break | 426 break |
545 else: | 427 else: |
546 logging.warning('Using default 1 minute timeout for %s', test_name) | 428 logging.warning('Using default 1 minute timeout for %s', test_name) |
547 timeout = 60 | 429 timeout = 60 |
548 | 430 |
549 timeout *= cls._GetTimeoutScaleFromAnnotations(annotations) | 431 timeout *= cls._GetTimeoutScaleFromAnnotations(annotations) |
550 | 432 |
551 return timeout | 433 return timeout |
552 | 434 |
OLD | NEW |