/*
 * Copyright (c) 2015 Oleg Morozenkov
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <boost/test/unit_test.hpp>

#include <tgbot/net/HttpParser.h>

#include "utils.h"

using namespace std;
using namespace TgBot;

BOOST_AUTO_TEST_SUITE(tHttpParser)

BOOST_AUTO_TEST_CASE(generateRequest) {
	vector<HttpReqArg> args = { HttpReqArg("email", "test@example.com"), HttpReqArg("text", "Hello, world!") };
	string t = HttpParser::getInstance().generateRequest(Url("http://example.com/index.html"), args, true);
	string e = ""
		"POST /index.html HTTP/1.1\r\n"
		"Host: example.com\r\n"
		"Connection: keep-alive\r\n"
		"Content-Type: application/x-www-form-urlencoded\r\n"
		"Content-Length: 49\r\n"
		"\r\n"
		"email=test%40example.com&text=Hello%2c%20world%21";
	BOOST_CHECK_MESSAGE(t == e, diffS(t, e));
}

BOOST_AUTO_TEST_CASE(generateMultipartFormData) {
	vector<HttpReqArg> args = { HttpReqArg("email", "test@example.com"), HttpReqArg("text", "Hello, world!", true) };
	string boundary = HttpParser::getInstance().generateMultipartBoundary(args);
	string t = HttpParser::getInstance().generateMultipartFormData(args, boundary);
	string e = ""
		"--" + boundary + "\r\n"
		"Content-Disposition: form-data; name=\"email\"\r\n"
		"\r\n"
		"test@example.com\r\n"
		"--" + boundary + "\r\n"
		"Content-Disposition: form-data; name=\"text\"\r\n"
		"Content-Type: text/plain\r\n"
		"\r\n"
		"Hello, world!\r\n";
	BOOST_CHECK_MESSAGE(t == e, diffS(t, e));
}

BOOST_AUTO_TEST_CASE(generateWwwFormUrlencoded) {
	vector<HttpReqArg> args = { HttpReqArg("email", "test@example.com"), HttpReqArg("text", "Hello, world!") };
	string t = HttpParser::getInstance().generateWwwFormUrlencoded(args);
	string e = "email=test%40example.com&text=Hello%2c%20world%21";
	BOOST_CHECK_MESSAGE(t == e, diffS(t, e));
}

BOOST_AUTO_TEST_CASE(generateResponse) {
	string t = HttpParser::getInstance().generateResponse("testdata");
	string e = ""
		"HTTP/1.1 200 OK\r\n"
		"Content-Type: text/plain\r\n"
		"Content-Length: 8\r\n"
		"\r\n"
		"testdata";
	BOOST_CHECK_MESSAGE(t == e, diffS(t, e));
}

BOOST_AUTO_TEST_CASE(parseRequest) {
	string data = ""
		"POST /index.html HTTP/1.1\r\n"
		"Host: example.com\r\n"
		"Connection: keep-alive\r\n"
		"Content-Type: text/plain\r\n"
		"Content-Length: 8\r\n"
		"\r\n"
		"testdata";

	map<string, string> tHeaders;
	string t = HttpParser::getInstance().parseRequest(data, tHeaders);

	map<string, string> eHeaders = {
		{ "method", "POST" },
		{ "path", "/index.html" },
		{ "host", "example.com" },
		{ "connection", "keep-alive" },
		{ "content-type", "text/plain" },
		{ "content-length", "8" }
	};
	string e = "testdata";

	BOOST_CHECK_MESSAGE(t == e, diffS(t, e));
	BOOST_CHECK_MESSAGE(tHeaders == eHeaders, diff(tHeaders, eHeaders, [](const pair<const string, string>& item) -> string {
		return item.first + '=' + item.second;
	}));
}

BOOST_AUTO_TEST_CASE(parseResponse) {
	string data = ""
		"HTTP/1.1 200 OK\r\n"
		"Content-Type: text/plain\r\n"
		"Content-Length: 8\r\n"
		"\r\n"
		"testdata";

	map<string, string> tHeaders;
	string t = HttpParser::getInstance().parseResponse(data, tHeaders);

	map<string, string> eHeaders = {
		{ "status", "200" },
		{ "content-type", "text/plain" },
		{ "content-length", "8" }
	};
	string e = "testdata";

	BOOST_CHECK_MESSAGE(t == e, diffS(t, e));
	BOOST_CHECK_MESSAGE(tHeaders == eHeaders, diff(tHeaders, eHeaders, [](const pair<const string, string>& item) -> string {
		return item.first + '=' + item.second;
	}));
}

BOOST_AUTO_TEST_SUITE_END()