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/public/test/content_browser_test.h"
12 #include "content/public/test/content_browser_test_utils.h"
13 #include "content/shell/browser/shell.h"
14
15 namespace content {
16
17 class RendererOverridesHandlerTest : public ContentBrowserTest {
18 protected:
SendCommand(const std::string & method,base::DictionaryValue * params)19 scoped_refptr<DevToolsProtocol::Response> SendCommand(
20 const std::string& method,
21 base::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,base::DictionaryValue * params)28 void SendAsyncCommand(const std::string& method,
29 base::DictionaryValue* params) {
30 scoped_ptr<RendererOverridesHandler> handler(CreateHandler());
31 scoped_refptr<DevToolsProtocol::Command> command(
32 DevToolsProtocol::CreateCommand(1, method, params));
33 scoped_refptr<DevToolsProtocol::Response> response =
34 handler->HandleCommand(command);
35 EXPECT_TRUE(response->is_async_promise());
36 base::MessageLoop::current()->Run();
37 }
38
HasValue(const std::string & path)39 bool HasValue(const std::string& path) {
40 base::Value* value = 0;
41 return result_->Get(path, &value);
42 }
43
HasListItem(const std::string & path_to_list,const std::string & name,const std::string & value)44 bool HasListItem(const std::string& path_to_list,
45 const std::string& name,
46 const std::string& value) {
47 base::ListValue* list;
48 if (!result_->GetList(path_to_list, &list))
49 return false;
50
51 for (size_t i = 0; i != list->GetSize(); i++) {
52 base::DictionaryValue* item;
53 if (!list->GetDictionary(i, &item))
54 return false;
55 std::string id;
56 if (!item->GetString(name, &id))
57 return false;
58 if (id == value)
59 return true;
60 }
61 return false;
62 }
63
64 scoped_ptr<base::DictionaryValue> result_;
65
66 private:
CreateHandler()67 RendererOverridesHandler* CreateHandler() {
68 RenderViewHost* rvh = shell()->web_contents()->GetRenderViewHost();
69 DevToolsAgentHost* agent = DevToolsAgentHost::GetOrCreateFor(rvh).get();
70 scoped_ptr<RendererOverridesHandler> handler(
71 new RendererOverridesHandler(agent));
72 handler->SetNotifier(base::Bind(
73 &RendererOverridesHandlerTest::OnMessageSent, base::Unretained(this)));
74 return handler.release();
75 }
76
OnMessageSent(const std::string & message)77 void OnMessageSent(const std::string& message) {
78 scoped_ptr<base::DictionaryValue> root(
79 static_cast<base::DictionaryValue*>(base::JSONReader::Read(message)));
80 base::DictionaryValue* result;
81 root->GetDictionary("result", &result);
82 result_.reset(result->DeepCopy());
83 base::MessageLoop::current()->QuitNow();
84 }
85 };
86
IN_PROC_BROWSER_TEST_F(RendererOverridesHandlerTest,QueryUsageAndQuota)87 IN_PROC_BROWSER_TEST_F(RendererOverridesHandlerTest, QueryUsageAndQuota) {
88 base::DictionaryValue* params = new base::DictionaryValue();
89 params->SetString("securityOrigin", "http://example.com");
90 SendAsyncCommand("Page.queryUsageAndQuota", params);
91
92 EXPECT_TRUE(HasValue("quota.persistent"));
93 EXPECT_TRUE(HasValue("quota.temporary"));
94 EXPECT_TRUE(HasListItem("usage.temporary", "id", "appcache"));
95 EXPECT_TRUE(HasListItem("usage.temporary", "id", "database"));
96 EXPECT_TRUE(HasListItem("usage.temporary", "id", "indexeddatabase"));
97 EXPECT_TRUE(HasListItem("usage.temporary", "id", "filesystem"));
98 EXPECT_TRUE(HasListItem("usage.persistent", "id", "filesystem"));
99 }
100
101 } // namespace content
102