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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
// Copyright (c) 2010 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 "chrome/app/chrome_dll_resource.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/common/url_constants.h"
#include "chrome/test/automation/tab_proxy.h"
#include "chrome/test/ui/ui_layout_test.h"
#include "chrome/test/ui_test_utils.h"
#include "net/base/escape.h"
#include "net/test/test_server.h"
struct IsSearchProviderTestData;
class SearchProviderTest : public UITest {
protected:
SearchProviderTest();
IsSearchProviderTestData StartIsSearchProviderInstalledTest(
BrowserProxy* browser_proxy,
const char* host,
const char* expected_result);
void FinishIsSearchProviderInstalledTest(
const IsSearchProviderTestData& data);
net::TestServer test_server_;
private:
DISALLOW_COPY_AND_ASSIGN(SearchProviderTest);
};
SearchProviderTest::SearchProviderTest()
: test_server_(net::TestServer::TYPE_HTTP,
FilePath(FILE_PATH_LITERAL("chrome/test/data"))) {
// Enable the search provider additions.
launch_arguments_.AppendSwitch(switches::kEnableSearchProviderApiV2);
// Map all hosts to our local server.
std::string host_rule("MAP * " + test_server_.host_port_pair().ToString());
launch_arguments_.AppendSwitchASCII(switches::kHostRules, host_rule);
}
struct IsSearchProviderTestData {
IsSearchProviderTestData() {
}
IsSearchProviderTestData(TabProxy* t,
std::string h,
GURL url)
: tab(t),
host(h),
test_url(url) {
}
scoped_refptr<TabProxy> tab;
std::string host;
GURL test_url;
};
IsSearchProviderTestData SearchProviderTest::StartIsSearchProviderInstalledTest(
BrowserProxy* browser_proxy,
const char* host,
const char* expected_result) {
// Set-up a new tab for the navigation.
int num_tabs = 0;
if (!browser_proxy->GetTabCount(&num_tabs)) {
ADD_FAILURE() << "BrowserProxy::GetTabCount failed.";
return IsSearchProviderTestData();
}
GURL blank(chrome::kAboutBlankURL);
if (!browser_proxy->AppendTab(blank)) {
ADD_FAILURE() << "BrowserProxy::AppendTab failed.";
return IsSearchProviderTestData();
}
scoped_refptr<TabProxy> tab(browser_proxy->GetTab(num_tabs));
if (!tab.get()) {
ADD_FAILURE() << "BrowserProxy::GetTab for the new tab failed.";
return IsSearchProviderTestData();
}
// Go to the test page.
GURL local_url =
test_server_.GetURL("files/is_search_provider_installed.html");
GURL test_url(std::string("http://") + host + local_url.path() +
"#" + expected_result);
EXPECT_TRUE(tab->NavigateToURLAsync(test_url));
// Bundle up information needed to verify the result.
return IsSearchProviderTestData(tab, host, test_url);
}
void SearchProviderTest::FinishIsSearchProviderInstalledTest(
const IsSearchProviderTestData& data) {
ASSERT_TRUE(data.tab.get());
std::string cookie_name = data.host + "testResult";
std::string escaped_value =
WaitUntilCookieNonEmpty(data.tab, data.test_url,
cookie_name.c_str(), action_max_timeout_ms());
// Unescapes and normalizes the actual result.
std::string value = UnescapeURLComponent(
escaped_value,
UnescapeRule::NORMAL | UnescapeRule::SPACES |
UnescapeRule::URL_SPECIAL_CHARS | UnescapeRule::CONTROL_CHARS);
value += "\n";
ReplaceSubstringsAfterOffset(&value, 0, "\r", "");
EXPECT_STREQ("1\n", value.c_str());
}
// Flaky, http://crbug.com/57405.
TEST_F(SearchProviderTest, FLAKY_TestIsSearchProviderInstalled) {
ASSERT_TRUE(test_server_.Start());
// Use the default search provider, other installed search provider, and
// one not installed as well. (Note that yahoo isn't tested because the
// its host name varies a lot for different locales unlike Google and Bing,
// which would make the test fail depending on the machine's locale.)
const char* test_hosts[] = { "www.google.com",
"www.bing.com",
"localhost" };
const char* expected_results[] = { "2",
"1",
"0" };
COMPILE_ASSERT(arraysize(test_hosts) == arraysize(expected_results),
there_should_be_a_result_for_each_host);
IsSearchProviderTestData test_data[2 * arraysize(test_hosts)];
// Start results for the normal mode.
scoped_refptr<BrowserProxy> browser(automation()->GetBrowserWindow(0));
ASSERT_TRUE(browser.get());
for (size_t i = 0; i < arraysize(test_hosts); ++i) {
test_data[i] = StartIsSearchProviderInstalledTest(
browser, test_hosts[i], expected_results[i]);
}
// Start tests for incognito mode (and verify the result is 0).
ASSERT_TRUE(browser->RunCommand(IDC_NEW_INCOGNITO_WINDOW));
scoped_refptr<BrowserProxy> incognito(automation()->GetBrowserWindow(1));
ASSERT_TRUE(incognito.get());
for (size_t i = 0; i < arraysize(test_hosts); ++i) {
test_data[i + arraysize(test_hosts)] = StartIsSearchProviderInstalledTest(
incognito, test_hosts[i], "0");
}
// Do the verification.
for (size_t i = 0; i < arraysize(test_data); ++i) {
FinishIsSearchProviderInstalledTest(test_data[i]);
}
}
|
