chrome/browser/browsing_data/browsing_data_channel_id_helper_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/browsing_data/browsing_data_channel_id_helper.h"
   6
   7 #include "base/bind.h"
   8 #include "base/run_loop.h"
   9 #include "chrome/test/base/testing_profile.h"
  10 #include "content/public/browser/browser_thread.h"
  11 #include "content/public/test/test_browser_thread_bundle.h"
  12 #include "net/ssl/channel_id_service.h"
  13 #include "net/url_request/url_request_context.h"
  14 #include "net/url_request/url_request_context_getter.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16
  17 using content::BrowserThread;
  18
  19 class BrowsingDataChannelIDHelperTest
  20     : public testing::Test,
  21       public net::SSLConfigService::Observer {
  22  public:
  23   BrowsingDataChannelIDHelperTest() : ssl_config_changed_count_(0) {
  24   }
  25
  26   virtual void SetUp() override {
  27     testing_profile_.reset(new TestingProfile());
  28
  29     testing_profile_->GetSSLConfigService()->AddObserver(this);
  30   }
  31
  32   virtual void TearDown() override {
  33     testing_profile_->GetSSLConfigService()->RemoveObserver(this);
  34   }
  35
  36   void CreateChannelIDsForTest() {
  37     net::URLRequestContext* context =
  38         testing_profile_->GetRequestContext()->GetURLRequestContext();
  39     net::ChannelIDStore* channel_id_store =
  40         context->channel_id_service()->GetChannelIDStore();
  41     channel_id_store->SetChannelID("https://www.google.com:443",
  42                                    base::Time(), base::Time(),
  43                                    "key", "cert");
  44     channel_id_store->SetChannelID("https://www.youtube.com:443",
  45                                    base::Time(), base::Time(),
  46                                    "key", "cert");
  47   }
  48
  49   void FetchCallback(
  50       const net::ChannelIDStore::ChannelIDList& channel_ids) {
  51     DCHECK_CURRENTLY_ON(BrowserThread::UI);
  52     channel_id_list_ = channel_ids;
  53   }
  54
  55   // net::SSLConfigService::Observer implementation:
  56   virtual void OnSSLConfigChanged() override {
  57     ssl_config_changed_count_++;
  58   }
  59
  60  protected:
  61   content::TestBrowserThreadBundle thread_bundle_;
  62   scoped_ptr<TestingProfile> testing_profile_;
  63
  64   net::ChannelIDStore::ChannelIDList channel_id_list_;
  65
  66   int ssl_config_changed_count_;
  67 };
  68
  69 TEST_F(BrowsingDataChannelIDHelperTest, FetchData) {
  70   CreateChannelIDsForTest();
  71   scoped_refptr<BrowsingDataChannelIDHelper> helper(
  72       BrowsingDataChannelIDHelper::Create(
  73           testing_profile_->GetRequestContext()));
  74
  75   helper->StartFetching(
  76       base::Bind(&BrowsingDataChannelIDHelperTest::FetchCallback,
  77                  base::Unretained(this)));
  78
  79   // Blocks until BrowsingDataChannelIDHelperTest::FetchCallback is
  80   // notified.
  81   base::RunLoop().RunUntilIdle();
  82
  83   ASSERT_EQ(2UL, channel_id_list_.size());
  84   net::ChannelIDStore::ChannelIDList::const_iterator it =
  85       channel_id_list_.begin();
  86
  87   // Correct because fetching channel_id_list_ will get them out in the
  88   // same order CreateChannelIDsForTest put them in.
  89   ASSERT_TRUE(it != channel_id_list_.end());
  90   EXPECT_EQ("https://www.google.com:443", it->server_identifier());
  91
  92   ASSERT_TRUE(++it != channel_id_list_.end());
  93   EXPECT_EQ("https://www.youtube.com:443", it->server_identifier());
  94
  95   ASSERT_TRUE(++it == channel_id_list_.end());
  96
  97   EXPECT_EQ(0, ssl_config_changed_count_);
  98 }
  99
 100 TEST_F(BrowsingDataChannelIDHelperTest, DeleteChannelID) {
 101   CreateChannelIDsForTest();
 102   scoped_refptr<BrowsingDataChannelIDHelper> helper(
 103       BrowsingDataChannelIDHelper::Create(
 104           testing_profile_->GetRequestContext()));
 105
 106   helper->DeleteChannelID("https://www.google.com:443");
 107
 108   helper->StartFetching(
 109       base::Bind(&BrowsingDataChannelIDHelperTest::FetchCallback,
 110                  base::Unretained(this)));
 111   base::RunLoop().RunUntilIdle();
 112
 113   EXPECT_EQ(1, ssl_config_changed_count_);
 114   ASSERT_EQ(1UL, channel_id_list_.size());
 115   net::ChannelIDStore::ChannelIDList::const_iterator it =
 116       channel_id_list_.begin();
 117
 118   ASSERT_TRUE(it != channel_id_list_.end());
 119   EXPECT_EQ("https://www.youtube.com:443", it->server_identifier());
 120
 121   ASSERT_TRUE(++it == channel_id_list_.end());
 122
 123   helper->DeleteChannelID("https://www.youtube.com:443");
 124
 125   helper->StartFetching(
 126       base::Bind(&BrowsingDataChannelIDHelperTest::FetchCallback,
 127                  base::Unretained(this)));
 128   base::RunLoop().RunUntilIdle();
 129
 130   EXPECT_EQ(2, ssl_config_changed_count_);
 131   ASSERT_EQ(0UL, channel_id_list_.size());
 132 }
 133
 134 TEST_F(BrowsingDataChannelIDHelperTest, CannedEmpty) {
 135   std::string origin = "https://www.google.com";
 136
 137   scoped_refptr<CannedBrowsingDataChannelIDHelper> helper(
 138       new CannedBrowsingDataChannelIDHelper());
 139
 140   ASSERT_TRUE(helper->empty());
 141   helper->AddChannelID(net::ChannelIDStore::ChannelID(
 142       origin, base::Time(), base::Time(), "key", "cert"));
 143   ASSERT_FALSE(helper->empty());
 144   helper->Reset();
 145   ASSERT_TRUE(helper->empty());
 146 }