1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/basictypes.h"
6 #include "base/json/json_reader.h"
7 #include "base/memory/scoped_ptr.h"
8 #include "content/browser/devtools/renderer_overrides_handler.h"
9 #include "content/public/browser/devtools_agent_host.h"
10 #include "content/public/browser/web_contents.h"
11 #include "content/shell/browser/shell.h"
12 #include "content/test/content_browser_test.h"
13 #include "content/test/content_browser_test_utils.h"
14
15 namespace content {
16
17 class RendererOverridesHandlerTest : public ContentBrowserTest {
18 protected:
SendCommand(const std::string & method,DictionaryValue * params)19 scoped_refptr<DevToolsProtocol::Response> SendCommand(
20 const std::string& method,
21 DictionaryValue* params) {
22 scoped_ptr<RendererOverridesHandler> handler(CreateHandler());
23 scoped_refptr<DevToolsProtocol::Command> command(
24 DevToolsProtocol::CreateCommand(1, method, params));
25 return handler->HandleCommand(command);
26 }
27
SendAsyncCommand(const std::string & method,DictionaryValue * params)28 void SendAsyncCommand(const std::string& method, DictionaryValue* params) {
29 scoped_ptr<RendererOverridesHandler> handler(CreateHandler());
30 scoped_refptr<DevToolsProtocol::Command> command(
31 DevToolsProtocol::CreateCommand(1, method, params));
32 scoped_refptr<DevToolsProtocol::Response> response =
33 handler->HandleCommand(command);
34 EXPECT_TRUE(response->is_async_promise());
35 base::MessageLoop::current()->Run();
36 }
37
HasValue(const std::string & path)38 bool HasValue(const std::string& path) {
39 base::Value* value = 0;
40 return result_->Get(path, &value);
41 }
42
HasListItem(const std::string & path_to_list,const std::string & name,const std::string & value)43 bool HasListItem(const std::string& path_to_list,
44 const std::string& name,
45 const std::string& value) {
46 base::ListValue* list;
47 if (!result_->GetList(path_to_list, &list))
48 return false;
49
50 for (size_t i = 0; i != list->GetSize(); i++) {
51 base::DictionaryValue* item;
52 if (!list->GetDictionary(i, &item))
53 return false;
54 std::string id;
55 if (!item->GetString(name, &id))
56 return false;
57 if (id == value)
58 return true;
59 }
60 return false;
61 }
62
63 scoped_ptr<base::DictionaryValue> result_;
64
65 private:
CreateHandler()66 RendererOverridesHandler* CreateHandler() {
67 RenderViewHost* rvh = shell()->web_contents()->GetRenderViewHost();
68 DevToolsAgentHost* agent = DevToolsAgentHost::GetOrCreateFor(rvh).get();
69 scoped_ptr<RendererOverridesHandler> handler(
70 new RendererOverridesHandler(agent));
71 handler->SetNotifier(base::Bind(
72 &RendererOverridesHandlerTest::OnMessageSent, base::Unretained(this)));
73 return handler.release();
74 }
75
OnMessageSent(const std::string & message)76 void OnMessageSent(const std::string& message) {
77 scoped_ptr<base::DictionaryValue> root(
78 static_cast<base::DictionaryValue*>(base::JSONReader::Read(message)));
79 base::DictionaryValue* result;
80 root->GetDictionary("result", &result);
81 result_.reset(result->DeepCopy());
82 base::MessageLoop::current()->QuitNow();
83 }
84 };
85
IN_PROC_BROWSER_TEST_F(RendererOverridesHandlerTest,QueryUsageAndQuota)86 IN_PROC_BROWSER_TEST_F(RendererOverridesHandlerTest, QueryUsageAndQuota) {
87 DictionaryValue* params = new DictionaryValue();
88 params->SetString("securityOrigin", "http://example.com");
89 SendAsyncCommand("Page.queryUsageAndQuota", params);
90
91 EXPECT_TRUE(HasValue("quota.persistent"));
92 EXPECT_TRUE(HasValue("quota.temporary"));
93 EXPECT_TRUE(HasListItem("usage.temporary", "id", "appcache"));
94 EXPECT_TRUE(HasListItem("usage.temporary", "id", "database"));
95 EXPECT_TRUE(HasListItem("usage.temporary", "id", "indexeddatabase"));
96 EXPECT_TRUE(HasListItem("usage.temporary", "id", "filesystem"));
97 EXPECT_TRUE(HasListItem("usage.persistent", "id", "filesystem"));
98 }
99
100 } // namespace content
101