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
|
// Copyright 2015 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/browser/ui/search_engines/search_engine_tab_helper.h"
#include <utility>
#include "base/macros.h"
#include "base/strings/stringprintf.h"
#include "chrome/browser/search_engines/template_url_service_factory.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "components/search_engines/template_url.h"
#include "components/search_engines/template_url_service.h"
#include "net/test/embedded_test_server/embedded_test_server.h"
#include "net/test/embedded_test_server/http_request.h"
#include "net/test/embedded_test_server/http_response.h"
using net::test_server::BasicHttpResponse;
using net::test_server::HttpRequest;
using net::test_server::HttpResponse;
namespace {
class TemplateURLServiceObserver {
public:
TemplateURLServiceObserver(TemplateURLService* service, base::RunLoop* loop)
: runner_(loop) {
DCHECK(loop);
template_url_sub_ = service->RegisterOnLoadedCallback(base::Bind(
&TemplateURLServiceObserver::StopLoop, base::Unretained(this)));
service->Load();
}
~TemplateURLServiceObserver() {}
private:
void StopLoop() { runner_->Quit(); }
base::RunLoop* runner_;
std::unique_ptr<TemplateURLService::Subscription> template_url_sub_;
DISALLOW_COPY_AND_ASSIGN(TemplateURLServiceObserver);
};
testing::AssertionResult VerifyTemplateURLServiceLoad(
TemplateURLService* service) {
if (service->loaded())
return testing::AssertionSuccess();
base::RunLoop runner;
TemplateURLServiceObserver observer(service, &runner);
runner.Run();
if (service->loaded())
return testing::AssertionSuccess();
return testing::AssertionFailure() << "TemplateURLService isn't loaded";
}
} // namespace
class SearchEngineTabHelperBrowserTest : public InProcessBrowserTest {
public:
SearchEngineTabHelperBrowserTest() {}
~SearchEngineTabHelperBrowserTest() override {}
private:
std::unique_ptr<HttpResponse> HandleRequest(const GURL& osdd_xml_url,
const HttpRequest& request) {
std::string html = base::StringPrintf(
"<html>"
"<head>"
" <link rel='search' type='application/opensearchdescription+xml'"
" href='%s'"
" title='ExampleSearch'>"
"</head>"
"</html>",
osdd_xml_url.spec().c_str());
std::unique_ptr<BasicHttpResponse> http_response(new BasicHttpResponse());
http_response->set_code(net::HTTP_OK);
http_response->set_content(html);
http_response->set_content_type("text/html");
return std::move(http_response);
}
// Starts a test server that serves a page pointing to a opensearch descriptor
// from a file:// url.
bool StartTestServer() {
GURL file_url = ui_test_utils::GetTestUrl(
base::FilePath(),
base::FilePath().AppendASCII("simple_open_search.xml"));
embedded_test_server()->RegisterRequestHandler(
base::Bind(&SearchEngineTabHelperBrowserTest::HandleRequest,
base::Unretained(this), file_url));
return embedded_test_server()->Start();
}
void SetUpOnMainThread() override { ASSERT_TRUE(StartTestServer()); }
DISALLOW_COPY_AND_ASSIGN(SearchEngineTabHelperBrowserTest);
};
IN_PROC_BROWSER_TEST_F(SearchEngineTabHelperBrowserTest,
IgnoreSearchDescriptionsFromFileURLs) {
TemplateURLService* url_service =
TemplateURLServiceFactory::GetForProfile(browser()->profile());
ASSERT_TRUE(url_service);
VerifyTemplateURLServiceLoad(url_service);
TemplateURLService::TemplateURLVector template_urls =
url_service->GetTemplateURLs();
// Navigate to a page with a search descriptor. Path doesn't matter as the
// test server always serves the same HTML.
GURL url(embedded_test_server()->GetURL("/"));
ui_test_utils::NavigateToURL(browser(), url);
// No new search engines should be added.
EXPECT_EQ(template_urls, url_service->GetTemplateURLs());
}
|
