brpc_http_parser_unittest.cpp 3.97 KB
Newer Older
gejun's avatar
gejun committed
1
// Copyright (c) 2014 Baidu, Inc.
gejun's avatar
gejun committed
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158
// File test_http_parser.cpp
// Date 2014/10/22 09:58:14

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

#include "base/time.h"
#include "base/logging.h"
#include "brpc/details/http_parser.h"
#include "brpc/builtin/common.h"  // AppendFileName

using brpc::http_parser;
using brpc::http_parser_init;
using brpc::http_parser_settings;

class HttpParserTest : public testing::Test {
protected:
    void SetUp() {}
    void TearDown() {}
};

TEST_F(HttpParserTest, init_perf) {
    const size_t loops = 10000000;
    base::Timer timer;
    timer.start();
    for (size_t i = 0; i < loops; ++i) {
        http_parser parser;
        http_parser_init(&parser, brpc::HTTP_REQUEST);
    }
    timer.stop();
    std::cout << "It takes " << timer.n_elapsed() / loops
              << "ns to init a http_parser"
              << std::endl;
}

int on_message_begin(http_parser *) {
    LOG(INFO) << "Start parsing message";
    return 0;
}

int on_url(http_parser *, const char *at, const size_t length) {
    LOG(INFO) << "Get url " << std::string(at, length);
    return 0;
}

int on_headers_complete(http_parser *) {
    LOG(INFO) << "Header complete";
    return 0;
}

int on_message_complete(http_parser *) {
    LOG(INFO) << "Message complete";
    return 0;
}

int on_header_field(http_parser *, const char *at, const size_t length) {
    LOG(INFO) << "Get header field " << std::string(at, length);
    return 0;
}

int on_header_value(http_parser *, const char *at, const size_t length) {
    LOG(INFO) << "Get header value " << std::string(at, length);
    return 0;
}

int on_body(http_parser *, const char *at, const size_t length) {
    LOG(INFO) << "Get body " << std::string(at, length);
    return 0;
}

TEST_F(HttpParserTest, http_example) {
    const char *http_request = 
        "GET /path/file.html?sdfsdf=sdfs HTTP/1.0\r\n"
        "From: someuser@jmarshall.com\r\n"
        "User-Agent: HTTPTool/1.0\r\n"
        "Content-Type: json\r\n"
        "Content-Length: 19\r\n"
        "Host: sdlfjslfd\r\n"
        "Accept: */*\r\n"
        "\r\n"
        "Message Body sdfsdf\r\n"
    ;
    std::cout << http_request << std::endl;

    http_parser parser;
    http_parser_init(&parser, brpc::HTTP_REQUEST);
    http_parser_settings settings;
    memset(&settings, 0, sizeof(settings));
    settings.on_message_begin = on_message_begin;
    settings.on_url = on_url;
    settings.on_headers_complete = on_headers_complete;
    settings.on_message_complete = on_message_complete;
    settings.on_header_field = on_header_field;
    settings.on_header_value = on_header_value;
    settings.on_body = on_body;
    LOG(INFO) << http_parser_execute(&parser, &settings, http_request, strlen(http_request));
}

TEST_F(HttpParserTest, append_filename) {
    std::string dir;

    dir = "/home/someone/.bsvn/..";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("/home", dir);

    dir = "/home/someone/.bsvn/../";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("/home", dir);

    dir = "/home/someone/./..";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("/", dir);
    
    dir = "/home/someone/./../";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("/", dir);

    dir = "/foo/bar";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("/foo", dir);

    dir = "/foo/bar/";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("/foo", dir);

    dir = "/foo";
    brpc::AppendFileName(&dir, ".");
    ASSERT_EQ("/foo", dir);

    dir = "/foo/";
    brpc::AppendFileName(&dir, ".");
    ASSERT_EQ("/foo/", dir);

    dir = "foo";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("", dir);

    dir = "foo/";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("", dir);

    dir = "foo/..";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("..", dir);

    dir = "foo/../";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("..", dir);
    
    dir = "/foo";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("/", dir);

    dir = "/foo/";
    brpc::AppendFileName(&dir, "..");
    ASSERT_EQ("/", dir);
}