Allow HttpServer response to include custom headers.

net/server/http_server_response_info_unittest.cc

+// Copyright (c) 2013 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 "net/http/http_status_code.h"
+#include "net/server/http_server_response_info.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace net {
+
+TEST(HttpServerResponseInfoTest, StatusLine) {
+  HttpServerResponseInfo response;
+  ASSERT_EQ(HTTP_OK, response.status_code());
+  ASSERT_EQ("HTTP/1.1 200 OK\r\n\r\n", response.Serialize());
+}
+
+TEST(HttpServerResponseInfoTest, Headers) {
+  HttpServerResponseInfo response;
+  response.AddHeader("A", "1");
+  response.AddHeader("A", "2");
+  ASSERT_EQ("HTTP/1.1 200 OK\r\nA:1\r\nA:2\r\n\r\n", response.Serialize());
+}
+
+TEST(HttpServerResponseInfoTest, Body) {
+  HttpServerResponseInfo response;
+  ASSERT_EQ(std::string(), response.body());
+  response.SetBody("body", "type");
+  ASSERT_EQ("body", response.body());
+  ASSERT_EQ(
+      "HTTP/1.1 200 OK\r\nContent-Length:4\r\nContent-Type:type\r\n\r\nbody",
+      response.Serialize());
+}
+
+TEST(HttpServerResponseInfoTest, CreateFor404) {
+  HttpServerResponseInfo response = HttpServerResponseInfo::CreateFor404();
+  ASSERT_EQ(
+      "HTTP/1.1 404 Not Found\r\n"
+      "Content-Length:0\r\nContent-Type:text/html\r\n\r\n",
+      response.Serialize());
+}
+
+TEST(HttpServerResponseInfoTest, CreateFor500) {
+  HttpServerResponseInfo response =
+      HttpServerResponseInfo::CreateFor500("mess");
+  ASSERT_EQ(
+      "HTTP/1.1 500 Internal Server Error\r\n"
+      "Content-Length:4\r\nContent-Type:text/html\r\n\r\nmess",
+      response.Serialize());
+}
+
+}  // namespace net