components/variations/net/variations_http_header_provider_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 "components/variations/net/variations_http_header_provider.h"
   6
   7 #include <string>
   8
   9 #include "base/base64.h"
  10 #include "base/message_loop/message_loop.h"
  11 #include "base/metrics/field_trial.h"
  12 #include "base/run_loop.h"
  13 #include "components/variations/entropy_provider.h"
  14 #include "components/variations/proto/client_variations.pb.h"
  15 #include "components/variations/variations_associated_data.h"
  16 #include "net/http/http_request_headers.h"
  17 #include "testing/gtest/include/gtest/gtest.h"
  18 #include "url/gurl.h"
  19
  20 namespace variations {
  21
  22 namespace {
  23
  24 // Decodes the variations header and extracts the variation ids.
  25 bool ExtractVariationIds(const std::string& variations,
  26                          std::set<VariationID>* variation_ids,
  27                          std::set<VariationID>* trigger_ids) {
  28   std::string serialized_proto;
  29   if (!base::Base64Decode(variations, &serialized_proto))
  30     return false;
  31   ClientVariations proto;
  32   if (!proto.ParseFromString(serialized_proto)) return false;
  33   for (int i = 0; i < proto.variation_id_size(); ++i)
  34     variation_ids->insert(proto.variation_id(i));
  35   for (int i = 0; i < proto.trigger_variation_id_size(); ++i)
  36     trigger_ids->insert(proto.trigger_variation_id(i));
  37   return true;
  38 }
  39
  40 scoped_refptr<base::FieldTrial> CreateTrialAndAssociateId(
  41     const std::string& trial_name,
  42     const std::string& default_group_name,
  43     IDCollectionKey key,
  44     VariationID id) {
  45   scoped_refptr<base::FieldTrial> trial(
  46       base::FieldTrialList::CreateFieldTrial(trial_name, default_group_name));
  47
  48   AssociateGoogleVariationID(key, trial->trial_name(), trial->group_name(), id);
  49
  50   return trial;
  51 }
  52
  53 }  // namespace
  54
  55 class VariationsHttpHeaderProviderTest : public ::testing::Test {
  56  public:
  57   VariationsHttpHeaderProviderTest() {}
  58
  59   ~VariationsHttpHeaderProviderTest() override {}
  60
  61   void TearDown() override { testing::ClearAllVariationIDs(); }
  62 };
  63
  64 TEST_F(VariationsHttpHeaderProviderTest, ShouldAppendHeaders) {
  65   struct {
  66     const char* url;
  67     bool should_append_headers;
  68   } cases[] = {
  69         {"http://google.com", true},
  70         {"http://www.google.com", true},
  71         {"http://m.google.com", true},
  72         {"http://google.ca", true},
  73         {"https://google.ca", true},
  74         {"http://google.co.uk", true},
  75         {"http://google.co.uk:8080/", true},
  76         {"http://www.google.co.uk:8080/", true},
  77         {"http://google", false},
  78
  79         {"http://youtube.com", true},
  80         {"http://www.youtube.com", true},
  81         {"http://www.youtube.ca", true},
  82         {"http://www.youtube.co.uk:8080/", true},
  83         {"https://www.youtube.com", true},
  84         {"http://youtube", false},
  85
  86         {"http://www.yahoo.com", false},
  87
  88         {"http://ad.doubleclick.net", true},
  89         {"https://a.b.c.doubleclick.net", true},
  90         {"https://a.b.c.doubleclick.net:8081", true},
  91         {"http://www.doubleclick.com", true},
  92         {"http://www.doubleclick.org", false},
  93         {"http://www.doubleclick.net.com", false},
  94         {"https://www.doubleclick.net.com", false},
  95
  96         {"http://ad.googlesyndication.com", true},
  97         {"https://a.b.c.googlesyndication.com", true},
  98         {"https://a.b.c.googlesyndication.com:8080", true},
  99         {"http://www.doubleclick.edu", false},
 100         {"http://www.googlesyndication.com.edu", false},
 101         {"https://www.googlesyndication.com.com", false},
 102
 103         {"http://www.googleadservices.com", true},
 104         {"http://www.googleadservices.com:8080", true},
 105         {"https://www.googleadservices.com", true},
 106         {"https://www.internal.googleadservices.com", true},
 107         {"https://www2.googleadservices.com", true},
 108         {"https://www.googleadservices.org", false},
 109         {"https://www.googleadservices.com.co.uk", false},
 110
 111         {"http://WWW.ANDROID.COM", true},
 112         {"http://www.android.com", true},
 113         {"http://www.doubleclick.com", true},
 114         {"http://www.doubleclick.net", true},
 115         {"http://www.ggpht.com", true},
 116         {"http://www.googleadservices.com", true},
 117         {"http://www.googleapis.com", true},
 118         {"http://www.googlesyndication.com", true},
 119         {"http://www.googleusercontent.com", true},
 120         {"http://www.googlevideo.com", true},
 121         {"http://ssl.gstatic.com", true},
 122         {"http://www.gstatic.com", true},
 123         {"http://www.ytimg.com", true},
 124         {"http://wwwytimg.com", false},
 125         {"http://ytimg.com", false},
 126
 127         {"http://www.android.org", false},
 128         {"http://www.doubleclick.org", false},
 129         {"http://www.doubleclick.net", true},
 130         {"http://www.ggpht.org", false},
 131         {"http://www.googleadservices.org", false},
 132         {"http://www.googleapis.org", false},
 133         {"http://www.googlesyndication.org", false},
 134         {"http://www.googleusercontent.org", false},
 135         {"http://www.googlevideo.org", false},
 136         {"http://ssl.gstatic.org", false},
 137         {"http://www.gstatic.org", false},
 138         {"http://www.ytimg.org", false},
 139
 140         {"http://a.b.android.com", true},
 141         {"http://a.b.doubleclick.com", true},
 142         {"http://a.b.doubleclick.net", true},
 143         {"http://a.b.ggpht.com", true},
 144         {"http://a.b.googleadservices.com", true},
 145         {"http://a.b.googleapis.com", true},
 146         {"http://a.b.googlesyndication.com", true},
 147         {"http://a.b.googleusercontent.com", true},
 148         {"http://a.b.googlevideo.com", true},
 149         {"http://ssl.gstatic.com", true},
 150         {"http://a.b.gstatic.com", true},
 151         {"http://a.b.ytimg.com", true},
 152     };
 153
 154   for (size_t i = 0; i < arraysize(cases); ++i) {
 155     const GURL url(cases[i].url);
 156     EXPECT_EQ(cases[i].should_append_headers,
 157               VariationsHttpHeaderProvider::ShouldAppendHeaders(url))
 158         << url;
 159   }
 160 }
 161
 162 TEST_F(VariationsHttpHeaderProviderTest, SetDefaultVariationIds_Valid) {
 163   base::MessageLoop loop;
 164   VariationsHttpHeaderProvider provider;
 165   GURL url("http://www.google.com");
 166   net::HttpRequestHeaders headers;
 167   std::string variations;
 168
 169   // Valid experiment ids.
 170   EXPECT_TRUE(provider.SetDefaultVariationIds("12,456,t789"));
 171   provider.InitVariationIDsCacheIfNeeded();
 172   provider.AppendHeaders(url, false, false, &headers);
 173   EXPECT_TRUE(headers.HasHeader("X-Client-Data"));
 174   headers.GetHeader("X-Client-Data", &variations);
 175   std::set<VariationID> variation_ids;
 176   std::set<VariationID> trigger_ids;
 177   ASSERT_TRUE(ExtractVariationIds(variations, &variation_ids, &trigger_ids));
 178   EXPECT_TRUE(variation_ids.find(12) != variation_ids.end());
 179   EXPECT_TRUE(variation_ids.find(456) != variation_ids.end());
 180   EXPECT_TRUE(trigger_ids.find(789) != trigger_ids.end());
 181   EXPECT_FALSE(variation_ids.find(789) != variation_ids.end());
 182 }
 183
 184 TEST_F(VariationsHttpHeaderProviderTest, SetDefaultVariationIds_Invalid) {
 185   base::MessageLoop loop;
 186   VariationsHttpHeaderProvider provider;
 187   GURL url("http://www.google.com");
 188   net::HttpRequestHeaders headers;
 189
 190   // Invalid experiment ids.
 191   EXPECT_FALSE(provider.SetDefaultVariationIds("abcd12,456"));
 192   provider.InitVariationIDsCacheIfNeeded();
 193   provider.AppendHeaders(url, false, false, &headers);
 194   EXPECT_FALSE(headers.HasHeader("X-Client-Data"));
 195
 196   // Invalid trigger experiment id
 197   EXPECT_FALSE(provider.SetDefaultVariationIds("12,tabc456"));
 198   provider.InitVariationIDsCacheIfNeeded();
 199   provider.AppendHeaders(url, false, false, &headers);
 200   EXPECT_FALSE(headers.HasHeader("X-Client-Data"));
 201 }
 202
 203 TEST_F(VariationsHttpHeaderProviderTest, OnFieldTrialGroupFinalized) {
 204   base::MessageLoop loop;
 205   base::FieldTrialList field_trial_list(
 206       new metrics::SHA1EntropyProvider("test"));
 207   VariationsHttpHeaderProvider provider;
 208   provider.InitVariationIDsCacheIfNeeded();
 209
 210   const std::string default_name = "default";
 211   scoped_refptr<base::FieldTrial> trial_1(CreateTrialAndAssociateId(
 212       "t1", default_name, GOOGLE_WEB_PROPERTIES, 123));
 213
 214   ASSERT_EQ(default_name, trial_1->group_name());
 215
 216   scoped_refptr<base::FieldTrial> trial_2(CreateTrialAndAssociateId(
 217       "t2", default_name, GOOGLE_WEB_PROPERTIES_TRIGGER, 456));
 218
 219   ASSERT_EQ(default_name, trial_2->group_name());
 220
 221   // Run the message loop to make sure OnFieldTrialGroupFinalized is called for
 222   // the two field trials.
 223   base::RunLoop().RunUntilIdle();
 224
 225   GURL url("http://www.google.com");
 226   net::HttpRequestHeaders headers;
 227   provider.AppendHeaders(url, false, false, &headers);
 228   std::string variations;
 229   headers.GetHeader("X-Client-Data", &variations);
 230
 231   std::set<VariationID> variation_ids;
 232   std::set<VariationID> trigger_ids;
 233   ASSERT_TRUE(ExtractVariationIds(variations, &variation_ids, &trigger_ids));
 234   EXPECT_TRUE(variation_ids.find(123) != variation_ids.end());
 235   EXPECT_TRUE(trigger_ids.find(456) != trigger_ids.end());
 236 }
 237
 238 }  // namespace variations