Add client for background testing of HTTP pipelining.

chrome/browser/net/http_pipelining_compatibility_client_unittest.cc

+// 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 <map>
+#include <string>
+#include <vector>
+
+#include "base/basictypes.h"
+#include "base/message_loop.h"
+#include "base/metrics/histogram.h"
+#include "base/stl_util.h"
+#include "base/stringprintf.h"
+#include "chrome/browser/net/http_pipelining_compatibility_client.h"

[willchan no longer on Chromium, 2012/02/08 15:58:19]

should go first

[James Simonsen, 2012/02/10 01:28:48]

Done.

+#include "chrome/test/base/test_url_request_context_getter.h"
+#include "content/test/test_browser_thread.h"
+#include "net/base/net_errors.h"
+#include "net/base/test_completion_callback.h"
+#include "net/url_request/url_request_context_getter.h"
+#include "net/test/test_server.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace chrome_browser_net {
+
+namespace {
+
+static const char* kHistogramNames[] = {

[willchan no longer on Chromium, 2012/02/08 15:58:19]

static const char* const

[James Simonsen, 2012/02/10 01:28:48]

Done.

+  "NetConnectivity.Pipeline.0.NetworkError",
+  "NetConnectivity.Pipeline.0.ResponseCode",
+  "NetConnectivity.Pipeline.0.Status",
+  "NetConnectivity.Pipeline.1.NetworkError",
+  "NetConnectivity.Pipeline.1.ResponseCode",
+  "NetConnectivity.Pipeline.1.Status",
+  "NetConnectivity.Pipeline.2.NetworkError",
+  "NetConnectivity.Pipeline.2.ResponseCode",
+  "NetConnectivity.Pipeline.2.Status",
+};
+
+enum HistogramField {
+  FIELD_NETWORK_ERROR,
+  FIELD_RESPONSE_CODE,
+  FIELD_STATUS,
+};
+
+class HttpPipeliningCompatibilityClientTest : public testing::Test {
+ public:
+  HttpPipeliningCompatibilityClientTest()
+      : test_server_(
+          net::TestServer::TYPE_HTTP,
+          FilePath(FILE_PATH_LITERAL("chrome/test/data/http_pipelining"))) {
+  }
+
+ protected:
+  virtual void SetUp() {

[mmenke1, 2012/02/08 16:07:47]

nit:  These should be OVERRIDE.

[James Simonsen, 2012/02/10 01:28:48]

Done.

+    ASSERT_TRUE(test_server_.Start());
+    context_ = new TestURLRequestContextGetter;
+    context_->AddRef();
+
+    for (size_t i = 0; i < arraysize(kHistogramNames); ++i) {
+      const char* name = kHistogramNames[i];
+      base::Histogram::SampleSet sample = GetHistogram(name);
+      if (sample.TotalCount() > 0) {
+        original_histograms_[name] = sample;
+      }
+    }
+  }
+
+  virtual void TearDown() {
+    content::BrowserThread::ReleaseSoon(content::BrowserThread::IO,
+                                        FROM_HERE, context_);
+    message_loop_.RunAllPending();
+  }
+
+  void RunTest(
+      std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests) {
+    HttpPipeliningCompatibilityClient client;
+    net::TestCompletionCallback callback;
+    client.Start(test_server_.GetURL("").spec(),
+                 requests, callback.callback(),
+                 context_->GetURLRequestContext());
+    callback.WaitForResult();
+
+    for (size_t i = 0; i < arraysize(kHistogramNames); ++i) {
+      const char* name = kHistogramNames[i];
+      base::Histogram::SampleSet sample = GetHistogram(name);
+      if (ContainsKey(original_histograms_, name)) {
+        sample.Subtract(original_histograms_[name]);
+      }
+      histograms_[name] = sample;
+    }
+  }
+
+  base::Histogram::SampleSet GetHistogramValue(int request_id,
+                                               HistogramField field) {
+    const char* field_str = "";
+    switch (field) {
+      case FIELD_STATUS:
+        field_str = "Status";
+        break;
+
+      case FIELD_NETWORK_ERROR:
+        field_str = "NetworkError";
+        break;
+
+      case FIELD_RESPONSE_CODE:
+        field_str = "ResponseCode";
+        break;
+
+      default:
+        NOTREACHED();
+        break;
+    }
+
+    std::string name = base::StringPrintf("NetConnectivity.Pipeline.%d.%s",
+                                          request_id, field_str);
+    return histograms_[name];
+  }
+
+ private:

[willchan no longer on Chromium, 2012/02/08 15:58:19]

it shouldn't go protected=>private=>protected=>private. you should move the data
members up to the appropriate sections and ditch the extra protected/private
sections.

[James Simonsen, 2012/02/10 01:28:48]

Done.

+  base::Histogram::SampleSet GetHistogram(const char* name) {
+    base::Histogram::SampleSet sample;
+    base::Histogram* histogram;
+    if (base::StatisticsRecorder::FindHistogram(name, &histogram)) {
+      histogram->SnapshotSample(&sample);
+    }
+    return sample;
+  }
+
+ protected:
+  MessageLoopForIO message_loop_;
+  net::TestServer test_server_;
+  TestURLRequestContextGetter* context_;
+
+ private:
+  std::map<std::string, base::Histogram::SampleSet> histograms_;
+  std::map<std::string, base::Histogram::SampleSet> original_histograms_;
+};
+
+TEST_F(HttpPipeliningCompatibilityClientTest, Success) {
+  HttpPipeliningCompatibilityClient::RequestInfo info;
+  info.filename = "files/alphabet.txt";
+  info.expected_response = "abcdefghijklmnopqrstuvwxyz";
+  std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests;
+  requests.push_back(info);
+
+  RunTest(requests);
+
+  base::Histogram::SampleSet status_sample =
+      GetHistogramValue(0, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample.TotalCount());
+  EXPECT_EQ(1, status_sample.counts(
+      HttpPipeliningCompatibilityClient::SUCCESS));
+
+  base::Histogram::SampleSet network_sample =
+      GetHistogramValue(0, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(0, network_sample.TotalCount());
+
+  base::Histogram::SampleSet response_sample =
+      GetHistogramValue(0, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(1, response_sample.TotalCount());
+  EXPECT_EQ(1, response_sample.counts(200));
+}
+
+TEST_F(HttpPipeliningCompatibilityClientTest, TooSmall) {
+  HttpPipeliningCompatibilityClient::RequestInfo info;
+  info.filename = "files/alphabet.txt";
+  info.expected_response = "abcdefghijklmnopqrstuvwxyz26";
+  std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests;
+  requests.push_back(info);
+
+  RunTest(requests);
+
+  base::Histogram::SampleSet status_sample =
+      GetHistogramValue(0, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample.TotalCount());
+  EXPECT_EQ(1, status_sample.counts(
+      HttpPipeliningCompatibilityClient::TOO_SMALL));
+
+  base::Histogram::SampleSet network_sample =
+      GetHistogramValue(0, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(0, network_sample.TotalCount());
+
+  base::Histogram::SampleSet response_sample =
+      GetHistogramValue(0, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(1, response_sample.TotalCount());
+  EXPECT_EQ(1, response_sample.counts(200));
+}
+
+TEST_F(HttpPipeliningCompatibilityClientTest, TooLarge) {
+  HttpPipeliningCompatibilityClient::RequestInfo info;
+  info.filename = "files/alphabet.txt";
+  info.expected_response = "abc";
+  std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests;
+  requests.push_back(info);
+
+  RunTest(requests);
+
+  base::Histogram::SampleSet status_sample =
+      GetHistogramValue(0, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample.TotalCount());
+  EXPECT_EQ(1, status_sample.counts(
+      HttpPipeliningCompatibilityClient::TOO_LARGE));
+
+  base::Histogram::SampleSet network_sample =
+      GetHistogramValue(0, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(0, network_sample.TotalCount());
+
+  base::Histogram::SampleSet response_sample =
+      GetHistogramValue(0, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(1, response_sample.TotalCount());
+  EXPECT_EQ(1, response_sample.counts(200));
+}
+
+TEST_F(HttpPipeliningCompatibilityClientTest, Mismatch) {
+  HttpPipeliningCompatibilityClient::RequestInfo info;
+  info.filename = "files/alphabet.txt";
+  info.expected_response = "zyxwvutsrqponmlkjihgfedcba";
+  std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests;
+  requests.push_back(info);
+
+  RunTest(requests);
+
+  base::Histogram::SampleSet status_sample =
+      GetHistogramValue(0, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample.TotalCount());
+  EXPECT_EQ(1, status_sample.counts(
+      HttpPipeliningCompatibilityClient::CONTENT_MISMATCH));
+
+  base::Histogram::SampleSet network_sample =
+      GetHistogramValue(0, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(0, network_sample.TotalCount());
+
+  base::Histogram::SampleSet response_sample =
+      GetHistogramValue(0, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(1, response_sample.TotalCount());
+  EXPECT_EQ(1, response_sample.counts(200));
+}
+
+TEST_F(HttpPipeliningCompatibilityClientTest, Redirect) {
+  HttpPipeliningCompatibilityClient::RequestInfo info;
+  info.filename = "server-redirect?http://foo.bar/asdf";
+  info.expected_response = "shouldn't matter";
+  std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests;
+  requests.push_back(info);
+
+  RunTest(requests);
+
+  base::Histogram::SampleSet status_sample =
+      GetHistogramValue(0, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample.TotalCount());
+  EXPECT_EQ(1, status_sample.counts(
+      HttpPipeliningCompatibilityClient::REDIRECTED));
+
+  base::Histogram::SampleSet network_sample =
+      GetHistogramValue(0, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(0, network_sample.TotalCount());
+
+  base::Histogram::SampleSet response_sample =
+      GetHistogramValue(0, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(0, response_sample.TotalCount());
+}
+
+TEST_F(HttpPipeliningCompatibilityClientTest, AuthRequired) {
+  HttpPipeliningCompatibilityClient::RequestInfo info;
+  info.filename = "auth-basic";
+  info.expected_response = "shouldn't matter";
+  std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests;
+  requests.push_back(info);
+
+  RunTest(requests);
+
+  base::Histogram::SampleSet status_sample =
+      GetHistogramValue(0, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample.TotalCount());
+  EXPECT_EQ(1, status_sample.counts(
+      HttpPipeliningCompatibilityClient::BAD_RESPONSE_CODE));
+
+  base::Histogram::SampleSet network_sample =
+      GetHistogramValue(0, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(0, network_sample.TotalCount());
+
+  base::Histogram::SampleSet response_sample =
+      GetHistogramValue(0, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(1, response_sample.TotalCount());
+  EXPECT_EQ(1, response_sample.counts(401));
+}
+
+TEST_F(HttpPipeliningCompatibilityClientTest, NoContent) {
+  HttpPipeliningCompatibilityClient::RequestInfo info;
+  info.filename = "nocontent";
+  info.expected_response = "shouldn't matter";
+  std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests;
+  requests.push_back(info);
+
+  RunTest(requests);
+
+  base::Histogram::SampleSet status_sample =
+      GetHistogramValue(0, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample.TotalCount());
+  EXPECT_EQ(1, status_sample.counts(
+      HttpPipeliningCompatibilityClient::BAD_RESPONSE_CODE));
+
+  base::Histogram::SampleSet network_sample =
+      GetHistogramValue(0, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(0, network_sample.TotalCount());
+
+  base::Histogram::SampleSet response_sample =
+      GetHistogramValue(0, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(1, response_sample.TotalCount());
+  EXPECT_EQ(1, response_sample.counts(204));
+}
+
+TEST_F(HttpPipeliningCompatibilityClientTest, CloseSocket) {
+  HttpPipeliningCompatibilityClient::RequestInfo info;
+  info.filename = "close-socket";
+  info.expected_response = "shouldn't matter";
+  std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests;
+  requests.push_back(info);
+
+  RunTest(requests);
+
+  base::Histogram::SampleSet status_sample =
+      GetHistogramValue(0, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample.TotalCount());
+  EXPECT_EQ(1, status_sample.counts(
+      HttpPipeliningCompatibilityClient::NETWORK_ERROR));
+
+  base::Histogram::SampleSet network_sample =
+      GetHistogramValue(0, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(1, network_sample.TotalCount());
+  EXPECT_EQ(1, network_sample.counts(-net::ERR_EMPTY_RESPONSE));
+
+  base::Histogram::SampleSet response_sample =
+      GetHistogramValue(0, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(0, response_sample.TotalCount());
+}
+
+TEST_F(HttpPipeliningCompatibilityClientTest, OldHttpVersion) {
+  HttpPipeliningCompatibilityClient::RequestInfo info;
+  info.filename = "http-1.0";
+  info.expected_response = "abcdefghijklmnopqrstuvwxyz";
+  std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests;
+  requests.push_back(info);
+
+  RunTest(requests);
+
+  base::Histogram::SampleSet status_sample =
+      GetHistogramValue(0, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample.TotalCount());
+  EXPECT_EQ(1, status_sample.counts(
+      HttpPipeliningCompatibilityClient::BAD_HTTP_VERSION));
+
+  base::Histogram::SampleSet network_sample =
+      GetHistogramValue(0, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(0, network_sample.TotalCount());
+
+  base::Histogram::SampleSet response_sample =
+      GetHistogramValue(0, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(1, response_sample.TotalCount());
+  EXPECT_EQ(1, response_sample.counts(200));
+}
+
+TEST_F(HttpPipeliningCompatibilityClientTest, MultipleRequests) {
+  std::vector<HttpPipeliningCompatibilityClient::RequestInfo> requests;
+
+  HttpPipeliningCompatibilityClient::RequestInfo info1;
+  info1.filename = "files/alphabet.txt";
+  info1.expected_response = "abcdefghijklmnopqrstuvwxyz";
+  requests.push_back(info1);
+
+  HttpPipeliningCompatibilityClient::RequestInfo info2;
+  info2.filename = "close-socket";
+  info2.expected_response = "shouldn't matter";
+  requests.push_back(info2);
+
+  HttpPipeliningCompatibilityClient::RequestInfo info3;
+  info3.filename = "auth-basic";
+  info3.expected_response = "shouldn't matter";
+  requests.push_back(info3);
+
+  RunTest(requests);
+
+  base::Histogram::SampleSet status_sample1 =
+      GetHistogramValue(0, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample1.TotalCount());
+  EXPECT_EQ(1, status_sample1.counts(
+      HttpPipeliningCompatibilityClient::SUCCESS));
+
+  base::Histogram::SampleSet network_sample1 =
+      GetHistogramValue(0, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(0, network_sample1.TotalCount());
+
+  base::Histogram::SampleSet response_sample1 =
+      GetHistogramValue(0, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(1, response_sample1.TotalCount());
+  EXPECT_EQ(1, response_sample1.counts(200));
+
+  base::Histogram::SampleSet status_sample2 =
+      GetHistogramValue(1, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample2.TotalCount());
+  EXPECT_EQ(1, status_sample2.counts(
+      HttpPipeliningCompatibilityClient::NETWORK_ERROR));
+
+  base::Histogram::SampleSet network_sample2 =
+      GetHistogramValue(1, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(1, network_sample2.TotalCount());
+  EXPECT_EQ(1, network_sample2.counts(-net::ERR_EMPTY_RESPONSE));
+
+  base::Histogram::SampleSet response_sample2 =
+      GetHistogramValue(1, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(0, response_sample2.TotalCount());
+
+  base::Histogram::SampleSet status_sample3 =
+      GetHistogramValue(2, FIELD_STATUS);
+  EXPECT_EQ(1, status_sample3.TotalCount());
+  EXPECT_EQ(1, status_sample3.counts(
+      HttpPipeliningCompatibilityClient::BAD_RESPONSE_CODE));
+
+  base::Histogram::SampleSet network_sample3 =
+      GetHistogramValue(2, FIELD_NETWORK_ERROR);
+  EXPECT_EQ(0, network_sample3.TotalCount());
+
+  base::Histogram::SampleSet response_sample3 =
+      GetHistogramValue(2, FIELD_RESPONSE_CODE);
+  EXPECT_EQ(1, response_sample3.TotalCount());
+  EXPECT_EQ(1, response_sample3.counts(401));
+}
+
+}  // anonymous namespace
+
+}  // namespace chrome_browser_net