Clean up compositor initialization/destruction.

content/public/test/browser_test_base.cc

Index: content/public/test/browser_test_base.cc
diff --git a/content/public/test/browser_test_base.cc b/content/public/test/browser_test_base.cc
index 25d6330103946d9ed304bf68e87c2fe9efb9219d..ae04f505c4d5b22c4240d4cd5ad23411f9acd146 100644
--- a/content/public/test/browser_test_base.cc
+++ b/content/public/test/browser_test_base.cc
@@ -13,6 +13,9 @@
 #include "content/public/common/main_function_params.h"
 #include "content/public/test/test_utils.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
+#include "ui/compositor/compositor_switches.h"
+#include "ui/gl/gl_implementation.h"
+#include "ui/gl/gl_switches.h"
 
 #if defined(OS_POSIX)
 #include "base/process/process_handle.h"
@@ -29,6 +32,10 @@
 #include "content/public/browser/browser_thread.h"
 #endif
 
+#if defined(OS_CHROMEOS)
+#include "base/chromeos/chromeos_version.h"
+#endif
+
 namespace content {
 namespace {
 
@@ -64,7 +71,9 @@ extern int BrowserMain(const MainFunctionParams&);
 BrowserTestBase::BrowserTestBase()
     : embedded_test_server_(
         new net::test_server::EmbeddedTestServer(
-            BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO))) {
+            BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO))),
+      allow_test_contexts_(true),
+      allow_osmesa_(true) {
 #if defined(OS_MACOSX)
   base::mac::SetOverrideAmIBundled(true);
   base::PowerMonitorDeviceSource::AllocateSystemIOPorts();
@@ -98,6 +107,50 @@ void BrowserTestBase::SetUp() {
       new base::Closure(
           base::Bind(&BrowserTestBase::ProxyRunTestOnMainThreadLoop, this));
 
+#if defined(USE_AURA)
+  // Use test contexts for browser tests unless they override and force us to
+  // use a real context.
+  if (allow_test_contexts_)
+    command_line->AppendSwitch(switches::kTestCompositor);
+#endif
+
+  // When using real GL contexts, we usually use OSMesa as this works on all
+  // bots. The command line can override this behaviour to use a real GPU.
+  if (command_line->HasSwitch(switches::kUseGpuInTests))
+    allow_osmesa_ = false;
+
+  // Some bots pass this flag when they want to use a real GPU.
+  if (command_line->HasSwitch("enable-gpu"))
+    allow_osmesa_ = false;
+
+#if defined(OS_MACOSX)
+  // On Mac we always use a real GPU.
+  allow_osmesa_ = false;
+#endif
+
+#if defined(OS_ANDROID)
+  // On Android we always use a real GPU.
+  allow_osmesa_ = false;
+#endif
+
+#if defined(OS_CHROMEOS)
+  // If the test is running on the chromeos envrionment (such as
+  // device or vm bots), the compositor will use real GL contexts, and
+  // we should use real GL bindings with it.
+  if (base::chromeos::IsRunningOnChromeOS())
+    allow_osmesa_ = false;
+#endif
+
+  if (command_line->HasSwitch(switches::kUseGL)) {
+    NOTREACHED() <<
+        "kUseGL should not be used with tests. Try kUseGpuInTests instead.";
+  }
+
+  if (allow_osmesa_) {
+    command_line->AppendSwitchASCII(
+        switches::kUseGL, gfx::kGLImplementationOSMesaName);
+  }
+
   SetUpInProcessBrowserTestFixture();
 #if defined(OS_ANDROID)
   BrowserMainRunner::Create()->Initialize(params);