Backend for DevTools quota managements.

content/browser/devtools/renderer_overrides_handler_browsertest.cc

+// Copyright 2013 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 "base/basictypes.h"
+#include "base/memory/scoped_ptr.h"
+#include "content/browser/devtools/renderer_overrides_handler.h"
+#include "content/public/browser/devtools_agent_host.h"
+#include "content/public/browser/web_contents.h"
+#include "content/shell/browser/shell.h"
+#include "content/test/content_browser_test.h"
+
+namespace content {
+
+class RendererOverridesHandlerTest : public ContentBrowserTest {
+ public:
+  std::string last_message_;
+
+  void OnMessageSent(const std::string& message) {
+    last_message_ = message;
+    base::MessageLoop::current()->QuitNow();
+  }
+
+  bool MessageContains(const std::string& substring) {
+    return last_message_.find(substring) != std::string::npos;
+  }
+};
+
+IN_PROC_BROWSER_TEST_F(RendererOverridesHandlerTest, QueryUsageAndQuota) {
+
+  RenderViewHost* rvh = shell()->web_contents()->GetRenderViewHost();
+  DevToolsAgentHost* agent_raw = DevToolsAgentHost::GetOrCreateFor(rvh).get();
+  scoped_ptr<RendererOverridesHandler> handler(
+      new RendererOverridesHandler(agent_raw));
+
+  handler->SetNotifier(base::Bind(
+      &RendererOverridesHandlerTest::OnMessageSent, base::Unretained(this)));
+
+  std::string error_response;
+  scoped_refptr<DevToolsProtocol::Command> command =
+      DevToolsProtocol::ParseCommand("{"
+              "\"id\": 1,"
+              "\"method\": \"Page.queryUsageAndQuota\","
+              "\"params\": {"
+                  "\"securityOrigin\": \"http://example.com\""
+              "}"
+          "}", &error_response);
+  ASSERT_TRUE(command.get());
+
+  scoped_refptr<DevToolsProtocol::Response> response =
+      handler->HandleCommand(command);
+
+  base::MessageLoop::current()->Run();
+
+  EXPECT_TRUE(MessageContains("\"availableSpace\""));
+  EXPECT_TRUE(MessageContains("\"persistentHostQuota\""));
+  EXPECT_TRUE(MessageContains("\"temporaryHostQuota\""));
+  EXPECT_TRUE(MessageContains("\"appcacheUsage\""));
+  EXPECT_TRUE(MessageContains("\"databaseUsage\""));
+  EXPECT_TRUE(MessageContains("\"indexedDatabaseUsage\""));
+  EXPECT_TRUE(MessageContains("\"persistentFileSystemUsage\""));
+  EXPECT_TRUE(MessageContains("\"syncableFileSystemUsage\""));
+  EXPECT_TRUE(MessageContains("\"temporaryFileSystemUsage\""));
+}
+
+}  // namespace content
+