Add test simulating a renderer process crash after browser startup.

content/test/gpu/gpu_tests/gpu_integration_test_unittest.py

 # Copyright 2016 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import json
 import logging
 import mock
 import os
 import tempfile
 import unittest
@@ skipping 110 matching lines @@
     # This test causes the browser to try and restart the browser 3 times.
     yield ('restart', 'restart.html', ())
 
   def RunActualGpuTest(self, file_path, *args):
     # The logic of this test is run when the browser starts, it fails twice
     # and then succeeds on the third time so we are just testing that this
     # is successful based on the parameters.
     pass
 
 
+class BrowserCrashAfterStartIntegrationUnittest(
+  gpu_integration_test.GpuIntegrationTest):
+
+  _num_browser_crashes = 0
+  _num_browser_starts = 0
+
+  @classmethod
+  def setUpClass(cls):
+    cls._fake_browser_options = fakes.CreateBrowserFinderOptions(
+      execute_after_browser_creation=cls.CrashAfterStart)
+    cls._fake_browser_options.browser_options.platform = \
+        fakes.FakeLinuxPlatform()
+    cls._fake_browser_options.output_formats = ['none']
+    cls._fake_browser_options.suppress_gtest_report = True
+    cls._fake_browser_options.output_dir = None
+    cls._fake_browser_options .upload_bucket = 'public'
+    cls._fake_browser_options .upload_results = False
+    cls._finder_options = cls._fake_browser_options
+    cls.platform = None
+    cls.browser = None
+    cls.SetBrowserOptions(cls._finder_options)
+    cls.StartBrowser()
+
+  @classmethod
+  def _CreateExpectations(cls):
+    return gpu_test_expectations.GpuTestExpectations()
+
+  @classmethod
+  def CrashAfterStart(cls, browser):
+    cls._num_browser_starts += 1
+    if cls._num_browser_crashes < 2:
+      cls._num_browser_crashes += 1
+      # This simulates the first tab's renderer process crashing upon
+      # startup. The try/catch forces the GpuIntegrationTest's first
+      # fetch of this tab to fail. crbug.com/682819
+      try:
+        browser.tabs[0].Navigate('chrome://crash')
+      except Exception:
+        pass
+
+  @classmethod
+  def Name(cls):
+    return 'browser_crash_after_start_integration_unittest'
+
+  @classmethod
+  def GenerateGpuTests(cls, options):
+    # This test causes the browser to try and restart the browser 3 times.
+    yield ('restart', 'restart.html', ())
+
+  def RunActualGpuTest(self, file_path, *args):
+    # The logic of this test is run when the browser starts, it fails twice
+    # and then succeeds on the third time so we are just testing that this
+    # is successful based on the parameters.
+    pass
+
+
 class GpuIntegrationTestUnittest(unittest.TestCase):
   @mock.patch('telemetry.internal.util.binary_manager.InitDependencyManager')
   def testSimpleIntegrationUnittest(self, mockInitDependencyManager):
     self._RunIntegrationTest(
       'simple_integration_unittest', [
         'unexpected_error',
         'unexpected_failure'
       ], [
         'expected_failure',
         'expected_flaky',
       ])
     # It might be nice to be more precise about the order of operations
     # with these browser restarts, but this is at least a start.
     self.assertEquals(SimpleIntegrationUnittest._num_browser_starts, 6)
 
   @mock.patch('telemetry.internal.util.binary_manager.InitDependencyManager')
   def testIntegrationUnittestWithBrowserFailure(
       self, mockInitDependencyManager):
     self._RunIntegrationTest(
       'browser_start_failure_integration_unittest', [], ['restart'])
     self.assertEquals( \
       BrowserStartFailureIntegrationUnittest._num_browser_crashes, 2)
     self.assertEquals( \
       BrowserStartFailureIntegrationUnittest._num_browser_starts, 3)
 
-  # TODO(kbr): write a new test utilizing the
-  # execute_after_browser_creation argument to
-  # fakes.CreateBrowserFinderOptions once that is available.
-  # crbug.com/682819
+  @mock.patch('telemetry.internal.util.binary_manager.InitDependencyManager')
+  def testIntegrationUnittestWithBrowserCrashUponStart(
+      self, mockInitDependencyManager):
+    self._RunIntegrationTest(
+      'browser_crash_after_start_integration_unittest', [], ['restart'])
+    self.assertEquals( \
+      BrowserCrashAfterStartIntegrationUnittest._num_browser_crashes, 2)
+    self.assertEquals( \
+      BrowserCrashAfterStartIntegrationUnittest._num_browser_starts, 3)
 
   def _RunIntegrationTest(self, test_name, failures, successes):
     options = browser_test_runner.TestRunOptions()
     # Suppress printing out information for passing tests.
     options.verbosity = 0
     config = gpu_project_config.CONFIG
     temp_file = tempfile.NamedTemporaryFile(delete=False)
     temp_file.close()
     temp_file_name = temp_file.name
     try:
       browser_test_runner.Run(
           config, options,
           [test_name,
            '--write-abbreviated-json-results-to=%s' % temp_file_name])
       with open(temp_file_name) as f:
         test_result = json.load(f)
       self.assertEquals(test_result['failures'], failures)
       self.assertEquals(test_result['successes'], successes)
       self.assertEquals(test_result['valid'], True)
 
     finally:
       os.remove(temp_file_name)