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