Provide webkit_unit_tests environment via a TestSuite specialization.

Update the blink/renderer/controller/tests/run_all_tests.cc e.g. used by
webkit_unit_tests, to provide the expected test environment by deriving
from base::TestSuite and overriding Initialize() and Shutdown().

Bug: 918724
Change-Id: I711e29f70517ec793b08441b05fa82e33e461613
Reviewed-on: https://chromium-review.googlesource.com/c/1395194Reviewed-by: Daniel Cheng <dcheng@chromium.org>
Commit-Queue: Wez <wez@chromium.org>
Cr-Commit-Position: refs/heads/master@{#619857}

third_party/blink/renderer/controller/tests/run_all_tests.cc

@@ -41,12 +41,18 @@
 
 namespace {
 
-int runHelper(base::TestSuite* testSuite) {
-  content::SetUpBlinkTestEnvironment();
-  blink::SchemeRegistry::Initialize();
+class BlinkUnitTestSuite : public base::TestSuite {
+ public:
+  BlinkUnitTestSuite(int argc, char** argv) : base::TestSuite(argc, argv) {}
 
-  int result = testSuite->Run();
+ private:
+  void Initialize() override {
+    base::TestSuite::Initialize();
 
+    content::SetUpBlinkTestEnvironment();
+    blink::SchemeRegistry::Initialize();
+  }
+  void Shutdown() override {
     // Tickle EndOfTaskRunner which among other things will flush the queue
     // of error messages via V8Initializer::reportRejectedPromisesOnMainThread.
     base::ThreadTaskRunnerHandle::Get()->PostTask(FROM_HERE, base::DoNothing());
@@ -61,13 +67,17 @@ int runHelper(base::TestSuite* testSuite) {
 
     content::TearDownBlinkTestEnvironment();
 
-  return result;
-}
+    base::TestSuite::Shutdown();
+  }
+
+  DISALLOW_COPY_AND_ASSIGN(BlinkUnitTestSuite);
+};
 
 }  // namespace
 
 int main(int argc, char** argv) {
-  base::TestSuite testSuite(argc, argv);
+  BlinkUnitTestSuite test_suite(argc, argv);
   return base::LaunchUnitTests(
-      argc, argv, base::Bind(&runHelper, base::Unretained(&testSuite)));
+      argc, argv,
+      base::BindOnce(&BlinkUnitTestSuite::Run, base::Unretained(&test_suite)));
 }