#include <gtest/gtest.h>
#include <httplib.h>
#include <future>
#include <iostream>

using namespace std;
using namespace httplib;

TEST(SplitTest, ParseQueryString)
{
    string s = "key1=val1&key2=val2&key3=val3";
    map<string, string> dic;

    split(&s[0], &s[s.size()], '&', [&](const char* b, const char* e) {
        string key, val;
        split(b, e, '=', [&](const char* b, const char* e) {
            if (key.empty()) {
                key.assign(b, e);
            } else {
                val.assign(b, e);
            }
        });
        dic[key] = val;
    });

    EXPECT_EQ("val1", dic["key1"]);
    EXPECT_EQ("val2", dic["key2"]);
    EXPECT_EQ("val3", dic["key3"]);
}

TEST(ParseQueryTest, ParseQueryString)
{
    string s = "key1=val1&key2=val2&key3=val3";
    map<string, string> dic;

    parse_query_text(&s[0], &s[s.size()], dic);

    EXPECT_EQ("val1", dic["key1"]);
    EXPECT_EQ("val2", dic["key2"]);
    EXPECT_EQ("val3", dic["key3"]);
}

TEST(SocketTest, OpenClose)
{
    socket_t sock = create_server_socket("localhost", 1914);
    ASSERT_NE(-1, sock);

    auto ret = close_socket(sock);
    EXPECT_EQ(0, ret);
}

TEST(GetHeaderValueTest, DefaultValue)
{
    MultiMap map = {{"Dummy","Dummy"}};
    auto val = get_header_value_text(map, "Content-Type", "text/plain");
    ASSERT_STREQ("text/plain", val);
}

TEST(GetHeaderValueTest, DefaultValueInt)
{
    MultiMap map = {{"Dummy","Dummy"}};
    auto val = get_header_value_int(map, "Content-Length", 100);
    EXPECT_EQ(100, val);
}

TEST(GetHeaderValueTest, RegularValue)
{
    MultiMap map = {{"Content-Type","text/html"}, {"Dummy", "Dummy"}};
    auto val = get_header_value_text(map, "Content-Type", "text/plain");
    ASSERT_STREQ("text/html", val);
}

TEST(GetHeaderValueTest, RegularValueInt)
{
    MultiMap map = {{"Content-Length","100"}, {"Dummy", "Dummy"}};
    auto val = get_header_value_int(map, "Content-Length", 0);
    EXPECT_EQ(100, val);
}

class ServerTest : public ::testing::Test {
protected:
    ServerTest() : svr_(host_, port_) {
        persons_["john"] = "programmer";
    }

    virtual void SetUp() {
        svr_.get("/hi", [&](httplib::Connection& c) {
            c.response.set_content("Hello World!", "text/plain");
        });
        svr_.get("/", [&](httplib::Connection& c) {
            c.response.set_redirect("/hi");
        });
        svr_.post("/person", [&](httplib::Connection& c) {
            const auto& req = c.request;
            if (req.has_param("name") && req.has_param("note")) {
                persons_[req.params.at("name")] = req.params.at("note");
            } else {
                c.response.status = 400;
            }
        });
        svr_.get("/person/(.*)", [&](httplib::Connection& c) {
            const auto& req = c.request;
            std::string name = req.matches[1];
            if (persons_.find(name) != persons_.end()) {
                auto note = persons_[name];
                c.response.set_content(note, "text/plain");
            } else {
                c.response.status = 404;
            }
        });
        f_ = async([&](){ svr_.run(); });
    }

    virtual void TearDown() {
        svr_.stop();
        f_.get();
    }

    const char*                        host_ = "localhost";
    int                                port_ = 1914;
    std::map<std::string, std::string> persons_;
    Server                             svr_;
    std::future<void>                  f_;
};

TEST_F(ServerTest, GetMethod200)
{
    Response res;
    bool ret = Client(host_, port_).get("/hi", res);
    ASSERT_EQ(true, ret);
    EXPECT_EQ(200, res.status);
    EXPECT_EQ("text/plain", res.get_header_value("Content-Type"));
    EXPECT_EQ("Hello World!", res.body);
}

TEST_F(ServerTest, GetMethod302)
{
    Response res;
    bool ret = Client(host_, port_).get("/", res);
    ASSERT_EQ(true, ret);
    EXPECT_EQ(302, res.status);
    EXPECT_EQ("/hi", res.get_header_value("Location"));
}

TEST_F(ServerTest, GetMethod404)
{
    Response res;
    bool ret = Client(host_, port_).get("/invalid", res);
    ASSERT_EQ(true, ret);
    EXPECT_EQ(404, res.status);
}

TEST_F(ServerTest, GetMethodPersonJohn)
{
    Response res;
    bool ret = Client(host_, port_).get("/person/john", res);
    ASSERT_EQ(true, ret);
    EXPECT_EQ(200, res.status);
    EXPECT_EQ("text/plain", res.get_header_value("Content-Type"));
    EXPECT_EQ("programmer", res.body);
}

TEST_F(ServerTest, PostMethod)
{
    {
        Response res;
        bool ret = Client(host_, port_).get("/person/john2", res);
        ASSERT_EQ(true, ret);
        ASSERT_EQ(404, res.status);
    }
    {
        auto content = "name=john2&note=coder";
        auto content_type = "application/x-www-form-urlencoded";
        Response res;
        bool ret = Client(host_, port_).post("/person", content, content_type, res);
        ASSERT_EQ(true, ret);
        ASSERT_EQ(200, res.status);
    }
    {
        Response res;
        bool ret = Client(host_, port_).get("/person/john2", res);
        ASSERT_EQ(true, ret);
        ASSERT_EQ(200, res.status);
        ASSERT_EQ("text/plain", res.get_header_value("Content-Type"));
        ASSERT_EQ("coder", res.body);
    }
}

// vim: et ts=4 sw=4 cin cino={1s ff=unix