diff options
author | rsesek@chromium.org <rsesek@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2010-01-07 14:19:35 +0000 |
---|---|---|
committer | rsesek@chromium.org <rsesek@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98> | 2010-01-07 14:19:35 +0000 |
commit | f27fce5ea23a46ce60e213927e1b903b88c596ff (patch) | |
tree | c0987d1ee064da2969633b46e50398de13304814 /chrome/browser/cookies_tree_model_unittest.cc | |
parent | 415a01c0101f4323f2288f20a842dac54d1ac242 (diff) | |
download | chromium_src-f27fce5ea23a46ce60e213927e1b903b88c596ff.zip chromium_src-f27fce5ea23a46ce60e213927e1b903b88c596ff.tar.gz chromium_src-f27fce5ea23a46ce60e213927e1b903b88c596ff.tar.bz2 |
Revert 35703 - Move code duplicated in two tests up into the TestingProfile.
Move TestURLRequestContextGetter and friends from CookieTreeModelTest and
CookiesWindowControllerTest into TestingProfile. In the TestingProfile, return
a valid CookieMonster and URLRequstContextGetter.
BUG=None
TEST=Covered by unit tests
Review URL: http://codereview.chromium.org/525072
TBR=rsesek@chromium.org
Review URL: http://codereview.chromium.org/521056
git-svn-id: svn://svn.chromium.org/chrome/trunk/src@35704 0039d316-1c4b-4281-b951-d872f2087c98
Diffstat (limited to 'chrome/browser/cookies_tree_model_unittest.cc')
-rw-r--r-- | chrome/browser/cookies_tree_model_unittest.cc | 41 |
1 files changed, 38 insertions, 3 deletions
diff --git a/chrome/browser/cookies_tree_model_unittest.cc b/chrome/browser/cookies_tree_model_unittest.cc index 8ed45a5..317c003 100644 --- a/chrome/browser/cookies_tree_model_unittest.cc +++ b/chrome/browser/cookies_tree_model_unittest.cc @@ -1,4 +1,4 @@ -// Copyright (c) 2009-2010 The Chromium Authors. All rights reserved. +// Copyright (c) 2009 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. @@ -15,6 +15,41 @@ namespace { +class TestURLRequestContext : public URLRequestContext { + public: + TestURLRequestContext() { + cookie_store_ = new net::CookieMonster(); + } +}; + +class TestURLRequestContextGetter : public URLRequestContextGetter { + public: + virtual URLRequestContext* GetURLRequestContext() { + if (!context_) + context_ = new TestURLRequestContext(); + return context_.get(); + } + private: + scoped_refptr<URLRequestContext> context_; +}; + +class CookieTestingProfile : public TestingProfile { + public: + virtual URLRequestContextGetter* GetRequestContext() { + if (!url_request_context_getter_.get()) + url_request_context_getter_ = new TestURLRequestContextGetter; + return url_request_context_getter_.get(); + } + virtual ~CookieTestingProfile() {} + + net::CookieMonster* GetCookieMonster() { + return GetRequestContext()->GetCookieStore()->GetCookieMonster(); + } + + private: + scoped_refptr<URLRequestContextGetter> url_request_context_getter_; +}; + class CookiesTreeModelTest : public testing::Test { public: CookiesTreeModelTest() : io_thread_(ChromeThread::IO, &message_loop_) { @@ -24,7 +59,7 @@ class CookiesTreeModelTest : public testing::Test { } virtual void SetUp() { - profile_.reset(new TestingProfile()); + profile_.reset(new CookieTestingProfile()); } // Get the cookie names in the cookie list, as a comma seperated string. @@ -81,7 +116,7 @@ class CookiesTreeModelTest : public testing::Test { MessageLoop message_loop_; ChromeThread io_thread_; - scoped_ptr<TestingProfile> profile_; + scoped_ptr<CookieTestingProfile> profile_; }; TEST_F(CookiesTreeModelTest, RemoveAll) { |