response_test.cc

#include "gtest/gtest.h"
#include "response.h"


class ResponseDefaultResponseTest : public ::testing::Test {
protected:

    void VerifyContents(Response::ResponseCode status) {
        r = Response::DefaultResponse(status);

        ASSERT_EQ(status, r.GetStatus());
        ASSERT_EQ(DefaultResponse::ToHtml(r.GetStatus()), r.GetBody());
        ASSERT_EQ(2, r.GetHeaders().size());
        EXPECT_EQ("Content-Length", r.GetHeaders()[0].first);
        EXPECT_EQ(std::to_string(r.GetBody().size()), r.GetHeaders()[0].second);
        EXPECT_EQ("Content-Type", r.GetHeaders()[1].first);
        EXPECT_EQ("text/html", r.GetHeaders()[1].second);
    }

    Response r;
};


TEST_F(ResponseDefaultResponseTest, OkStatus) {
    VerifyContents(Response::ResponseCode::ok);
}


TEST_F(ResponseDefaultResponseTest, CreatedStatus) {
    VerifyContents(Response::ResponseCode::created);
}


TEST_F(ResponseDefaultResponseTest, AcceptedStatus) {
    VerifyContents(Response::ResponseCode::accepted);
}


TEST_F(ResponseDefaultResponseTest, NoContentStatus) {
    VerifyContents(Response::ResponseCode::no_content);
}


TEST_F(ResponseDefaultResponseTest, MovedPermanentlyStatus) {
    VerifyContents(Response::ResponseCode::moved_permanently);
}


TEST_F(ResponseDefaultResponseTest, MovedTemporarilyStatus) {
    VerifyContents(Response::ResponseCode::moved_temporarily);
}


TEST_F(ResponseDefaultResponseTest, NotModifiedStatus) {
    VerifyContents(Response::ResponseCode::not_modified);
}


TEST_F(ResponseDefaultResponseTest, BadRequestStatus) {
    VerifyContents(Response::ResponseCode::bad_request);
}


TEST_F(ResponseDefaultResponseTest, UnauthorizedStatus) {
    VerifyContents(Response::ResponseCode::unauthorized);
}


TEST_F(ResponseDefaultResponseTest, ForbiddenStatus) {
    VerifyContents(Response::ResponseCode::forbidden);
}


TEST_F(ResponseDefaultResponseTest, NotFoundStatus) {
    VerifyContents(Response::ResponseCode::not_found);
}


TEST_F(ResponseDefaultResponseTest, InternalServerErrorStatus) {
    VerifyContents(Response::ResponseCode::internal_server_error);
}


TEST_F(ResponseDefaultResponseTest, NotImplementederrorStatus) {
    VerifyContents(Response::ResponseCode::not_implemented);
}


TEST_F(ResponseDefaultResponseTest, BadGatewayStatus) {
    VerifyContents(Response::ResponseCode::bad_gateway);
}


TEST_F(ResponseDefaultResponseTest, ServiceUnavailableStatus) {
    VerifyContents(Response::ResponseCode::service_unavailable);
}


TEST(ResponseTest, html) {
    Response r = Response::HtmlResponse(
        DefaultResponse::ToHtml(Response::ResponseCode::ok));

    ASSERT_EQ(Response::ResponseCode::ok, r.GetStatus());
    ASSERT_EQ(DefaultResponse::ToHtml(Response::ResponseCode::ok), r.GetBody());
    ASSERT_EQ(2, r.GetHeaders().size());
    EXPECT_EQ("Content-Length", r.GetHeaders()[0].first);
    EXPECT_EQ(std::to_string(r.GetBody().size()), r.GetHeaders()[0].second);
    EXPECT_EQ("Content-Type", r.GetHeaders()[1].first);
    EXPECT_EQ("text/html", r.GetHeaders()[1].second);
}


TEST(ResponseTest, plain) {
    Response r = Response::PlainTextResponse(
        DefaultResponse::ToHtml(Response::ResponseCode::ok));

    ASSERT_EQ(Response::ResponseCode::ok, r.GetStatus());
    ASSERT_EQ(DefaultResponse::ToHtml(Response::ResponseCode::ok), r.GetBody());
    ASSERT_EQ(2, r.GetHeaders().size());
    EXPECT_EQ("Content-Length", r.GetHeaders()[0].first);
    EXPECT_EQ(std::to_string(r.GetBody().size()), r.GetHeaders()[0].second);
    EXPECT_EQ("Content-Type", r.GetHeaders()[1].first);
    EXPECT_EQ("text/plain", r.GetHeaders()[1].second);
}


TEST(ToStringTest, generaltest) {
    Response r = Response::PlainTextResponse(
        DefaultResponse::ToHtml(Response::ResponseCode::ok));

    ASSERT_EQ(Response::ResponseCode::ok, r.GetStatus());
    ASSERT_EQ(DefaultResponse::ToHtml(Response::ResponseCode::ok), r.GetBody());
    ASSERT_EQ(2, r.GetHeaders().size());
    EXPECT_EQ("Content-Length", r.GetHeaders()[0].first);
    EXPECT_EQ(std::to_string(r.GetBody().size()), r.GetHeaders()[0].second);
    EXPECT_EQ("Content-Type", r.GetHeaders()[1].first);
    EXPECT_EQ("text/plain", r.GetHeaders()[1].second);


    EXPECT_EQ("HTTP/1.0 200 OK\r\nContent-Length: 0\r\nContent-Type: text/plain\r\n\r\n", r.ToString());
}