1 // Copyright (c) 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/strings/stringprintf.h"
6 #include "chrome/browser/extensions/activity_log/activity_log.h"
7 #include "chrome/browser/extensions/extension_apitest.h"
8 #include "chrome/browser/extensions/extension_browsertest.h"
9 #include "chrome/browser/prerender/prerender_handle.h"
10 #include "chrome/browser/prerender/prerender_manager.h"
11 #include "chrome/browser/prerender/prerender_manager_factory.h"
12 #include "chrome/browser/ui/tabs/tab_strip_model.h"
13 #include "chrome/common/chrome_switches.h"
14 #include "chrome/test/base/ui_test_utils.h"
15 #include "content/public/browser/browser_context.h"
16 #include "content/public/browser/browser_thread.h"
17 #include "content/public/browser/notification_service.h"
18 #include "content/public/browser/storage_partition.h"
19 #include "content/public/test/browser_test_utils.h"
20 #include "net/dns/mock_host_resolver.h"
21 #include "net/test/embedded_test_server/embedded_test_server.h"
22
23 namespace extensions {
24
25 // Only the prerender tests are in this file. To add tests for activity
26 // logging please see:
27 // chrome/test/data/extensions/api_test/activity_log_private/README
28
29 class ActivityLogPrerenderTest : public ExtensionApiTest {
30 protected:
31 // Make sure the activity log is turned on.
SetUpCommandLine(CommandLine * command_line)32 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
33 ExtensionBrowserTest::SetUpCommandLine(command_line);
34 command_line->AppendSwitch(switches::kEnableExtensionActivityLogging);
35 command_line->AppendSwitchASCII(switches::kPrerenderMode,
36 switches::kPrerenderModeSwitchValueEnabled);
37 }
38
Prerender_Arguments(const std::string & extension_id,int port,scoped_ptr<std::vector<scoped_refptr<Action>>> i)39 static void Prerender_Arguments(
40 const std::string& extension_id,
41 int port,
42 scoped_ptr<std::vector<scoped_refptr<Action> > > i) {
43 // This is to exit RunLoop (base::MessageLoop::current()->Run()) below
44 base::MessageLoop::current()->PostTask(
45 FROM_HERE, base::MessageLoop::QuitClosure());
46
47 ASSERT_TRUE(i->size());
48 scoped_refptr<Action> last = i->front();
49
50 ASSERT_EQ(extension_id, last->extension_id());
51 ASSERT_EQ(Action::ACTION_CONTENT_SCRIPT, last->action_type());
52 ASSERT_EQ("[\"/google_cs.js\"]",
53 ActivityLogPolicy::Util::Serialize(last->args()));
54 ASSERT_EQ(
55 base::StringPrintf("http://www.google.com.bo:%d/test.html", port),
56 last->SerializePageUrl());
57 ASSERT_EQ(
58 base::StringPrintf("www.google.com.bo:%d/test.html", port),
59 last->page_title());
60 ASSERT_EQ("{\"prerender\":true}",
61 ActivityLogPolicy::Util::Serialize(last->other()));
62 ASSERT_EQ("", last->api_name());
63 ASSERT_EQ("", last->SerializeArgUrl());
64 }
65 };
66
IN_PROC_BROWSER_TEST_F(ActivityLogPrerenderTest,TestScriptInjected)67 IN_PROC_BROWSER_TEST_F(ActivityLogPrerenderTest, TestScriptInjected) {
68 host_resolver()->AddRule("*", "127.0.0.1");
69 ASSERT_TRUE(StartEmbeddedTestServer());
70 int port = embedded_test_server()->port();
71
72 // Get the extension (chrome/test/data/extensions/activity_log)
73 const Extension* ext =
74 LoadExtension(test_data_dir_.AppendASCII("activity_log"));
75 ASSERT_TRUE(ext);
76
77 ActivityLog* activity_log = ActivityLog::GetInstance(profile());
78 ASSERT_TRUE(activity_log);
79
80 //Disable rate limiting in PrerenderManager
81 prerender::PrerenderManager* prerender_manager =
82 prerender::PrerenderManagerFactory::GetForProfile(profile());
83 ASSERT_TRUE(prerender_manager);
84 prerender_manager->mutable_config().rate_limit_enabled = false;
85 // Increase prerenderer limits, otherwise this test fails
86 // on Windows XP.
87 prerender_manager->mutable_config().max_bytes = 1000 * 1024 * 1024;
88 prerender_manager->mutable_config().time_to_live =
89 base::TimeDelta::FromMinutes(10);
90 prerender_manager->mutable_config().abandon_time_to_live =
91 base::TimeDelta::FromMinutes(10);
92
93 content::WebContents* web_contents =
94 browser()->tab_strip_model()->GetActiveWebContents();
95 ASSERT_TRUE(web_contents);
96
97 content::WindowedNotificationObserver page_observer(
98 content::NOTIFICATION_LOAD_STOP,
99 content::NotificationService::AllSources());
100
101 GURL url(base::StringPrintf(
102 "http://www.google.com.bo:%d/test.html",
103 port));
104
105 if (!prerender_manager->cookie_store_loaded()) {
106 base::RunLoop loop;
107 prerender_manager->set_on_cookie_store_loaded_cb_for_testing(
108 loop.QuitClosure());
109 loop.Run();
110 }
111
112 const gfx::Size kSize(640, 480);
113 scoped_ptr<prerender::PrerenderHandle> prerender_handle(
114 prerender_manager->AddPrerenderFromLocalPredictor(
115 url,
116 web_contents->GetController().GetDefaultSessionStorageNamespace(),
117 kSize));
118
119 page_observer.Wait();
120
121 activity_log->GetFilteredActions(
122 ext->id(),
123 Action::ACTION_ANY,
124 "",
125 "",
126 "",
127 -1,
128 base::Bind(
129 ActivityLogPrerenderTest::Prerender_Arguments, ext->id(), port));
130
131 // Allow invocation of Prerender_Arguments
132 base::MessageLoop::current()->Run();
133 }
134
135 } // namespace extensions
136