commit e2518494afe2da9fc98a8011c17f94d708ff10ba
author David Haddock <dhaddock@chromium.org> Wed Jan 09 12:13:03 2019 -0800
committer David Haddock <dhaddock@chromium.org> Tue Jan 22 22:06:30 2019 +0000
tree 3a5c1d23176204a6ea77982be5fc779e2e1a61cc
parent 10963ac41abc0a5b3ba819be2f9a5c4a35d7893b

Wrap screenshot code in try/catch.

BUG=None
TEST=None

Change-Id: I3509b426cf7c35e47431f1f2ffc199b02dbc8c7e
Reviewed-on: https://chromium-review.googlesource.com/c/1403778
Reviewed-by: David Haddock <dhaddock@chromium.org>
Commit-Queue: David Haddock <dhaddock@chromium.org>
Tested-by: David Haddock <dhaddock@chromium.org>
Trybot-Ready: David Haddock <dhaddock@chromium.org>
(cherry picked from commit fa283898e8639cb07768300102bdeabecb7fa695)
Reviewed-on: https://chromium-review.googlesource.com/c/1427430

client/site_tests/cheets_AppCompatTest/cheets_AppCompatTest.py

diff --git a/client/site_tests/cheets_AppCompatTest/cheets_AppCompatTest.py b/client/site_tests/cheets_AppCompatTest/cheets_AppCompatTest.py
index 0cc3bcc..357c20b 100644
--- a/client/site_tests/cheets_AppCompatTest/cheets_AppCompatTest.py
+++ b/client/site_tests/cheets_AppCompatTest/cheets_AppCompatTest.py
@@ -186,6 +186,13 @@
                                           self._pkg_name)
 
 
+    def _take_screenshot(self, name):
+        try:
+            graphics_utils.take_screenshot(self.resultsdir, name)
+        except:
+            logging.info('Failed to take screenshot')
+
+
     def run_once(self, retries=3):
         self._increase_logcat_buffer()
         self._copy_resources_to_dut()
@@ -196,12 +203,10 @@
 
             # Bring Play Store to front.
             arc.adb_shell('am start %s' % self._PLAY_STORE_ACTIVITY)
-            graphics_utils.take_screenshot(self.resultsdir,
-                                           'test_start')
+            self._take_screenshot('test_start')
             self._start_test()
             logging.info('Iteration %d: Test finished' % trial)
-            graphics_utils.take_screenshot(self.resultsdir,
-                                           'test_end')
+            self._take_screenshot('test_end')
             self._get_app_version()
             self._capture_bugreport()
             self._grab_screenshots()