#include "shrpx_router_test.h"
#include "munitxx.h"
#include "shrpx_router.h"
using namespace std::literals;
namespace shrpx {
namespace {
const MunitTest tests[]{
munit_void_test(test_shrpx_router_match),
munit_void_test(test_shrpx_router_match_wildcard),
munit_void_test(test_shrpx_router_match_prefix),
munit_test_end(),
};
}
const MunitSuite router_suite{
"/router", tests, nullptr, 1, MUNIT_SUITE_OPTION_NONE,
};
struct Pattern {
std::string_view pattern;
size_t idx;
bool wildcard;
};
void test_shrpx_router_match(void) {
auto patterns = std::vector<Pattern>{
{"nghttp2.org/"sv, 0},
{"nghttp2.org/alpha"sv, 1},
{"nghttp2.org/alpha/"sv, 2},
{"nghttp2.org/alpha/bravo/"sv, 3},
{"www.nghttp2.org/alpha/"sv, 4},
{"/alpha"sv, 5},
{"example.com/alpha/"sv, 6},
{"nghttp2.org/alpha/bravo2/"sv, 7},
{"www2.nghttp2.org/alpha/"sv, 8},
{"www2.nghttp2.org/alpha2/"sv, 9},
};
Router router;
for (auto &p : patterns) {
router.add_route(p.pattern, p.idx);
}
ssize_t idx;
idx = router.match("nghttp2.org"sv, "/"sv);
assert_ssize(0, ==, idx);
idx = router.match("nghttp2.org"sv, "/alpha"sv);
assert_ssize(1, ==, idx);
idx = router.match("nghttp2.org"sv, "/alpha/"sv);
assert_ssize(2, ==, idx);
idx = router.match("nghttp2.org"sv, "/alpha/charlie"sv);
assert_ssize(2, ==, idx);
idx = router.match("nghttp2.org"sv, "/alpha/bravo/"sv);
assert_ssize(3, ==, idx);
idx = router.match("nghttp2.org"sv, "/alpha/bravo"sv);
assert_ssize(3, ==, idx);
idx = router.match("www2.nghttp2.org"sv, "/alpha"sv);
assert_ssize(8, ==, idx);
idx = router.match(""sv, "/alpha"sv);
assert_ssize(5, ==, idx);
}
void test_shrpx_router_match_wildcard(void) {
constexpr auto patterns = std::to_array<Pattern>({
{"nghttp2.org/"sv, 0},
{"nghttp2.org/"sv, 1, true},
{"nghttp2.org/alpha/"sv, 2},
{"nghttp2.org/alpha/"sv, 3, true},
{"nghttp2.org/bravo"sv, 4},
{"nghttp2.org/bravo"sv, 5, true},
});
Router router;
for (auto &p : patterns) {
router.add_route(p.pattern, p.idx, p.wildcard);
}
assert_ssize(0, ==, router.match("nghttp2.org"sv, "/"sv));
assert_ssize(1, ==, router.match("nghttp2.org"sv, "/a"sv));
assert_ssize(1, ==, router.match("nghttp2.org"sv, "/charlie"sv));
assert_ssize(2, ==, router.match("nghttp2.org"sv, "/alpha"sv));
assert_ssize(2, ==, router.match("nghttp2.org"sv, "/alpha/"sv));
assert_ssize(3, ==, router.match("nghttp2.org"sv, "/alpha/b"sv));
assert_ssize(4, ==, router.match("nghttp2.org"sv, "/bravo"sv));
assert_ssize(5, ==, router.match("nghttp2.org"sv, "/bravocharlie"sv));
assert_ssize(5, ==, router.match("nghttp2.org"sv, "/bravo/"sv));
}
void test_shrpx_router_match_prefix(void) {
auto patterns = std::vector<Pattern>{
{"gro.2ptthgn."sv, 0},
{"gro.2ptthgn.www."sv, 1},
{"gro.2ptthgn.gmi."sv, 2},
{"gro.2ptthgn.gmi.ahpla."sv, 3},
};
Router router;
for (auto &p : patterns) {
router.add_route(p.pattern, p.idx);
}
ssize_t idx;
const RNode *node;
size_t nread;
node = nullptr;
idx = router.match_prefix(&nread, &node, "gro.2ptthgn.gmi.ahpla.ovarb"sv);
assert_ssize(0, ==, idx);
assert_size(12, ==, nread);
idx = router.match_prefix(&nread, &node, "gmi.ahpla.ovarb"sv);
assert_ssize(2, ==, idx);
assert_size(4, ==, nread);
idx = router.match_prefix(&nread, &node, "ahpla.ovarb"sv);
assert_ssize(3, ==, idx);
assert_size(6, ==, nread);
}
}