Index: chrome/browser/browsing_data/browsing_data_channel_id_helper_unittest.cc |
diff --git a/chrome/browser/browsing_data/browsing_data_channel_id_helper_unittest.cc b/chrome/browser/browsing_data/browsing_data_channel_id_helper_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..3ce192c1526f007df46e1c93710ddb06cab2cf18 |
--- /dev/null |
+++ b/chrome/browser/browsing_data/browsing_data_channel_id_helper_unittest.cc |
@@ -0,0 +1,144 @@ |
+// Copyright (c) 2012 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/browsing_data/browsing_data_channel_id_helper.h" |
+ |
+#include "base/bind.h" |
+#include "base/run_loop.h" |
+#include "chrome/test/base/testing_profile.h" |
+#include "content/public/browser/browser_thread.h" |
+#include "content/public/test/test_browser_thread_bundle.h" |
+#include "net/ssl/channel_id_service.h" |
+#include "net/url_request/url_request_context.h" |
+#include "net/url_request/url_request_context_getter.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+using content::BrowserThread; |
+ |
+class BrowsingDataChannelIDHelperTest |
+ : public testing::Test, |
+ public net::SSLConfigService::Observer { |
+ public: |
+ BrowsingDataChannelIDHelperTest() : ssl_config_changed_count_(0) { |
+ } |
+ |
+ virtual void SetUp() OVERRIDE { |
+ testing_profile_.reset(new TestingProfile()); |
+ |
+ testing_profile_->GetSSLConfigService()->AddObserver(this); |
+ } |
+ |
+ virtual void TearDown() OVERRIDE { |
+ testing_profile_->GetSSLConfigService()->RemoveObserver(this); |
+ } |
+ |
+ void CreateChannelIDsForTest() { |
+ net::URLRequestContext* context = |
+ testing_profile_->GetRequestContext()->GetURLRequestContext(); |
+ net::ChannelIDStore* channel_id_store = |
+ context->channel_id_service()->GetChannelIDStore(); |
+ channel_id_store->SetChannelID("https://www.google.com:443", |
+ base::Time(), base::Time(), |
+ "key", "cert"); |
+ channel_id_store->SetChannelID("https://www.youtube.com:443", |
+ base::Time(), base::Time(), |
+ "key", "cert"); |
+ } |
+ |
+ void FetchCallback( |
+ const net::ChannelIDStore::ChannelIDList& channel_ids) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
+ channel_id_list_ = channel_ids; |
+ } |
+ |
+ // net::SSLConfigService::Observer implementation: |
+ virtual void OnSSLConfigChanged() OVERRIDE { |
+ ssl_config_changed_count_++; |
+ } |
+ |
+ protected: |
+ content::TestBrowserThreadBundle thread_bundle_; |
+ scoped_ptr<TestingProfile> testing_profile_; |
+ |
+ net::ChannelIDStore::ChannelIDList channel_id_list_; |
+ |
+ int ssl_config_changed_count_; |
+}; |
+ |
+TEST_F(BrowsingDataChannelIDHelperTest, FetchData) { |
+ CreateChannelIDsForTest(); |
+ scoped_refptr<BrowsingDataChannelIDHelper> helper( |
+ BrowsingDataChannelIDHelper::Create(testing_profile_.get())); |
+ |
+ helper->StartFetching( |
+ base::Bind(&BrowsingDataChannelIDHelperTest::FetchCallback, |
+ base::Unretained(this))); |
+ |
+ // Blocks until BrowsingDataChannelIDHelperTest::FetchCallback is |
+ // notified. |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ ASSERT_EQ(2UL, channel_id_list_.size()); |
+ net::ChannelIDStore::ChannelIDList::const_iterator it = |
+ channel_id_list_.begin(); |
+ |
+ // Correct because fetching channel_id_list_ will get them out in the |
+ // same order CreateChannelIDsForTest put them in. |
+ ASSERT_TRUE(it != channel_id_list_.end()); |
+ EXPECT_EQ("https://www.google.com:443", it->server_identifier()); |
+ |
+ ASSERT_TRUE(++it != channel_id_list_.end()); |
+ EXPECT_EQ("https://www.youtube.com:443", it->server_identifier()); |
+ |
+ ASSERT_TRUE(++it == channel_id_list_.end()); |
+ |
+ EXPECT_EQ(0, ssl_config_changed_count_); |
+} |
+ |
+TEST_F(BrowsingDataChannelIDHelperTest, DeleteChannelID) { |
+ CreateChannelIDsForTest(); |
+ scoped_refptr<BrowsingDataChannelIDHelper> helper( |
+ BrowsingDataChannelIDHelper::Create(testing_profile_.get())); |
+ |
+ helper->DeleteChannelID("https://www.google.com:443"); |
+ |
+ helper->StartFetching( |
+ base::Bind(&BrowsingDataChannelIDHelperTest::FetchCallback, |
+ base::Unretained(this))); |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ EXPECT_EQ(1, ssl_config_changed_count_); |
+ ASSERT_EQ(1UL, channel_id_list_.size()); |
+ net::ChannelIDStore::ChannelIDList::const_iterator it = |
+ channel_id_list_.begin(); |
+ |
+ ASSERT_TRUE(it != channel_id_list_.end()); |
+ EXPECT_EQ("https://www.youtube.com:443", it->server_identifier()); |
+ |
+ ASSERT_TRUE(++it == channel_id_list_.end()); |
+ |
+ helper->DeleteChannelID("https://www.youtube.com:443"); |
+ |
+ helper->StartFetching( |
+ base::Bind(&BrowsingDataChannelIDHelperTest::FetchCallback, |
+ base::Unretained(this))); |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ EXPECT_EQ(2, ssl_config_changed_count_); |
+ ASSERT_EQ(0UL, channel_id_list_.size()); |
+} |
+ |
+TEST_F(BrowsingDataChannelIDHelperTest, CannedEmpty) { |
+ std::string origin = "https://www.google.com"; |
+ |
+ scoped_refptr<CannedBrowsingDataChannelIDHelper> helper( |
+ new CannedBrowsingDataChannelIDHelper()); |
+ |
+ ASSERT_TRUE(helper->empty()); |
+ helper->AddChannelID(net::ChannelIDStore::ChannelID( |
+ origin, base::Time(), base::Time(), "key", "cert")); |
+ ASSERT_FALSE(helper->empty()); |
+ helper->Reset(); |
+ ASSERT_TRUE(helper->empty()); |
+} |