Make GTK and Aura parts orthogonal to OS parts

chrome/browser/chrome_browser_main_unittest.cc

 // Copyright (c) 2011 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.
 
 #include <string>
 #include <vector>
 #include "base/command_line.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/memory/scoped_vector.h"
 #include "chrome/browser/chrome_browser_main.h"
 #include "chrome/browser/chrome_content_browser_client.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/testing_pref_service.h"
 #include "content/common/main_function_params.h"
 #include "content/common/sandbox_init_wrapper.h"
 #include "content/public/browser/content_browser_client.h"
 #include "net/socket/client_socket_pool_base.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 class BrowserMainTest : public testing::Test {
  public:
   BrowserMainTest() : command_line_(CommandLine::NO_PROGRAM) {}
  protected:
   virtual void SetUp() {
     sandbox_init_wrapper_.reset(new SandboxInitWrapper());
   }
 
   scoped_ptr<SandboxInitWrapper> sandbox_init_wrapper_;
   TestingPrefService pref_service_;
   CommandLine command_line_;
 };
 
 TEST_F(BrowserMainTest, WarmConnectionFieldTrial_WarmestSocket) {
   command_line_.AppendSwitchASCII(switches::kSocketReusePolicy, "0");
 
   scoped_ptr<MainFunctionParams> params(
       new MainFunctionParams(command_line_, *sandbox_init_wrapper_, NULL));
-  scoped_ptr<content::BrowserMainParts> bw(
-      content::GetContentClient()->browser()->CreateBrowserMainParts(
-          *params));
-
-  ChromeBrowserMainParts* cbw = static_cast<ChromeBrowserMainParts*>(bw.get());
-  cbw->WarmConnectionFieldTrial();
-
-  EXPECT_EQ(0, net::GetSocketReusePolicy());
+  ScopedVector<content::BrowserMainParts> bwv;
+  content::GetContentClient()->browser()->CreateBrowserMainParts(
+      *params, &(bwv.get()));
+  ChromeBrowserMainParts* cbw = NULL;
+  if (bwv.size() >= 1)
+    cbw = static_cast<ChromeBrowserMainParts*>(bwv[0]);
+  EXPECT_TRUE(cbw);
+  if (cbw) {
+    cbw->WarmConnectionFieldTrial();
+    EXPECT_EQ(0, net::GetSocketReusePolicy());
+  }
 }
 
 TEST_F(BrowserMainTest, WarmConnectionFieldTrial_Random) {
   scoped_ptr<MainFunctionParams> params(
       new MainFunctionParams(command_line_, *sandbox_init_wrapper_, NULL));
-  scoped_ptr<content::BrowserMainParts> bw(
-      content::GetContentClient()->browser()->CreateBrowserMainParts(
-          *params));
-  ChromeBrowserMainParts* cbw = static_cast<ChromeBrowserMainParts*>(bw.get());
-
-  const int kNumRuns = 1000;
-  for (int i = 0; i < kNumRuns; i++) {
-    cbw->WarmConnectionFieldTrial();
-    int val = net::GetSocketReusePolicy();
-    EXPECT_LE(val, 2);
-    EXPECT_GE(val, 0);
+  ScopedVector<content::BrowserMainParts> bwv;
+  content::GetContentClient()->browser()->CreateBrowserMainParts(
+      *params, &(bwv.get()));
+  ChromeBrowserMainParts* cbw = NULL;
+  if (bwv.size() >= 1)
+    cbw = static_cast<ChromeBrowserMainParts*>(bwv[0]);
+  EXPECT_TRUE(cbw);
+  if (cbw) {
+    const int kNumRuns = 1000;
+    for (int i = 0; i < kNumRuns; i++) {
+      cbw->WarmConnectionFieldTrial();
+      int val = net::GetSocketReusePolicy();
+      EXPECT_LE(val, 2);
+      EXPECT_GE(val, 0);
+    }
   }
 }
 
 TEST_F(BrowserMainTest, WarmConnectionFieldTrial_Invalid) {
   command_line_.AppendSwitchASCII(switches::kSocketReusePolicy, "100");
 
   scoped_ptr<MainFunctionParams> params(
       new MainFunctionParams(command_line_, *sandbox_init_wrapper_, NULL));
   // This test ends up launching a new process, and that doesn't initialize the
   // ContentClient interfaces.
-  scoped_ptr<content::BrowserMainParts> bw;
+  ScopedVector<content::BrowserMainParts> bwv;
   if (content::GetContentClient()) {
-    bw.reset(content::GetContentClient()->browser()->CreateBrowserMainParts(
-        *params));
+    content::GetContentClient()->browser()->CreateBrowserMainParts(
+        *params, &(bwv.get()));
   } else {
     chrome::ChromeContentBrowserClient ccbc;
-    bw.reset(ccbc.CreateBrowserMainParts(*params));
+    ccbc.CreateBrowserMainParts(*params, &(bwv.get()));
   }
-  ChromeBrowserMainParts* cbw = static_cast<ChromeBrowserMainParts*>(bw.get());
+  ChromeBrowserMainParts* cbw = NULL;
+  if (bwv.size() >= 1)
+    cbw = static_cast<ChromeBrowserMainParts*>(bwv[0]);
+  EXPECT_TRUE(cbw);
+  if (cbw) {
 #if defined(NDEBUG) && defined(DCHECK_ALWAYS_ON)
-  EXPECT_DEATH(cbw->WarmConnectionFieldTrial(),
-                     "Not a valid socket reuse policy group");
+    EXPECT_DEATH(cbw->WarmConnectionFieldTrial(),
+                 "Not a valid socket reuse policy group");
 #else
-  EXPECT_DEBUG_DEATH(cbw->WarmConnectionFieldTrial(),
-                     "Not a valid socket reuse policy group");
+    EXPECT_DEBUG_DEATH(cbw->WarmConnectionFieldTrial(),
+                       "Not a valid socket reuse policy group");
 #endif
+  }
 }