net/http/http_vary_data_unittest.cc

   1 // Copyright (c) 2006-2008 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 <algorithm>
   6
   7 #include "net/http/http_request_info.h"
   8 #include "net/http/http_response_headers.h"
   9 #include "net/http/http_vary_data.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 namespace net {
  13
  14 namespace {
  15
  16 typedef testing::Test HttpVaryDataTest;
  17
  18 struct TestTransaction {
  19   HttpRequestInfo request;
  20   scoped_refptr<HttpResponseHeaders> response;
  21
  22   void Init(const std::string& request_headers,
  23             const std::string& response_headers) {
  24     std::string temp(response_headers);
  25     std::replace(temp.begin(), temp.end(), '\n', '\0');
  26     response = new HttpResponseHeaders(temp);
  27
  28     request.extra_headers.Clear();
  29     request.extra_headers.AddHeadersFromString(request_headers);
  30   }
  31 };
  32
  33 }  // namespace
  34
  35 TEST(HttpVaryDataTest, IsInvalid) {
  36   // All of these responses should result in an invalid vary data object.
  37   const char* const kTestResponses[] = {
  38     "HTTP/1.1 200 OK\n\n",
  39     "HTTP/1.1 200 OK\nVary: *\n\n",
  40     "HTTP/1.1 200 OK\nVary: cookie, *, bar\n\n",
  41     "HTTP/1.1 200 OK\nVary: cookie\nFoo: 1\nVary: *\n\n",
  42   };
  43
  44   for (size_t i = 0; i < arraysize(kTestResponses); ++i) {
  45     TestTransaction t;
  46     t.Init(std::string(), kTestResponses[i]);
  47
  48     HttpVaryData v;
  49     EXPECT_FALSE(v.is_valid());
  50     EXPECT_FALSE(v.Init(t.request, *t.response.get()));
  51     EXPECT_FALSE(v.is_valid());
  52   }
  53 }
  54
  55 TEST(HttpVaryDataTest, MultipleInit) {
  56   HttpVaryData v;
  57
  58   // Init to something valid.
  59   TestTransaction t1;
  60   t1.Init("Foo: 1\r\nbar: 23", "HTTP/1.1 200 OK\nVary: foo, bar\n\n");
  61   EXPECT_TRUE(v.Init(t1.request, *t1.response.get()));
  62   EXPECT_TRUE(v.is_valid());
  63
  64   // Now overwrite by initializing to something invalid.
  65   TestTransaction t2;
  66   t2.Init("Foo: 1\r\nbar: 23", "HTTP/1.1 200 OK\nVary: *\n\n");
  67   EXPECT_FALSE(v.Init(t2.request, *t2.response.get()));
  68   EXPECT_FALSE(v.is_valid());
  69 }
  70
  71 TEST(HttpVaryDataTest, DoesVary) {
  72   TestTransaction a;
  73   a.Init("Foo: 1", "HTTP/1.1 200 OK\nVary: foo\n\n");
  74
  75   TestTransaction b;
  76   b.Init("Foo: 2", "HTTP/1.1 200 OK\nVary: foo\n\n");
  77
  78   HttpVaryData v;
  79   EXPECT_TRUE(v.Init(a.request, *a.response.get()));
  80
  81   EXPECT_FALSE(v.MatchesRequest(b.request, *b.response.get()));
  82 }
  83
  84 TEST(HttpVaryDataTest, DoesVary2) {
  85   TestTransaction a;
  86   a.Init("Foo: 1\r\nbar: 23", "HTTP/1.1 200 OK\nVary: foo, bar\n\n");
  87
  88   TestTransaction b;
  89   b.Init("Foo: 12\r\nbar: 3", "HTTP/1.1 200 OK\nVary: foo, bar\n\n");
  90
  91   HttpVaryData v;
  92   EXPECT_TRUE(v.Init(a.request, *a.response.get()));
  93
  94   EXPECT_FALSE(v.MatchesRequest(b.request, *b.response.get()));
  95 }
  96
  97 TEST(HttpVaryDataTest, DoesntVary) {
  98   TestTransaction a;
  99   a.Init("Foo: 1", "HTTP/1.1 200 OK\nVary: foo\n\n");
 100
 101   TestTransaction b;
 102   b.Init("Foo: 1", "HTTP/1.1 200 OK\nVary: foo\n\n");
 103
 104   HttpVaryData v;
 105   EXPECT_TRUE(v.Init(a.request, *a.response.get()));
 106
 107   EXPECT_TRUE(v.MatchesRequest(b.request, *b.response.get()));
 108 }
 109
 110 TEST(HttpVaryDataTest, DoesntVary2) {
 111   TestTransaction a;
 112   a.Init("Foo: 1\r\nbAr: 2", "HTTP/1.1 200 OK\nVary: foo, bar\n\n");
 113
 114   TestTransaction b;
 115   b.Init("Foo: 1\r\nbaR: 2", "HTTP/1.1 200 OK\nVary: foo\nVary: bar\n\n");
 116
 117   HttpVaryData v;
 118   EXPECT_TRUE(v.Init(a.request, *a.response.get()));
 119
 120   EXPECT_TRUE(v.MatchesRequest(b.request, *b.response.get()));
 121 }
 122
 123 TEST(HttpVaryDataTest, ImplicitCookieForRedirect) {
 124   TestTransaction a;
 125   a.Init("Cookie: 1", "HTTP/1.1 301 Moved\nLocation: x\n\n");
 126
 127   TestTransaction b;
 128   b.Init("Cookie: 2", "HTTP/1.1 301 Moved\nLocation: x\n\n");
 129
 130   HttpVaryData v;
 131   EXPECT_TRUE(v.Init(a.request, *a.response.get()));
 132
 133   EXPECT_FALSE(v.MatchesRequest(b.request, *b.response.get()));
 134 }
 135
 136 TEST(HttpVaryDataTest, ImplicitCookieForRedirect2) {
 137   // This should be no different than the test above
 138
 139   TestTransaction a;
 140   a.Init("Cookie: 1", "HTTP/1.1 301 Moved\nLocation: x\nVary: coOkie\n\n");
 141
 142   TestTransaction b;
 143   b.Init("Cookie: 2", "HTTP/1.1 301 Moved\nLocation: x\nVary: cooKie\n\n");
 144
 145   HttpVaryData v;
 146   EXPECT_TRUE(v.Init(a.request, *a.response.get()));
 147
 148   EXPECT_FALSE(v.MatchesRequest(b.request, *b.response.get()));
 149 }
 150
 151 }  // namespace net