#define private public

#include "server_http.hpp"
#include "client_http.hpp"
#include <iostream>

using namespace std;
using namespace SimpleWeb;

class ServerTest : public ServerBase<HTTP> {
public:
    ServerTest() : 
            ServerBase<HTTP>::ServerBase(8080, 1, 5, 300) {}
            
    void accept() {}
    
    bool parse_request_test() {
        std::shared_ptr<Request> request(new Request());
        
        stringstream ss;
        ss << "GET /test/ HTTP/1.1\r\n";
        ss << "TestHeader: test\r\n";
        ss << "TestHeader2:test2\r\n";
        ss << "\r\n";
        
        parse_request(request, ss);
        
        if(request->method!="GET")
            return 0;
        if(request->path!="/test/")
            return 0;
        if(request->http_version!="1.1")
            return 0;
        
        if(request->header.size()!=2)
            return 0;
        if(request->header.count("TestHeader")==0)
            return 0;
        if(request->header["TestHeader"]!="test")
            return 0;

        if(request->header.count("TestHeader2")==0)
            return 0;
        if(request->header["TestHeader2"]!="test2")
            return 0;
        
        return 1;
    }
};

class ClientTest : public ClientBase<HTTP> {
public:
    ClientTest(const std::string& server_port_path) : ClientBase<HTTP>::ClientBase(server_port_path, 80) {}
    
    void connect() {}
    
    bool constructor_parse_test1() {
        if(host!="test.org")
            return 0;
        if(port!=8080)
            return 0;
        
        return 1;
    }
    
    bool constructor_parse_test2() {
        if(host!="test.org")
            return 0;
        if(port!=80)
            return 0;
        
        return 1;
    }
    
    bool parse_response_header_test() {
        std::shared_ptr<Response> response(new Response());
        
        stringstream ss;
        ss << "HTTP/1.1 200 OK\r\n";
        ss << "TestHeader: test\r\n";
        ss << "TestHeader2:test2\r\n";
        ss << "\r\n";
        
        parse_response_header(response, ss);
        
        if(response->http_version!="1.1")
            return 0;
        if(response->status_code!="200 OK")
            return 0;
        
        if(response->header.size()!=2)
            return 0;
        if(response->header.count("TestHeader")==0)
            return 0;
        if(response->header["TestHeader"]!="test")
            return 0;

        if(response->header.count("TestHeader2")==0)
            return 0;
        if(response->header["TestHeader2"]!="test2")
            return 0;
        
        return 1;
    }
};

int main(int argc, char** argv) {
    ServerTest serverTest;
    
    if(!serverTest.parse_request_test()) {
        cerr << "FAIL Server::parse_request" << endl;
        return 1;
    }
    
    ClientTest clientTest("test.org:8080");
    if(!clientTest.constructor_parse_test1()) {
        cerr << "FAIL Client::Client" << endl;
        return 1;
    }
    
    ClientTest clientTest2("test.org");
    if(!clientTest2.constructor_parse_test2()) {
        cerr << "FAIL Client::Client" << endl;
        return 1;
    }
    
    if(!clientTest2.parse_response_header_test()) {
        cerr << "FAIL Client::parse_response_header" << endl;
        return 1;
    }
    
    return 0;
}