http_request_headers_unittest.cc 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172
  1. // Copyright (c) 2012 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. #include "net/http/http_request_headers.h"
  5. #include <memory>
  6. #include "base/values.h"
  7. #include "net/log/net_log_capture_mode.h"
  8. #include "testing/gtest/include/gtest/gtest.h"
  9. namespace net {
  10. namespace {
  11. TEST(HttpRequestHeaders, HasHeader) {
  12. HttpRequestHeaders headers;
  13. headers.SetHeader("Foo", "bar");
  14. EXPECT_TRUE(headers.HasHeader("foo"));
  15. EXPECT_TRUE(headers.HasHeader("Foo"));
  16. EXPECT_FALSE(headers.HasHeader("Fo"));
  17. const HttpRequestHeaders& headers_ref = headers;
  18. EXPECT_TRUE(headers_ref.HasHeader("foo"));
  19. EXPECT_TRUE(headers_ref.HasHeader("Foo"));
  20. EXPECT_FALSE(headers_ref.HasHeader("Fo"));
  21. }
  22. TEST(HttpRequestHeaders, SetHeader) {
  23. HttpRequestHeaders headers;
  24. headers.SetHeader("Foo", "bar");
  25. EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
  26. }
  27. TEST(HttpRequestHeaders, SetMultipleHeaders) {
  28. HttpRequestHeaders headers;
  29. headers.SetHeader("Cookie-Monster", "Nom nom nom");
  30. headers.SetHeader("Domo-Kun", "Loves Chrome");
  31. EXPECT_EQ("Cookie-Monster: Nom nom nom\r\nDomo-Kun: Loves Chrome\r\n\r\n",
  32. headers.ToString());
  33. }
  34. TEST(HttpRequestHeaders, SetHeaderTwice) {
  35. HttpRequestHeaders headers;
  36. headers.SetHeader("Foo", "bar");
  37. headers.SetHeader("Foo", "bar");
  38. EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
  39. }
  40. TEST(HttpRequestHeaders, SetHeaderTwiceCaseInsensitive) {
  41. HttpRequestHeaders headers;
  42. headers.SetHeader("Foo", "bar");
  43. headers.SetHeader("FoO", "Bar");
  44. EXPECT_EQ("Foo: Bar\r\n\r\n", headers.ToString());
  45. }
  46. TEST(HttpRequestHeaders, SetHeaderTwiceSamePrefix) {
  47. HttpRequestHeaders headers;
  48. headers.SetHeader("FooBar", "baz");
  49. headers.SetHeader("Foo", "qux");
  50. EXPECT_EQ("FooBar: baz\r\nFoo: qux\r\n\r\n", headers.ToString());
  51. const HttpRequestHeaders& headers_ref = headers;
  52. EXPECT_EQ("FooBar: baz\r\nFoo: qux\r\n\r\n", headers_ref.ToString());
  53. }
  54. TEST(HttpRequestHeaders, SetEmptyHeader) {
  55. HttpRequestHeaders headers;
  56. headers.SetHeader("Foo", "Bar");
  57. headers.SetHeader("Bar", "");
  58. EXPECT_EQ("Foo: Bar\r\nBar: \r\n\r\n", headers.ToString());
  59. }
  60. TEST(HttpRequestHeaders, SetHeaderIfMissing) {
  61. HttpRequestHeaders headers;
  62. headers.SetHeaderIfMissing("Foo", "Bar");
  63. EXPECT_EQ("Foo: Bar\r\n\r\n", headers.ToString());
  64. headers.SetHeaderIfMissing("Foo", "Baz");
  65. EXPECT_EQ("Foo: Bar\r\n\r\n", headers.ToString());
  66. }
  67. TEST(HttpRequestHeaders, RemoveHeader) {
  68. HttpRequestHeaders headers;
  69. headers.SetHeader("Foo", "bar");
  70. headers.RemoveHeader("Foo");
  71. EXPECT_EQ("\r\n", headers.ToString());
  72. }
  73. TEST(HttpRequestHeaders, RemoveHeaderMissingHeader) {
  74. HttpRequestHeaders headers;
  75. headers.SetHeader("Foo", "bar");
  76. headers.RemoveHeader("Bar");
  77. EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
  78. }
  79. TEST(HttpRequestHeaders, RemoveHeaderCaseInsensitive) {
  80. HttpRequestHeaders headers;
  81. headers.SetHeader("Foo", "bar");
  82. headers.SetHeader("All-Your-Base", "Belongs To Chrome");
  83. headers.RemoveHeader("foo");
  84. EXPECT_EQ("All-Your-Base: Belongs To Chrome\r\n\r\n", headers.ToString());
  85. }
  86. TEST(HttpRequestHeaders, AddHeaderFromString) {
  87. HttpRequestHeaders headers;
  88. headers.AddHeaderFromString("Foo: bar");
  89. EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
  90. }
  91. TEST(HttpRequestHeaders, AddHeaderFromStringNoLeadingWhitespace) {
  92. HttpRequestHeaders headers;
  93. headers.AddHeaderFromString("Foo:bar");
  94. EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
  95. }
  96. TEST(HttpRequestHeaders, AddHeaderFromStringMoreLeadingWhitespace) {
  97. HttpRequestHeaders headers;
  98. headers.AddHeaderFromString("Foo: \t \t bar");
  99. EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
  100. }
  101. TEST(HttpRequestHeaders, AddHeaderFromStringTrailingWhitespace) {
  102. HttpRequestHeaders headers;
  103. headers.AddHeaderFromString("Foo: bar \t \t ");
  104. EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
  105. }
  106. TEST(HttpRequestHeaders, AddHeaderFromStringLeadingTrailingWhitespace) {
  107. HttpRequestHeaders headers;
  108. headers.AddHeaderFromString("Foo: \t bar\t ");
  109. EXPECT_EQ("Foo: bar\r\n\r\n", headers.ToString());
  110. }
  111. TEST(HttpRequestHeaders, AddHeaderFromStringWithEmptyValue) {
  112. HttpRequestHeaders headers;
  113. headers.AddHeaderFromString("Foo:");
  114. EXPECT_EQ("Foo: \r\n\r\n", headers.ToString());
  115. }
  116. TEST(HttpRequestHeaders, AddHeaderFromStringWithWhitespaceValue) {
  117. HttpRequestHeaders headers;
  118. headers.AddHeaderFromString("Foo: ");
  119. EXPECT_EQ("Foo: \r\n\r\n", headers.ToString());
  120. }
  121. TEST(HttpRequestHeaders, MergeFrom) {
  122. HttpRequestHeaders headers;
  123. headers.SetHeader("A", "A");
  124. headers.SetHeader("B", "B");
  125. HttpRequestHeaders headers2;
  126. headers2.SetHeader("B", "b");
  127. headers2.SetHeader("C", "c");
  128. headers.MergeFrom(headers2);
  129. EXPECT_EQ("A: A\r\nB: b\r\nC: c\r\n\r\n", headers.ToString());
  130. }
  131. TEST(HttpRequestHeaders, CopyFrom) {
  132. HttpRequestHeaders headers;
  133. headers.SetHeader("A", "A");
  134. headers.SetHeader("B", "B");
  135. HttpRequestHeaders headers2;
  136. headers2.SetHeader("B", "b");
  137. headers2.SetHeader("C", "c");
  138. headers.CopyFrom(headers2);
  139. EXPECT_EQ("B: b\r\nC: c\r\n\r\n", headers.ToString());
  140. }
  141. } // namespace
  142. } // namespace net