Files
rest_rpc/tests/test_rest_rpc.cpp
T

798 lines
24 KiB
C++
Raw Normal View History

2025-09-20 08:12:32 +08:00
#define DOCTEST_CONFIG_IMPLEMENT
#include "doctest/doctest.h"
#include <asio/any_completion_handler.hpp>
2025-10-01 20:29:26 +08:00
#include <rest_rpc/rpc_client.hpp>
#include <rest_rpc/rpc_server.hpp>
2025-09-20 08:12:32 +08:00
#include <rest_rpc/traits.h>
using namespace rest_rpc;
2025-10-10 10:04:21 +08:00
struct person {
size_t id;
std::string name;
size_t age;
};
person get_person(person p) { return p; }
person modify_person(person p, std::string name) {
p.name = name;
return p;
}
2025-09-20 08:12:32 +08:00
struct dummy {
int add(int a, int b) { return a + b; }
void foo(std::string str) { std::cout << str << "\n"; }
2025-09-21 20:07:04 +08:00
std::string echo(std::string val) { return val; }
int round1(int i) { return i; }
2025-09-24 19:22:09 +08:00
asio::awaitable<void> no_arg_coro() {
std::cout << "no args\n";
co_return;
}
asio::awaitable<std::string> no_arg_coro1() {
std::cout << "no args\n";
co_return "test";
}
asio::awaitable<std::string> echo_coro(std::string str) { co_return str; }
asio::awaitable<int> add_coro(int a, int b) { co_return a + b; }
2025-09-20 08:12:32 +08:00
};
int add(int a, int b) { return a + b; }
2025-09-24 19:22:09 +08:00
asio::awaitable<int> add_coro(int a, int b) { co_return a + b; }
2025-09-20 08:12:32 +08:00
void foo(std::string str) { std::cout << str << "\n"; }
2025-09-21 20:07:04 +08:00
int round1(int i) { return i; }
std::string_view echo_sv(std::string_view str) { return str; }
2025-10-05 18:41:13 +08:00
template <auto func> asio::awaitable<void> response(auto ctx) {
2025-10-04 16:54:07 +08:00
auto ec = co_await ctx.template response_s<func>("test");
2025-09-30 11:24:41 +08:00
if (ec) {
REST_LOG_ERROR << "response error: " << ec.message();
}
2025-10-04 16:54:07 +08:00
ec = co_await ctx.template response_s<func>("test");
REST_LOG_ERROR << ec.message();
CHECK(ec);
ec = co_await ctx.response("test");
2025-09-30 11:24:41 +08:00
REST_LOG_ERROR << ec.message();
CHECK(ec);
}
2025-09-21 20:07:04 +08:00
std::string echo(std::string str) { return str; }
2025-09-24 08:29:49 +08:00
asio::awaitable<std::string> echo_coro(std::string str) { co_return str; }
2025-09-21 20:07:04 +08:00
void no_arg() { std::cout << "no args\n"; }
2025-10-08 09:05:23 +08:00
void exception_func() { throw std::invalid_argument("invalid"); }
void unknown_exception_func() { throw 1; }
2025-09-24 19:22:09 +08:00
asio::awaitable<void> no_arg_coro() {
std::cout << "no args\n";
co_return;
}
asio::awaitable<std::string> no_arg_coro1() {
std::cout << "no args\n";
co_return "test";
}
2025-10-12 10:23:42 +08:00
asio::awaitable<std::string> delay_response2(std::string_view str) {
auto coro = []() -> asio::awaitable<void> {
rpc_context ctx; // rpc_context init will be failed, it should be defined in
// io thread.
auto ret = co_await ctx.response("test");
REST_LOG_INFO << ret.message();
CHECK(ret);
};
2025-10-17 20:02:51 +08:00
sync_wait(coro());
2025-10-12 10:23:42 +08:00
2025-10-16 11:08:58 +08:00
rpc_context ctx; // right, created in io thread.
2025-10-12 10:23:42 +08:00
co_await ctx.response(str);
auto ret = co_await ctx.response(str);
CHECK(ret);
2025-10-16 11:41:21 +08:00
2025-10-16 11:08:58 +08:00
// this return value is meaningless, because it will response later, the
// return type is important for client, so just return an empty value here.
2025-10-12 10:23:42 +08:00
co_return "";
}
std::string delay_response3(std::string str) {
rpc_context ctx;
std::thread thd([ctx = std::move(ctx), str]() mutable {
sync_wait(ctx.get_executor(), ctx.response(std::move(str)));
});
thd.detach();
return "";
}
TEST_CASE("test delay response") {
using T = return_type_t<int>;
rpc_server server("127.0.0.1:9005");
server.register_handler<delay_response2>();
server.register_handler<delay_response3>();
server.async_start();
rpc_client client;
sync_wait(client.get_executor(), client.connect("127.0.0.1:9005"));
2025-10-13 20:52:27 +08:00
auto result1 = sync_wait(
client.get_executor(),
client.call_for<delay_response2>(std::chrono::minutes(2), "test"));
2025-10-12 10:23:42 +08:00
CHECK(result1.value == "test");
result1 =
sync_wait(client.get_executor(), client.call<delay_response2>("test"));
CHECK(result1.value == "test");
auto result2 = sync_wait(
client.get_executor(),
client.call_for<delay_response3>(std::chrono::minutes(2), "test"));
CHECK(result2.value == "test");
server.stop();
}
2025-10-07 17:23:53 +08:00
// TODO: client pool
2025-09-24 19:22:09 +08:00
asio::awaitable<void> test_router() {
2025-09-20 08:12:32 +08:00
rpc_router router;
router.register_handler<add>();
router.register_handler<foo>();
2025-09-21 20:07:04 +08:00
router.register_handler<round1>();
router.register_handler<echo>();
2025-09-24 19:22:09 +08:00
router.register_handler<echo_coro>();
router.register_handler<no_arg_coro>();
router.register_handler<no_arg_coro1>();
router.register_handler<add_coro>();
2025-10-08 09:05:23 +08:00
router.register_handler<no_arg>();
auto name = router.get_name_by_key(get_key<add>());
CHECK(name == "add");
router.remove_handler<add>();
name = router.get_name_by_key(get_key<add>());
CHECK(name != "add");
router.register_handler<add>();
name = router.get_name_by_key(get_key<add>());
CHECK(name == "add");
CHECK_THROWS_AS(router.register_handler<add>(), std::invalid_argument);
name = router.get_name_by_key(111);
CHECK(name == "111");
2025-09-24 19:22:09 +08:00
{
2025-10-08 09:05:23 +08:00
auto ret0 = co_await router.route(get_key<no_arg>(), "");
CHECK(ret0.ec == rpc_errc::ok);
2025-09-24 19:22:09 +08:00
auto ret = co_await router.route(get_key<no_arg_coro>(), "");
CHECK(ret.ec == rpc_errc::ok);
auto ret1 = co_await router.route(get_key<no_arg_coro1>(), "");
CHECK(ret1.ec == rpc_errc::ok);
auto ret2 = co_await router.route(get_key<echo_coro>(), "test");
CHECK(ret2.ec == rpc_errc::ok);
2025-10-08 09:05:23 +08:00
router.register_handler<exception_func>();
router.register_handler<unknown_exception_func>();
auto ret3 = co_await router.route(get_key<exception_func>(), "");
CHECK(ret3.ec == rpc_errc::function_exception);
auto ret4 = co_await router.route(get_key<unknown_exception_func>(), "");
CHECK(ret4.ec == rpc_errc::function_unknown_exception);
auto ret5 = co_await router.route(111, "");
CHECK(ret5.ec == rpc_errc::no_such_function);
2025-09-24 19:22:09 +08:00
}
2025-09-20 08:12:32 +08:00
dummy d{};
router.register_handler<&dummy::add>(&d);
router.register_handler<&dummy::foo>(&d);
2025-09-21 20:07:04 +08:00
router.register_handler<&dummy::round1>(&d);
router.register_handler<&dummy::echo>(&d);
2025-09-24 19:22:09 +08:00
router.register_handler<&dummy::no_arg_coro>(&d);
router.register_handler<&dummy::no_arg_coro1>(&d);
router.register_handler<&dummy::echo_coro>(&d);
router.register_handler<&dummy::add_coro>(&d);
{
auto ret = co_await router.route(get_key<&dummy::no_arg_coro>(), "");
CHECK(ret.ec == rpc_errc::ok);
auto ret1 = co_await router.route(get_key<&dummy::no_arg_coro1>(), "");
CHECK(ret1.ec == rpc_errc::ok);
auto ret2 = co_await router.route(get_key<&dummy::echo_coro>(), "test");
CHECK(ret2.ec == rpc_errc::ok);
}
2025-09-21 20:07:04 +08:00
{
2025-10-13 16:02:20 +08:00
auto s = rpc_codec::pack_args(1);
auto s1 = rpc_codec::pack_args("test");
2025-09-24 19:22:09 +08:00
auto r = co_await router.route(get_key<round1>(), s);
auto r1 = co_await router.route(get_key<echo>(), s1);
2025-09-21 20:07:04 +08:00
2025-09-24 19:22:09 +08:00
auto r2 = co_await router.route(get_key<&dummy::round1>(), s);
auto r3 = co_await router.route(get_key<&dummy::echo>(), s1);
2025-09-21 20:07:04 +08:00
std::cout << "\n";
}
2025-10-13 16:02:20 +08:00
auto args = rpc_codec::pack_args(1, 2);
2025-09-20 08:12:32 +08:00
std::string_view str(args.data(), args.size());
2025-09-24 19:22:09 +08:00
{
auto r1 = co_await router.route(get_key<add_coro>(), str);
auto r2 = co_await router.route(get_key<&dummy::add_coro>(), str);
std::cout << "\n";
}
2025-10-13 16:02:20 +08:00
auto args1 = rpc_codec::pack_args("it is a test");
2025-09-20 08:12:32 +08:00
std::string_view str1(args1.data(), args1.size());
{
2025-09-24 19:22:09 +08:00
auto result = co_await router.route(get_key<&dummy::add>(), str);
2025-10-13 16:02:20 +08:00
auto r = rpc_codec::unpack<int>(result.result);
2025-09-24 19:22:09 +08:00
auto result1 = co_await router.route(get_key<&dummy::foo>(), str1);
2025-09-20 08:12:32 +08:00
CHECK(r == 3);
CHECK(result1.ec == rpc_errc::ok);
}
2025-09-24 19:22:09 +08:00
auto result = co_await router.route(get_key<add>(), str);
2025-09-20 08:12:32 +08:00
CHECK(result.ec == rpc_errc::ok);
2025-09-24 19:22:09 +08:00
auto result1 = co_await router.route(get_key<foo>(), str1);
2025-09-20 08:12:32 +08:00
CHECK(result1.ec == rpc_errc::ok);
}
2025-10-17 20:02:51 +08:00
TEST_CASE("test router") { sync_wait(test_router()); }
2025-09-24 19:22:09 +08:00
2025-10-17 12:26:20 +08:00
asio::awaitable<void>
get_last_rwtime_coro(std::shared_ptr<rpc_connection> conn) {
conn->set_last_time();
asio::steady_timer timer(co_await asio::this_coro::executor);
timer.expires_after(std::chrono::milliseconds(100));
co_await timer.async_wait(asio::use_awaitable);
conn->set_last_time();
auto time = co_await conn->get_last_rwtime();
auto ms = std::chrono::duration_cast<std::chrono::milliseconds>(
time.time_since_epoch());
std::cout << "Time in coroutine: " << ms.count() << "ms\n";
}
2025-10-09 16:48:40 +08:00
TEST_CASE("test rpc_connection") {
uint64_t conn_id = 999;
size_t num_thread = 4;
asio::io_context io_ctx;
tcp_socket socket(io_ctx);
2025-10-17 12:26:20 +08:00
bool cross_ending = false;
2025-10-09 16:48:40 +08:00
rpc_router router;
2025-10-10 10:04:21 +08:00
router.register_handler<get_person>();
2025-10-12 10:51:23 +08:00
2025-10-10 10:04:21 +08:00
person p{1, "tom", 20};
2025-10-13 20:52:27 +08:00
auto buf = rpc_codec::pack_args(std::tuple(p));
2025-10-10 10:04:21 +08:00
auto ret = sync_wait(get_global_executor(),
router.route(get_key<get_person>(), buf));
2025-10-13 20:52:27 +08:00
auto tp = rpc_codec::unpack<person>(ret.data());
2025-10-09 16:48:40 +08:00
dummy d{};
router.register_handler<&dummy::add>(&d);
auto conn = std::make_shared<rpc_connection>(std::move(socket), conn_id,
2025-10-17 12:26:20 +08:00
router, cross_ending);
2025-10-09 16:48:40 +08:00
CHECK_EQ(conn->id(), conn_id);
conn->set_check_timeout(true);
2025-10-17 12:26:20 +08:00
asio::co_spawn(io_ctx, get_last_rwtime_coro(conn), asio::detached);
io_ctx.run();
2025-10-09 16:48:40 +08:00
std::cout << "topic_id: " << conn->topic_id() << std::endl;
conn->close();
2025-10-17 12:26:20 +08:00
io_ctx.stop();
2025-10-09 16:48:40 +08:00
}
2025-10-08 20:07:43 +08:00
TEST_CASE("test context pool") {
io_context_pool pool(0);
CHECK(pool.size() == 1);
}
2025-09-20 08:12:32 +08:00
TEST_CASE("test server start") {
2025-10-01 20:29:26 +08:00
rpc_server server("127.0.0.1:9005");
2025-09-20 08:12:32 +08:00
server.register_handler<add>();
server.register_handler<foo>();
2025-09-21 20:07:04 +08:00
server.register_handler<no_arg>();
server.register_handler<echo>();
server.register_handler<echo_sv>();
server.register_handler<round1>();
2025-10-10 10:04:21 +08:00
server.register_handler<get_person>();
server.register_handler<modify_person>();
2025-09-30 10:17:07 +08:00
server.set_conn_max_age(std::chrono::seconds(5));
server.set_check_conn_interval(std::chrono::seconds(1));
2025-09-20 08:12:32 +08:00
auto ec = server.async_start();
CHECK(!ec);
2025-10-01 20:29:26 +08:00
rpc_client cl;
2025-09-21 20:07:04 +08:00
static_assert(util::CharArrayRef<char const(&)[5]>);
static_assert(util::CharArray<const char[5]>);
2025-10-13 16:02:20 +08:00
rpc_codec::pack_args();
rpc_codec::pack_args(1, 2);
auto s1 = rpc_codec::pack_args("test");
auto s2 = rpc_codec::pack_args(std::string_view("test2"));
auto s3 = rpc_codec::pack_args(std::string("test2"));
auto s5 = rpc_codec::pack_args(123);
2025-09-21 20:07:04 +08:00
2025-09-20 08:12:32 +08:00
auto conn_ec = sync_wait(cl.get_executor(), cl.connect("127.0.0.1:9005"));
2025-10-16 11:08:58 +08:00
CHECK(!conn_ec);
2025-10-16 11:41:21 +08:00
2025-09-24 08:29:49 +08:00
{
2025-09-24 19:22:09 +08:00
auto result = sync_wait(cl.get_executor(),
cl.call_for<add>(std::chrono::minutes(2), 1, 2));
2025-09-24 08:29:49 +08:00
CHECK(result.ec == rpc_errc::ok);
2025-10-10 10:04:21 +08:00
CHECK(result.value == 3);
}
{
person p{1, "tom", 20};
auto result = sync_wait(
cl.get_executor(), cl.call_for<get_person>(std::chrono::minutes(2), p));
CHECK(result.ec == rpc_errc::ok);
auto result1 = sync_wait(
2025-10-14 08:43:40 +08:00
cl.get_executor(), cl.call_for<modify_person>(std::chrono::minutes(2),
p, std::string("jack")));
2025-10-10 10:04:21 +08:00
CHECK(result1.ec == rpc_errc::ok);
CHECK(result1.value.name == "jack");
2025-09-24 08:29:49 +08:00
}
2025-09-21 20:07:04 +08:00
{
auto result =
sync_wait(cl.get_executor(),
cl.call_for<echo_sv>(std::chrono::minutes(2), "test"));
CHECK(result.ec == rpc_errc::ok);
auto result1 = sync_wait(
cl.get_executor(), cl.call_for<echo>(std::chrono::minutes(2), "test"));
CHECK(result1.ec == rpc_errc::ok);
2025-10-08 20:07:43 +08:00
cl.enable_tcp_no_delay(false);
2025-09-21 20:07:04 +08:00
auto result2 = sync_wait(cl.get_executor(), cl.call<round1>(1));
CHECK(result2.ec == rpc_errc::ok);
}
2025-09-20 08:12:32 +08:00
2025-10-08 20:07:43 +08:00
{
auto result =
sync_wait(cl.get_executor(),
cl.call_for<add>(std::chrono::milliseconds(0), 1, 2));
CHECK(result.ec == rpc_errc::request_timeout);
}
2025-09-20 08:12:32 +08:00
ec = server.async_start();
CHECK(!ec);
ec = server.async_start();
CHECK(!ec);
server.stop();
server.stop();
2025-10-08 20:07:43 +08:00
{
auto result = sync_wait(cl.get_executor(), cl.call<add>(1, 2));
CHECK(result.ec != rpc_errc::ok);
}
2025-09-20 08:12:32 +08:00
ec = server.async_start();
CHECK(ec);
std::cout << ec.message() << "\n";
2025-10-01 20:29:26 +08:00
rpc_server server1("127.0.0.1:9007");
2025-09-20 08:12:32 +08:00
ec = server1.async_start();
CHECK(!ec);
2025-10-01 20:29:26 +08:00
rpc_server server2("127.0.0.1:9007");
2025-09-20 08:12:32 +08:00
ec = server2.async_start();
CHECK(ec);
std::cout << ec.message() << "\n";
ec = server2.async_start();
CHECK(ec);
std::cout << ec.message() << "\n";
// if server has stopped, start server will be failed.
2025-10-01 20:29:26 +08:00
rpc_server server3("127.0.0.1:9005");
2025-09-20 08:12:32 +08:00
server3.stop();
CHECK(server3.has_stopped());
2025-09-30 10:17:07 +08:00
ec = server3.async_start();
2025-09-20 08:12:32 +08:00
CHECK(ec);
std::cout << ec.message() << "\n";
}
2025-10-08 09:13:02 +08:00
TEST_CASE("test cross ending") {
2025-10-08 20:07:43 +08:00
rpc_server server("127.0.0.1:9004");
server.register_handler<echo>();
server.enable_cross_ending(true);
server.async_start();
2025-10-08 09:13:02 +08:00
2025-10-08 20:07:43 +08:00
rpc_client client{};
client.enable_cross_ending(true);
sync_wait(get_global_executor(), client.connect("127.0.0.1:9004"));
auto ret = sync_wait(client.get_executor(), client.call<echo>("test"));
CHECK(ret.value == "test");
2025-10-08 09:13:02 +08:00
}
2025-10-05 18:41:13 +08:00
TEST_CASE("test pub sub") {
rpc_server server("127.0.0.1:9004");
server.async_start();
rpc_client client{};
sync_wait(get_global_executor(), client.connect("127.0.0.1:9004"));
std::promise<void> promise;
auto sub = [&]() -> asio::awaitable<void> {
for (int i = 0; i < 5; i++) {
2025-10-06 13:01:10 +08:00
auto result = co_await client.subscribe<std::string>("topic1");
REST_LOG_INFO << result.value;
CHECK(result.ec == rpc_errc::ok);
CHECK(result.value == "publish message");
2025-10-05 18:41:13 +08:00
}
2025-10-08 20:07:43 +08:00
2025-10-05 18:41:13 +08:00
promise.set_value();
};
asio::co_spawn(client.get_executor(), sub(), asio::detached);
std::this_thread::sleep_for(std::chrono::seconds(2));
auto pub = [&]() -> asio::awaitable<void> {
for (int i = 0; i < 5; i++) {
co_await server.publish("topic1", "publish message");
}
};
asio::co_spawn(client.get_executor(), pub(), asio::detached);
promise.get_future().wait();
}
2025-10-17 12:26:20 +08:00
TEST_CASE("test sync pub sub") {
rpc_server server("127.0.0.1:9004");
server.async_start();
rpc_client client{};
sync_wait(get_global_executor(), client.connect("127.0.0.1:9004"));
std::promise<void> promise;
auto sub = [&]() -> asio::awaitable<void> {
for (int i = 0; i < 5; i++) {
auto result = co_await client.subscribe<std::string>("topic1");
REST_LOG_INFO << result.value;
CHECK(result.ec == rpc_errc::ok);
CHECK(result.value == "publish message");
}
promise.set_value();
};
asio::co_spawn(client.get_executor(), sub(), asio::detached);
std::this_thread::sleep_for(std::chrono::seconds(2));
for (int i = 0; i < 5; i++) {
server.sync_publish("topic1", "publish message");
}
promise.get_future().wait();
}
2025-10-06 21:28:28 +08:00
TEST_CASE("test reconnect") {
rpc_server server("127.0.0.1:9004");
server.async_start();
rpc_server server1("127.0.0.1:9005");
2025-10-09 17:47:42 +08:00
server1.set_check_conn_interval(std::chrono::milliseconds(100));
2025-10-13 20:52:27 +08:00
server1.set_conn_max_age(std::chrono::milliseconds(400));
2025-10-06 21:28:28 +08:00
server1.async_start();
rpc_client client{};
auto ec = sync_wait(get_global_executor(), client.connect("127.0.0.1:9004"));
CHECK(!ec);
ec = sync_wait(get_global_executor(), client.connect("127.0.0.1:9005"));
CHECK(!ec);
2025-10-08 20:07:43 +08:00
2025-10-09 17:47:42 +08:00
CHECK(server1.connection_count() == 1);
std::this_thread::sleep_for(std::chrono::milliseconds(600));
// expired connection has been removed.
CHECK(server1.connection_count() == 0);
ec = sync_wait(get_global_executor(), client.connect("127.0.0.1:9999"));
2025-10-08 20:07:43 +08:00
CHECK(ec);
REST_LOG_INFO << ec.message();
ec = sync_wait(
get_global_executor(),
client.connect("127.0.0.0:9006", std::chrono::milliseconds(200)));
CHECK(ec);
REST_LOG_INFO << ec.message();
ec =
sync_wait(get_global_executor(),
client.connect("127.0.0.x:9006", std::chrono::milliseconds(0)));
CHECK(ec);
REST_LOG_INFO << ec.message();
ec = sync_wait(
get_global_executor(),
client.connect("127.0.0.x:9006", std::chrono::milliseconds(200)));
CHECK(ec);
REST_LOG_INFO << ec.message();
2025-10-06 21:28:28 +08:00
}
2025-10-09 17:47:42 +08:00
TEST_CASE("test server address") {
rpc_server server("127.0.0.1", "9004");
server.enable_tcp_no_delay(false);
auto ec = server.async_start();
CHECK(!ec);
server.register_handler<add>();
server.remove_handler("add");
CHECK_NOTHROW(server.register_handler<add>());
server.remove_handler<add>();
CHECK_NOTHROW(server.register_handler<add>());
rpc_server server1("127.0.0.x:9004");
ec = server1.async_start();
CHECK(ec);
REST_LOG_INFO << ec.message();
rpc_server server2("127.0.0.1:9005");
std::thread thd([&] { server2.start(); });
std::this_thread::sleep_for(std::chrono::milliseconds(500));
std::thread thd2([&] { server2.stop(); });
thd2.join();
thd.join();
}
2025-10-17 12:26:20 +08:00
TEST_CASE("test wrong client address") {
rpc_client client{};
auto ec0 = sync_wait(get_global_executor(), client.connect("127.0.0.1"));
CHECK(ec0);
auto ec1 = sync_wait(get_global_executor(), client.connect("127.0.0.1:"));
CHECK(ec1);
auto ec2 = sync_wait(get_global_executor(), client.connect(":9005"));
CHECK(ec2);
auto ec3 = sync_wait(get_global_executor(), client.connect(":"));
CHECK(ec3);
}
TEST_CASE("test_string_view_array_has_func") {
constexpr std::array<std::string_view, 3> arr = {"hello", "world", "test"};
auto result = string_view_array_has(arr, "hello");
CHECK_EQ(result, "hello");
result = string_view_array_has(arr, "hello world");
CHECK_EQ(result, "hello");
result = string_view_array_has(arr, "say hello");
CHECK_EQ(result, "");
result = string_view_array_has(arr, "goodbye");
CHECK_EQ(result, "");
result = string_view_array_has(arr, "");
CHECK_EQ(result, "");
// empty arr
constexpr std::array<std::string_view, 0> empty_arr{};
auto result1 = string_view_array_has(empty_arr, "hello");
CHECK_EQ(result1, "");
result1 = string_view_array_has(empty_arr, "");
CHECK_EQ(result1, "");
}
#ifdef _MSC_VER
void __cdecl test_cdecl_func() {}
void __stdcall test_stdcall_func() {}
void __fastcall test_fastcall_func() {}
#else
void test_cdecl_func() {}
void test_stdcall_func() {}
void test_fastcall_func() {}
#endif
void normal_func() {}
namespace rest_rpc {
template <> struct qualified_name_of<test_cdecl_func> {
static constexpr std::string_view value = "__cdecl test_cdecl_func";
};
2025-10-09 17:47:42 +08:00
2025-10-17 12:26:20 +08:00
template <> struct qualified_name_of<test_stdcall_func> {
static constexpr std::string_view value = "__stdcall test_stdcall_func";
};
template <> struct qualified_name_of<test_fastcall_func> {
static constexpr std::string_view value = "__fastcall test_fastcall_func";
};
template <> struct qualified_name_of<normal_func> {
static constexpr std::string_view value = "normal_func";
};
struct TestStruct {};
template <> struct qualified_name_of<TestStruct{}> {
static constexpr std::string_view value = "__thiscall TestStruct::method";
};
template <> struct qualified_name_of<nullptr> {
static constexpr std::string_view value = "";
};
} // namespace rest_rpc
TEST_CASE("test_get_func_name") {
constexpr auto cdecl_name = get_func_name<test_cdecl_func>();
CHECK(cdecl_name == "test_cdecl_func");
constexpr auto stdcall_name = get_func_name<test_stdcall_func>();
CHECK(stdcall_name == "test_stdcall_func");
constexpr auto fastcall_name = get_func_name<test_fastcall_func>();
CHECK(fastcall_name == "test_fastcall_func");
constexpr auto normal_name = get_func_name<normal_func>();
CHECK(normal_name == "normal_func");
constexpr auto thiscall_name = get_func_name<rest_rpc::TestStruct{}>();
CHECK(thiscall_name == "TestStruct::method");
constexpr auto empty_name = get_func_name<nullptr>();
CHECK(empty_name == "");
}
TEST_CASE("test_error_code") {
const auto &cat = rest_rpc::category();
CHECK_EQ(cat.name(), "rest_rpc_error");
SUBCASE("rpc_errc::ok") {
CHECK_EQ(cat.message(static_cast<int>(rest_rpc::rpc_errc::ok)), "ok");
}
SUBCASE("rpc_errc::no_such_function") {
CHECK_EQ(
cat.message(static_cast<int>(rest_rpc::rpc_errc::no_such_function)),
"no such function");
}
SUBCASE("rpc_errc::no_such_key") {
CHECK_EQ(cat.message(static_cast<int>(rest_rpc::rpc_errc::no_such_key)),
"resolve failed");
}
SUBCASE("rpc_errc::invalid_req_type") {
CHECK_EQ(
cat.message(static_cast<int>(rest_rpc::rpc_errc::invalid_req_type)),
"invalid request type");
}
SUBCASE("rpc_errc::function_exception") {
CHECK_EQ(
cat.message(static_cast<int>(rest_rpc::rpc_errc::function_exception)),
"logic function exception happend");
}
SUBCASE("rpc_errc::function_unknown_exception") {
CHECK_EQ(cat.message(static_cast<int>(
rest_rpc::rpc_errc::function_unknown_exception)),
"unknown function exception happend");
}
SUBCASE("rpc_errc::invalid_argument") {
CHECK_EQ(
cat.message(static_cast<int>(rest_rpc::rpc_errc::invalid_argument)),
"invalid argument");
}
SUBCASE("rpc_errc::write_error") {
CHECK_EQ(cat.message(static_cast<int>(rest_rpc::rpc_errc::write_error)),
"write failed");
}
SUBCASE("rpc_errc::read_error") {
CHECK_EQ(cat.message(static_cast<int>(rest_rpc::rpc_errc::read_error)),
"read failed");
}
SUBCASE("rpc_errc::socket_closed") {
CHECK_EQ(cat.message(static_cast<int>(rest_rpc::rpc_errc::socket_closed)),
"socket closed");
}
SUBCASE("rpc_errc::resolve_timeout") {
CHECK_EQ(cat.message(static_cast<int>(rest_rpc::rpc_errc::resolve_timeout)),
"resolve timeout");
}
SUBCASE("rpc_errc::connection_timeout") {
CHECK_EQ(
cat.message(static_cast<int>(rest_rpc::rpc_errc::connection_timeout)),
"connect timeout");
}
SUBCASE("rpc_errc::request_timeout") {
CHECK_EQ(cat.message(static_cast<int>(rest_rpc::rpc_errc::request_timeout)),
"request timeout");
}
SUBCASE("rpc_errc::protocol_error") {
CHECK_EQ(cat.message(static_cast<int>(rest_rpc::rpc_errc::protocol_error)),
"protocol error");
}
SUBCASE("rpc_errc::has_response") {
CHECK_EQ(cat.message(static_cast<int>(rest_rpc::rpc_errc::has_response)),
"has response, duplicate response is not allowed");
}
SUBCASE("rpc_errc::duplicate_topic") {
CHECK_EQ(cat.message(static_cast<int>(rest_rpc::rpc_errc::duplicate_topic)),
"duplicate topic");
}
SUBCASE("rpc_errc::rpc_context_init_failed") {
CHECK_EQ(cat.message(
static_cast<int>(rest_rpc::rpc_errc::rpc_context_init_failed)),
"the rpc context init failed, it must be created in rpc handler "
"io thread, otherwise will init failed");
}
SUBCASE("negative error code") { CHECK_EQ(cat.message(-1), "unknown error"); }
SUBCASE("out of bounds positive error code") {
CHECK_EQ(cat.message(100), "unknown error");
}
SUBCASE("creates error_code with correct value and category") {
auto ec = rest_rpc::make_error_code(rest_rpc::rpc_errc::no_such_function);
CHECK_EQ(ec.value(),
static_cast<int>(rest_rpc::rpc_errc::no_such_function));
CHECK_EQ(ec.category().name(), "rest_rpc_error");
CHECK(ec);
}
SUBCASE("error_code for ok evaluates to true in boolean context") {
auto ec_ok = rest_rpc::make_error_code(rest_rpc::rpc_errc::ok);
CHECK_FALSE(ec_ok); // zero error code should be false in boolean context
auto ec_error = rest_rpc::make_error_code(rest_rpc::rpc_errc::read_error);
CHECK(ec_error); // non-zero error code should be true in boolean context
}
SUBCASE("operator== with matching error code") {
auto ec = rest_rpc::make_error_code(rest_rpc::rpc_errc::write_error);
CHECK(ec == rest_rpc::rpc_errc::write_error);
}
SUBCASE("operator== with non-matching error code") {
auto ec = rest_rpc::make_error_code(rest_rpc::rpc_errc::read_error);
CHECK_FALSE(ec == rest_rpc::rpc_errc::write_error);
}
SUBCASE("operator== with ok error code") {
auto ec_ok = rest_rpc::make_error_code(rest_rpc::rpc_errc::ok);
CHECK(ec_ok == rest_rpc::rpc_errc::ok);
}
}
2025-10-14 20:19:39 +08:00
// bool in_user_pack = false;
// bool in_user_unpack = false;
// namespace user_codec {
// // adl lookup in user_codec namespace
// template <typename... Args>
// std::string serialize(rest_adl_tag, Args &&...args) {
// in_user_pack = true;
// msgpack::sbuffer buffer(2 * 1024);
// if constexpr (sizeof...(Args) > 1) {
2025-10-16 11:41:21 +08:00
// msgpack::pack(buffer,
// std::forward_as_tuple(std::forward<Args>(args)...));
2025-10-14 20:19:39 +08:00
// } else {
// msgpack::pack(buffer, std::forward<Args>(args)...);
// }
// return std::string(buffer.data(), buffer.size());
// }
// template <typename T> T deserialize(rest_adl_tag, std::string_view data) {
// try {
// in_user_unpack = true;
// static msgpack::unpacked msg;
// msgpack::unpack(msg, data.data(), data.size());
// return msg.get().as<T>();
// } catch (...) {
// return T{};
// }
// }
// } // namespace user_codec
// TEST_CASE("test user codec") {
// auto buf = rpc_codec::pack_args(
// std::make_tuple<int, std::string, int>(1, "tom", 20));
// CHECK(in_user_pack);
// std::string_view str(buf.data(), buf.size());
// rpc_codec::unpack<std::tuple<int, std::string, int>>(str);
// CHECK(in_user_unpack);
// }
2025-10-13 16:02:20 +08:00
2025-09-20 08:12:32 +08:00
// doctest comments
// 'function' : must be 'attribute' - see issue #182
DOCTEST_MSVC_SUPPRESS_WARNING_WITH_PUSH(4007)
int main(int argc, char **argv) { return doctest::Context(argc, argv).run(); }
2025-10-06 19:28:29 +08:00
DOCTEST_MSVC_SUPPRESS_WARNING_POP