Add a JS API for detecting WebGL availability.

chrome/browser/extensions/extension_webstore_private_apitest.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 <vector>
 
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/stringprintf.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/extension_install_dialog.h"
 #include "chrome/browser/extensions/extension_install_ui.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/extension_webstore_private_api.h"
 #include "chrome/browser/extensions/webstore_installer.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_switches.h"
+#include "chrome/test/base/test_launcher_utils.h"
 #include "chrome/test/base/ui_test_utils.h"
+#include "content/browser/gpu/gpu_blacklist.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "net/base/mock_host_resolver.h"
+#include "ui/gfx/gl/gl_switches.h"
 
 namespace {
 
 class WebstoreInstallListener : public WebstoreInstaller::Delegate {
  public:
   WebstoreInstallListener()
       : received_failure_(false), received_success_(false), waiting_(false) {}
 
   void OnExtensionInstallSuccess(const std::string& id) OVERRIDE;
   void OnExtensionInstallFailure(const std::string& id,
@@ skipping 45 matching lines @@
 
 }  // namespace
 
 // A base class for tests below.
 class ExtensionWebstorePrivateApiTest : public ExtensionApiTest {
  public:
   void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
     ExtensionApiTest::SetUpCommandLine(command_line);
     command_line->AppendSwitchASCII(switches::kAppsGalleryURL,
         "http://www.example.com");
+
+    // In linux, we need to launch GPU process to decide if WebGL is allowed.
+    // Run it on top of osmesa to avoid bot driver issues.
+#if defined(OS_LINUX)
+    test_launcher_utils::OverrideGLImplementation(
+        command_line, gfx::kGLImplementationOSMesaName);
+#endif
   }
 
   void SetUpInProcessBrowserTestFixture() OVERRIDE {
     // Start up the test server and get us ready for calling the install
     // API functions.
     host_resolver()->AddRule("www.example.com", "127.0.0.1");
     ASSERT_TRUE(test_server()->Start());
     SetExtensionInstallDialogAutoConfirmForTests(true);
     ExtensionInstallUI::DisableFailureUIForTests();
   }
@@ skipping 123 matching lines @@
 IN_PROC_BROWSER_TEST_F(ExtensionWebstorePrivateApiTest,
                        IconUrl) {
   ASSERT_TRUE(RunInstallTest("icon_url.html", "extension.crx"));
 }
 
 // Tests using silentlyInstall to install extensions.
 IN_PROC_BROWSER_TEST_F(ExtensionWebstorePrivateBundleTest, SilentlyInstall) {
   WebstorePrivateApi::SetTrustTestIDsForTesting(true);
   ASSERT_TRUE(RunPageTest(GetTestServerURL("silently_install.html").spec()));
 }
+
+// Tests getWebGLStatus function when webgl is allowed.

[Mihai Parparita -not on Chrome, 2011/12/06 02:58:11]

Nit: Capitalize WebGL (applies below too).

[Zhenyao Mo, 2011/12/06 20:00:15]

Done.

+IN_PROC_BROWSER_TEST_F(ExtensionWebstorePrivateBundleTest,
+                       GetWebGLStatusAllowed) {
+  ASSERT_TRUE(
+      RunPageTest(GetTestServerURL("get_webgl_status_allowed.html").spec()));

[Mihai Parparita -not on Chrome, 2011/12/06 02:58:11]

For testing single function behavior, rather than using api tests and
RunPageTest, the preferred new way is to use RunFunctionAndReturnError to invoke
the ExtensionFunction implementation directly. See extension_tabs_test.cc or
socket_apitest.cc for samples of how it's used.

[Zhenyao Mo, 2011/12/06 20:00:15]

I tried, but it didn't work because RunFunctionAndReturnError or
RunFunctionAndReturnResult expect a sync function, so it expects HasResponse()
to be false, where in this function, HasResponse() is true.

[Zhenyao Mo, 2011/12/06 20:19:57]

Never mind.  It's a bug in my own code.  I switched to
RunFunctionAndReturnResult.

+}
+
+// Tests getWebGLStatus function when webgl is blacklisted.
+IN_PROC_BROWSER_TEST_F(ExtensionWebstorePrivateBundleTest,
+                       GetWebGLStatusBlocked) {
+  const std::string json_blacklist =
+      "{\n"
+      "  \"name\": \"gpu blacklist\",\n"
+      "  \"version\": \"1.0\",\n"
+      "  \"entries\": [\n"
+      "    {\n"
+      "      \"id\": 1,\n"
+      "      \"blacklist\": [\n"
+      "        \"webgl\"\n"
+      "      ]\n"
+      "    }\n"
+      "  ]\n"
+      "}";
+  scoped_ptr<Version> os_version(Version::GetVersionFromString("1.0"));
+  GpuBlacklist* blacklist = new GpuBlacklist("1.0");
+
+  ASSERT_TRUE(blacklist->LoadGpuBlacklist(
+      json_blacklist, GpuBlacklist::kAllOs));
+  GpuDataManager::GetInstance()->SetGpuBlacklist(blacklist);
+  GpuFeatureFlags flags = GpuDataManager::GetInstance()->GetGpuFeatureFlags();
+  EXPECT_EQ(
+      flags.flags(), static_cast<uint32>(GpuFeatureFlags::kGpuFeatureWebgl));
+
+  ASSERT_TRUE(
+      RunPageTest(GetTestServerURL("get_webgl_status_blocked.html").spec()));
+}
+