Add GPU memory usage contents browser test

chrome/browser/automation/testing_automation_provider.cc

 // Copyright 2012 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 "chrome/browser/automation/testing_automation_provider.h"
 
 #include <map>
 #include <set>
 #include <string>
 #include <vector>
@@ skipping 110 matching lines @@
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/permissions/permission_set.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/render_messages.h"
 #include "content/public/browser/browser_child_process_host_iterator.h"
 #include "content/public/browser/child_process_data.h"
 #include "content/public/browser/favicon_status.h"
 #include "content/public/browser/geolocation.h"
+#include "content/public/browser/gpu_data_manager.h"
+#include "content/public/browser/gpu_data_manager_observer.h"
 #include "content/public/browser/interstitial_page.h"
 #include "content/public/browser/interstitial_page_delegate.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/plugin_service.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/render_widget_host_view.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/child_process_host.h"
@@ skipping 1824 matching lines @@
       &TestingAutomationProvider::KillRendererProcess;
 
   browser_handler_map_["LaunchApp"] = &TestingAutomationProvider::LaunchApp;
   browser_handler_map_["SetAppLaunchType"] =
       &TestingAutomationProvider::SetAppLaunchType;
 
   browser_handler_map_["GetV8HeapStats"] =
       &TestingAutomationProvider::GetV8HeapStats;
   browser_handler_map_["GetFPS"] =
       &TestingAutomationProvider::GetFPS;
+  browser_handler_map_["GetGpuMemoryUsage"] =
+      &TestingAutomationProvider::GetGpuMemoryUsage;
 
   browser_handler_map_["IsFullscreenForBrowser"] =
       &TestingAutomationProvider::IsFullscreenForBrowser;
   browser_handler_map_["IsFullscreenForTab"] =
       &TestingAutomationProvider::IsFullscreenForTab;
   browser_handler_map_["IsMouseLocked"] =
       &TestingAutomationProvider::IsMouseLocked;
   browser_handler_map_["IsMouseLockPermissionRequested"] =
       &TestingAutomationProvider::IsMouseLockPermissionRequested;
   browser_handler_map_["IsFullscreenPermissionRequested"] =
@@ skipping 3109 matching lines @@
   int routing_id = render_view->GetRoutingID();
 
   // This observer will delete itself.
   new FPSObserver(
       this, reply_message,
       base::GetProcId(render_view->GetProcess()->GetHandle()),
       routing_id);
   render_view->Send(new ChromeViewMsg_GetFPS(routing_id));
 }
 
+class TestingAutomationGpuDataManagerObserver
+    : public content::GpuDataManagerObserver {
+ public:
+  TestingAutomationGpuDataManagerObserver(
+      TestingAutomationProvider* provider,
+      IPC::Message* reply_message)
+      : provider_(provider),
+        reply_message_(reply_message) {
+    content::GpuDataManager::GetInstance()->AddObserver(this);
+  }
+
+  virtual ~TestingAutomationGpuDataManagerObserver() {
+    content::GpuDataManager::GetInstance()->RemoveObserver(this);
+  }
+
+  virtual void OnGpuInfoUpdate() OVERRIDE {}
+
+  virtual void OnVideoMemoryUsageStatsUpdate(
+      const content::GPUVideoMemoryUsageStats& video_memory_usage_stats)
+          OVERRIDE {
+    DictionaryValue dict;
+    dict.SetInteger("gpu_memory_mbytes_allocated", static_cast<int>(
+        video_memory_usage_stats.bytes_allocated/1048576));
+    AutomationJSONReply(provider_, reply_message_).SendSuccess(&dict);
+    delete this;
+  }
+
+ private:
+  TestingAutomationProvider* provider_;
+  IPC::Message* reply_message_;
+};
+
+void TestingAutomationProvider::GetGpuMemoryUsage(
+    Browser* browser,
+    DictionaryValue* args,
+    IPC::Message* reply_message) {
+  if (!content::GpuDataManager::GetInstance()->GpuAccessAllowed()) {
+    AutomationJSONReply(this, reply_message).SendError("Cannot access GPU");
+    return;
+  }
+  // This observer will delete itself.
+  new TestingAutomationGpuDataManagerObserver(this, reply_message);
+  content::GpuDataManager::GetInstance()->RequestVideoMemoryUsageStatsUpdate();
+}
+
 void TestingAutomationProvider::IsFullscreenForBrowser(Browser* browser,
     base::DictionaryValue* args,
     IPC::Message* reply_message) {
   DictionaryValue dict;
   dict.SetBoolean("result",
                   browser->fullscreen_controller()->IsFullscreenForBrowser());
   AutomationJSONReply(this, reply_message).SendSuccess(&dict);
 }
 
 void TestingAutomationProvider::IsFullscreenForTab(Browser* browser,
@@ skipping 1162 matching lines @@
   if (g_browser_process)
     g_browser_process->GetAutomationProviderList()->RemoveProvider(this);
 }
 
 void TestingAutomationProvider::EnsureTabSelected(Browser* browser,
                                                   WebContents* tab) {
   TabStripModel* tab_strip = browser->tab_strip_model();
   if (tab_strip->GetActiveWebContents() != tab)
     tab_strip->ActivateTabAt(tab_strip->GetIndexOfWebContents(tab), true);
 }