| Index: chrome/browser/policy/url_blacklist_manager_unittest.cc
|
| diff --git a/chrome/browser/policy/url_blacklist_manager_unittest.cc b/chrome/browser/policy/url_blacklist_manager_unittest.cc
|
| index a1b72eb537b34d6eea150d9580155f9fe14ccf04..02debe137c148347e1b18dba7df8acd6297bd665 100644
|
| --- a/chrome/browser/policy/url_blacklist_manager_unittest.cc
|
| +++ b/chrome/browser/policy/url_blacklist_manager_unittest.cc
|
| @@ -8,10 +8,12 @@
|
|
|
| #include "base/basictypes.h"
|
| #include "base/message_loop/message_loop.h"
|
| +#include "base/message_loop/message_loop_proxy.h"
|
| #include "base/prefs/pref_registry_simple.h"
|
| #include "base/prefs/testing_pref_service.h"
|
| -#include "chrome/common/pref_names.h"
|
| -#include "content/public/test/test_browser_thread.h"
|
| +#include "chrome/browser/policy/policy_helpers.h"
|
| +#include "chrome/common/net/url_fixer_upper.h"
|
| +#include "components/policy/core/common/policy_pref_names.h"
|
| #include "google_apis/gaia/gaia_urls.h"
|
| #include "net/base/request_priority.h"
|
| #include "net/url_request/url_request.h"
|
| @@ -23,15 +25,20 @@ namespace policy {
|
|
|
| namespace {
|
|
|
| -using content::BrowserThread;
|
| +// Helper to get the disambiguated SegmentURL() function.
|
| +URLBlacklist::SegmentURLCallback GetSegmentURLCallback() {
|
| + return URLFixerUpper::SegmentURL;
|
| +}
|
|
|
| class TestingURLBlacklistManager : public URLBlacklistManager {
|
| public:
|
| explicit TestingURLBlacklistManager(PrefService* pref_service)
|
| - : URLBlacklistManager(pref_service),
|
| + : URLBlacklistManager(pref_service,
|
| + base::MessageLoopProxy::current(),
|
| + GetSegmentURLCallback(),
|
| + SkipBlacklistForURL),
|
| update_called_(0),
|
| - set_blacklist_called_(false) {
|
| - }
|
| + set_blacklist_called_(false) {}
|
|
|
| virtual ~TestingURLBlacklistManager() {
|
| }
|
| @@ -70,18 +77,12 @@ class TestingURLBlacklistManager : public URLBlacklistManager {
|
|
|
| class URLBlacklistManagerTest : public testing::Test {
|
| protected:
|
| - URLBlacklistManagerTest()
|
| - : loop_(base::MessageLoop::TYPE_IO),
|
| - ui_thread_(BrowserThread::UI, &loop_),
|
| - file_thread_(BrowserThread::FILE, &loop_),
|
| - io_thread_(BrowserThread::IO, &loop_) {
|
| - }
|
| + URLBlacklistManagerTest() : loop_(base::MessageLoop::TYPE_IO) {}
|
|
|
| virtual void SetUp() OVERRIDE {
|
| - pref_service_.registry()->RegisterListPref(prefs::kUrlBlacklist);
|
| - pref_service_.registry()->RegisterListPref(prefs::kUrlWhitelist);
|
| - blacklist_manager_.reset(
|
| - new TestingURLBlacklistManager(&pref_service_));
|
| + pref_service_.registry()->RegisterListPref(policy_prefs::kUrlBlacklist);
|
| + pref_service_.registry()->RegisterListPref(policy_prefs::kUrlWhitelist);
|
| + blacklist_manager_.reset(new TestingURLBlacklistManager(&pref_service_));
|
| loop_.RunUntilIdle();
|
| }
|
|
|
| @@ -97,13 +98,6 @@ class URLBlacklistManagerTest : public testing::Test {
|
| base::MessageLoop loop_;
|
| TestingPrefServiceSimple pref_service_;
|
| scoped_ptr<TestingURLBlacklistManager> blacklist_manager_;
|
| -
|
| - private:
|
| - content::TestBrowserThread ui_thread_;
|
| - content::TestBrowserThread file_thread_;
|
| - content::TestBrowserThread io_thread_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(URLBlacklistManagerTest);
|
| };
|
|
|
| // Parameters for the FilterToComponents test.
|
| @@ -171,8 +165,9 @@ TEST_P(URLBlacklistFilterToComponentsTest, FilterToComponents) {
|
| uint16 port = 42;
|
| std::string path;
|
|
|
| - URLBlacklist::FilterToComponents(GetParam().filter(), &scheme, &host,
|
| - &match_subdomains, &port, &path);
|
| + URLBlacklist::FilterToComponents(GetSegmentURLCallback(), GetParam().filter(),
|
| + &scheme, &host, &match_subdomains, &port,
|
| + &path);
|
| EXPECT_EQ(GetParam().scheme(), scheme);
|
| EXPECT_EQ(GetParam().host(), host);
|
| EXPECT_EQ(GetParam().match_subdomains(), match_subdomains);
|
| @@ -185,8 +180,8 @@ TEST_F(URLBlacklistManagerTest, SingleUpdateForTwoPrefChanges) {
|
| blacklist->Append(new StringValue("*.google.com"));
|
| ListValue* whitelist = new ListValue;
|
| whitelist->Append(new StringValue("mail.google.com"));
|
| - pref_service_.SetManagedPref(prefs::kUrlBlacklist, blacklist);
|
| - pref_service_.SetManagedPref(prefs::kUrlBlacklist, whitelist);
|
| + pref_service_.SetManagedPref(policy_prefs::kUrlBlacklist, blacklist);
|
| + pref_service_.SetManagedPref(policy_prefs::kUrlBlacklist, whitelist);
|
| loop_.RunUntilIdle();
|
|
|
| EXPECT_EQ(1, blacklist_manager_->update_called());
|
| @@ -310,7 +305,7 @@ INSTANTIATE_TEST_CASE_P(
|
| "/whatever")));
|
|
|
| TEST_F(URLBlacklistManagerTest, Filtering) {
|
| - URLBlacklist blacklist;
|
| + URLBlacklist blacklist(GetSegmentURLCallback());
|
|
|
| // Block domain and all subdomains, for any filtered scheme.
|
| scoped_ptr<base::ListValue> blocked(new base::ListValue);
|
| @@ -463,7 +458,7 @@ TEST_F(URLBlacklistManagerTest, Filtering) {
|
| }
|
|
|
| TEST_F(URLBlacklistManagerTest, BlockAllWithExceptions) {
|
| - URLBlacklist blacklist;
|
| + URLBlacklist blacklist(GetSegmentURLCallback());
|
|
|
| scoped_ptr<base::ListValue> blocked(new base::ListValue);
|
| scoped_ptr<base::ListValue> allowed(new base::ListValue);
|
| @@ -489,7 +484,7 @@ TEST_F(URLBlacklistManagerTest, BlockAllWithExceptions) {
|
| }
|
|
|
| TEST_F(URLBlacklistManagerTest, DontBlockResources) {
|
| - scoped_ptr<URLBlacklist> blacklist(new URLBlacklist());
|
| + scoped_ptr<URLBlacklist> blacklist(new URLBlacklist(GetSegmentURLCallback()));
|
| scoped_ptr<base::ListValue> blocked(new base::ListValue);
|
| blocked->Append(new base::StringValue("google.com"));
|
| blacklist->Block(blocked.get());
|
|
|