components/domain_reliability/monitor_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/domain_reliability/monitor.h"
   6
   7 #include <map>
   8 #include <string>
   9 #include <vector>
  10
  11 #include "base/bind.h"
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/test/test_simple_task_runner.h"
  14 #include "components/domain_reliability/baked_in_configs.h"
  15 #include "components/domain_reliability/beacon.h"
  16 #include "components/domain_reliability/config.h"
  17 #include "components/domain_reliability/test_util.h"
  18 #include "net/base/host_port_pair.h"
  19 #include "net/base/load_flags.h"
  20 #include "net/http/http_response_headers.h"
  21 #include "net/http/http_util.h"
  22 #include "net/url_request/url_request_context_getter.h"
  23 #include "net/url_request/url_request_status.h"
  24 #include "net/url_request/url_request_test_util.h"
  25 #include "testing/gtest/include/gtest/gtest.h"
  26
  27 namespace domain_reliability {
  28
  29 namespace {
  30
  31 typedef std::vector<DomainReliabilityBeacon> BeaconVector;
  32
  33 static const size_t kAlwaysReportIndex = 0u;
  34 static const size_t kNeverReportIndex = 1u;
  35
  36 scoped_refptr<net::HttpResponseHeaders> MakeHttpResponseHeaders(
  37     const std::string& headers) {
  38   return scoped_refptr<net::HttpResponseHeaders>(
  39       new net::HttpResponseHeaders(net::HttpUtil::AssembleRawHeaders(
  40           headers.c_str(), headers.length())));
  41 }
  42
  43 }  // namespace
  44
  45 class DomainReliabilityMonitorTest : public testing::Test {
  46  protected:
  47   typedef DomainReliabilityMonitor::RequestInfo RequestInfo;
  48
  49   DomainReliabilityMonitorTest()
  50       : pref_task_runner_(new base::TestSimpleTaskRunner()),
  51         network_task_runner_(new base::TestSimpleTaskRunner()),
  52         url_request_context_getter_(
  53             new net::TestURLRequestContextGetter(network_task_runner_)),
  54         time_(new MockTime()),
  55         monitor_("test-reporter",
  56                  pref_task_runner_,
  57                  network_task_runner_,
  58                  scoped_ptr<MockableTime>(time_)),
  59         context_(nullptr) {
  60     monitor_.MoveToNetworkThread();
  61     monitor_.InitURLRequestContext(url_request_context_getter_);
  62     monitor_.SetDiscardUploads(false);
  63     context_ = monitor_.AddContextForTesting(MakeTestConfig());
  64   }
  65
  66   static RequestInfo MakeRequestInfo() {
  67     RequestInfo request;
  68     request.status = net::URLRequestStatus();
  69     request.response_info.socket_address =
  70         net::HostPortPair::FromString("12.34.56.78:80");
  71     request.response_info.headers = MakeHttpResponseHeaders(
  72         "HTTP/1.1 200 OK\n\n");
  73     request.response_info.was_cached = false;
  74     request.response_info.network_accessed = true;
  75     request.response_info.was_fetched_via_proxy = false;
  76     request.load_flags = 0;
  77     request.is_upload = false;
  78     return request;
  79   }
  80
  81   void OnRequestLegComplete(const RequestInfo& info) {
  82     monitor_.OnRequestLegComplete(info);
  83   }
  84
  85   size_t CountPendingBeacons() {
  86     BeaconVector beacons;
  87     context_->GetQueuedBeaconsForTesting(&beacons);
  88     return beacons.size();
  89   }
  90
  91   bool CheckRequestCounts(size_t index,
  92                           uint32 expected_successful,
  93                           uint32 expected_failed) {
  94     return CheckRequestCounts(context_,
  95                               index,
  96                               expected_successful,
  97                               expected_failed);
  98   }
  99
 100   bool CheckRequestCounts(DomainReliabilityContext* context,
 101                           size_t index,
 102                           uint32 expected_successful,
 103                           uint32 expected_failed) {
 104     uint32 successful, failed;
 105     context->GetRequestCountsForTesting(index, &successful, &failed);
 106     EXPECT_EQ(expected_successful, successful);
 107     EXPECT_EQ(expected_failed, failed);
 108     return expected_successful == successful && expected_failed == failed;
 109   }
 110
 111   DomainReliabilityContext* CreateAndAddContext(const std::string& domain) {
 112     return monitor_.AddContextForTesting(MakeTestConfigWithDomain(domain));
 113   }
 114
 115   scoped_refptr<base::TestSimpleTaskRunner> pref_task_runner_;
 116   scoped_refptr<base::TestSimpleTaskRunner> network_task_runner_;
 117   scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_;
 118   MockTime* time_;
 119   DomainReliabilityMonitor monitor_;
 120   DomainReliabilityContext* context_;
 121   DomainReliabilityMonitor::RequestInfo request_;
 122 };
 123
 124 namespace {
 125
 126 TEST_F(DomainReliabilityMonitorTest, Create) {
 127   EXPECT_EQ(0u, CountPendingBeacons());
 128   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 0u));
 129   EXPECT_TRUE(CheckRequestCounts(kNeverReportIndex, 0u, 0u));
 130 }
 131
 132 TEST_F(DomainReliabilityMonitorTest, NoContext) {
 133   RequestInfo request = MakeRequestInfo();
 134   request.url = GURL("http://no-context/");
 135   OnRequestLegComplete(request);
 136
 137   EXPECT_EQ(0u, CountPendingBeacons());
 138   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 0u));
 139   EXPECT_TRUE(CheckRequestCounts(kNeverReportIndex, 0u, 0u));
 140 }
 141
 142 TEST_F(DomainReliabilityMonitorTest, NotReported) {
 143   RequestInfo request = MakeRequestInfo();
 144   request.url = GURL("http://example/never_report");
 145   OnRequestLegComplete(request);
 146
 147   EXPECT_EQ(0u, CountPendingBeacons());
 148   EXPECT_TRUE(CheckRequestCounts(kNeverReportIndex, 1u, 0u));
 149 }
 150
 151 TEST_F(DomainReliabilityMonitorTest, NetworkFailure) {
 152   RequestInfo request = MakeRequestInfo();
 153   request.url = GURL("http://example/always_report");
 154   request.status = net::URLRequestStatus::FromError(net::ERR_CONNECTION_RESET);
 155   request.response_info.headers = nullptr;
 156   OnRequestLegComplete(request);
 157
 158   EXPECT_EQ(1u, CountPendingBeacons());
 159   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 1u));
 160 }
 161
 162 TEST_F(DomainReliabilityMonitorTest, ServerFailure) {
 163   RequestInfo request = MakeRequestInfo();
 164   request.url = GURL("http://example/always_report");
 165   request.response_info.headers =
 166       MakeHttpResponseHeaders("HTTP/1.1 500 :(\n\n");
 167   OnRequestLegComplete(request);
 168
 169   EXPECT_EQ(1u, CountPendingBeacons());
 170   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 1u));
 171 }
 172
 173 TEST_F(DomainReliabilityMonitorTest, NotReportedFailure) {
 174   RequestInfo request = MakeRequestInfo();
 175   request.url = GURL("http://example/never_report");
 176   request.status = net::URLRequestStatus::FromError(net::ERR_CONNECTION_RESET);
 177   OnRequestLegComplete(request);
 178
 179   EXPECT_EQ(0u, CountPendingBeacons());
 180   EXPECT_TRUE(CheckRequestCounts(kNeverReportIndex, 0u, 1u));
 181 }
 182
 183 TEST_F(DomainReliabilityMonitorTest, Request) {
 184   RequestInfo request = MakeRequestInfo();
 185   request.url = GURL("http://example/always_report");
 186   OnRequestLegComplete(request);
 187
 188   EXPECT_EQ(1u, CountPendingBeacons());
 189   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 1u, 0u));
 190 }
 191
 192 // Make sure the monitor does not log requests that did not access the network.
 193 TEST_F(DomainReliabilityMonitorTest, DidNotAccessNetwork) {
 194   RequestInfo request = MakeRequestInfo();
 195   request.url = GURL("http://example/always_report");
 196   request.response_info.network_accessed = false;
 197   OnRequestLegComplete(request);
 198
 199   EXPECT_EQ(0u, CountPendingBeacons());
 200   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 0u));
 201 }
 202
 203 // Make sure the monitor does not log requests that don't send cookies.
 204 TEST_F(DomainReliabilityMonitorTest, DoNotSendCookies) {
 205   RequestInfo request = MakeRequestInfo();
 206   request.url = GURL("http://example/always_report");
 207   request.load_flags = net::LOAD_DO_NOT_SEND_COOKIES;
 208   OnRequestLegComplete(request);
 209
 210   EXPECT_EQ(0u, CountPendingBeacons());
 211   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 0u));
 212 }
 213
 214 // Make sure the monitor does not log upload requests.
 215 TEST_F(DomainReliabilityMonitorTest, IsUpload) {
 216   RequestInfo request = MakeRequestInfo();
 217   request.url = GURL("http://example/always_report");
 218   request.is_upload = true;
 219   OnRequestLegComplete(request);
 220
 221   EXPECT_EQ(0u, CountPendingBeacons());
 222   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 0u));
 223 }
 224
 225 // Make sure the monitor does not log a network-local error.
 226 TEST_F(DomainReliabilityMonitorTest, LocalError) {
 227   RequestInfo request = MakeRequestInfo();
 228   request.url = GURL("http://example/always_report");
 229   request.status =
 230       net::URLRequestStatus::FromError(net::ERR_PROXY_CONNECTION_FAILED);
 231   OnRequestLegComplete(request);
 232
 233   EXPECT_EQ(0u, CountPendingBeacons());
 234   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 0u));
 235 }
 236
 237 // Make sure the monitor does not log the proxy's IP if one was used.
 238 TEST_F(DomainReliabilityMonitorTest, WasFetchedViaProxy) {
 239   RequestInfo request = MakeRequestInfo();
 240   request.url = GURL("http://example/always_report");
 241   request.response_info.socket_address =
 242       net::HostPortPair::FromString("127.0.0.1:3128");
 243   request.response_info.was_fetched_via_proxy = true;
 244   OnRequestLegComplete(request);
 245
 246   BeaconVector beacons;
 247   context_->GetQueuedBeaconsForTesting(&beacons);
 248   EXPECT_EQ(1u, beacons.size());
 249   EXPECT_TRUE(beacons[0].server_ip.empty());
 250
 251   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 1u, 0u));
 252 }
 253
 254 // Make sure the monitor does not log the cached IP returned after a successful
 255 // cache revalidation request.
 256 TEST_F(DomainReliabilityMonitorTest,
 257        NoCachedIPFromSuccessfulRevalidationRequest) {
 258   RequestInfo request = MakeRequestInfo();
 259   request.url = GURL("http://example/always_report");
 260   request.response_info.was_cached = true;
 261   OnRequestLegComplete(request);
 262
 263   BeaconVector beacons;
 264   context_->GetQueuedBeaconsForTesting(&beacons);
 265   EXPECT_EQ(1u, beacons.size());
 266   EXPECT_TRUE(beacons[0].server_ip.empty());
 267 }
 268
 269 // Make sure the monitor does not log the cached IP returned with a failed
 270 // cache revalidation request.
 271 TEST_F(DomainReliabilityMonitorTest, NoCachedIPFromFailedRevalidationRequest) {
 272   RequestInfo request = MakeRequestInfo();
 273   request.url = GURL("http://example/always_report");
 274   request.response_info.was_cached = true;
 275   request.status =
 276       net::URLRequestStatus::FromError(net::ERR_NAME_RESOLUTION_FAILED);
 277   OnRequestLegComplete(request);
 278
 279   BeaconVector beacons;
 280   context_->GetQueuedBeaconsForTesting(&beacons);
 281   EXPECT_EQ(1u, beacons.size());
 282   EXPECT_TRUE(beacons[0].server_ip.empty());
 283 }
 284
 285 TEST_F(DomainReliabilityMonitorTest, AtLeastOneBakedInConfig) {
 286   DCHECK(kBakedInJsonConfigs[0] != nullptr);
 287 }
 288
 289 // Will fail when baked-in configs expire, as a reminder to update them.
 290 // (Contact ttuttle@chromium.org if this starts failing.)
 291 TEST_F(DomainReliabilityMonitorTest, AddBakedInConfigs) {
 292   // AddBakedInConfigs DCHECKs that the baked-in configs parse correctly, so
 293   // this unittest will fail if someone tries to add an invalid config to the
 294   // source tree.
 295   monitor_.AddBakedInConfigs();
 296
 297   // Count the number of baked-in configs.
 298   size_t num_baked_in_configs = 0;
 299   for (const char* const* p = kBakedInJsonConfigs; *p; ++p)
 300     ++num_baked_in_configs;
 301
 302   // The monitor should have contexts for all of the baked-in configs, plus the
 303   // test one added in the test constructor.
 304   EXPECT_EQ(num_baked_in_configs + 1, monitor_.contexts_size_for_testing());
 305 }
 306
 307 TEST_F(DomainReliabilityMonitorTest, ClearBeacons) {
 308   // Initially the monitor should have just the test context, with no beacons.
 309   EXPECT_EQ(1u, monitor_.contexts_size_for_testing());
 310   EXPECT_EQ(0u, CountPendingBeacons());
 311   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 0u));
 312   EXPECT_TRUE(CheckRequestCounts(kNeverReportIndex, 0u, 0u));
 313
 314   // Add a beacon.
 315   RequestInfo request = MakeRequestInfo();
 316   request.url = GURL("http://example/always_report");
 317   OnRequestLegComplete(request);
 318
 319   // Make sure it was added.
 320   EXPECT_EQ(1u, CountPendingBeacons());
 321   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 1u, 0u));
 322
 323   monitor_.ClearBrowsingData(CLEAR_BEACONS);
 324
 325   // Make sure the beacon was cleared, but not the contexts.
 326   EXPECT_EQ(1u, monitor_.contexts_size_for_testing());
 327   EXPECT_EQ(0u, CountPendingBeacons());
 328   EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 0u));
 329   EXPECT_TRUE(CheckRequestCounts(kNeverReportIndex, 0u, 0u));
 330 }
 331
 332 TEST_F(DomainReliabilityMonitorTest, ClearContexts) {
 333   // Initially the monitor should have just the test context.
 334   EXPECT_EQ(1u, monitor_.contexts_size_for_testing());
 335
 336   monitor_.ClearBrowsingData(CLEAR_CONTEXTS);
 337
 338   // Clearing contexts should leave the monitor with none.
 339   EXPECT_EQ(0u, monitor_.contexts_size_for_testing());
 340 }
 341
 342 TEST_F(DomainReliabilityMonitorTest, WildcardMatchesSelf) {
 343   DomainReliabilityContext* context = CreateAndAddContext("*.wildcard");
 344
 345   RequestInfo request = MakeRequestInfo();
 346   request.url = GURL("http://wildcard/always_report");
 347   OnRequestLegComplete(request);
 348   EXPECT_TRUE(CheckRequestCounts(context, kAlwaysReportIndex, 1u, 0u));
 349 }
 350
 351 TEST_F(DomainReliabilityMonitorTest, WildcardMatchesSubdomain) {
 352   DomainReliabilityContext* context = CreateAndAddContext("*.wildcard");
 353
 354   RequestInfo request = MakeRequestInfo();
 355   request.url = GURL("http://test.wildcard/always_report");
 356   OnRequestLegComplete(request);
 357   EXPECT_TRUE(CheckRequestCounts(context, kAlwaysReportIndex, 1u, 0u));
 358 }
 359
 360 TEST_F(DomainReliabilityMonitorTest, WildcardDoesntMatchSubsubdomain) {
 361   DomainReliabilityContext* context = CreateAndAddContext("*.wildcard");
 362
 363   RequestInfo request = MakeRequestInfo();
 364   request.url = GURL("http://test.test.wildcard/always_report");
 365   OnRequestLegComplete(request);
 366   EXPECT_TRUE(CheckRequestCounts(context, kAlwaysReportIndex, 0u, 0u));
 367 }
 368
 369 TEST_F(DomainReliabilityMonitorTest, WildcardPrefersSelfToSelfWildcard) {
 370   DomainReliabilityContext* context1 = CreateAndAddContext("wildcard");
 371   DomainReliabilityContext* context2 = CreateAndAddContext("*.wildcard");
 372
 373   RequestInfo request = MakeRequestInfo();
 374   request.url = GURL("http://wildcard/always_report");
 375   OnRequestLegComplete(request);
 376
 377   EXPECT_TRUE(CheckRequestCounts(context1, kAlwaysReportIndex, 1u, 0u));
 378   EXPECT_TRUE(CheckRequestCounts(context2, kAlwaysReportIndex, 0u, 0u));
 379 }
 380
 381 TEST_F(DomainReliabilityMonitorTest, WildcardPrefersSelfToParentWildcard) {
 382   DomainReliabilityContext* context1 = CreateAndAddContext("test.wildcard");
 383   DomainReliabilityContext* context2 = CreateAndAddContext("*.wildcard");
 384
 385   RequestInfo request = MakeRequestInfo();
 386   request.url = GURL("http://test.wildcard/always_report");
 387   OnRequestLegComplete(request);
 388
 389   EXPECT_TRUE(CheckRequestCounts(context1, kAlwaysReportIndex, 1u, 0u));
 390   EXPECT_TRUE(CheckRequestCounts(context2, kAlwaysReportIndex, 0u, 0u));
 391 }
 392
 393 TEST_F(DomainReliabilityMonitorTest,
 394     WildcardPrefersSelfWildcardToParentWildcard) {
 395   DomainReliabilityContext* context1 = CreateAndAddContext("*.test.wildcard");
 396   DomainReliabilityContext* context2 = CreateAndAddContext("*.wildcard");
 397
 398   RequestInfo request = MakeRequestInfo();
 399   request.url = GURL("http://test.wildcard/always_report");
 400   OnRequestLegComplete(request);
 401
 402   EXPECT_TRUE(CheckRequestCounts(context1, kAlwaysReportIndex, 1u, 0u));
 403   EXPECT_TRUE(CheckRequestCounts(context2, kAlwaysReportIndex, 0u, 0u));
 404 }
 405
 406 }  // namespace
 407
 408 }  // namespace domain_reliability