diff --git a/tests/test_rest_rpc.cpp b/tests/test_rest_rpc.cpp index b1f76ef..22e3856 100644 --- a/tests/test_rest_rpc.cpp +++ b/tests/test_rest_rpc.cpp @@ -29,21 +29,21 @@ TEST_CASE("test_client_default_constructor") { } TEST_CASE("test_client_async_connect") { - rpc_server server(9000, std::thread::hardware_concurrency()); - dummy d; - server.register_handler("add", &dummy::add, &d); - server.async_run(); - std::this_thread::sleep_for(std::chrono::milliseconds(300)); + rpc_server server(9000, std::thread::hardware_concurrency()); + dummy d; + server.register_handler("add", &dummy::add, &d); + server.async_run(); + std::this_thread::sleep_for(std::chrono::milliseconds(300)); - rpc_client client; - client.async_connect("127.0.0.1", 9000); + rpc_client client; + client.async_connect("127.0.0.1", 9000); - try { - auto result = client.call("add", 1, 2); - CHECK_EQ(result, 3); - } catch (const std::exception &ex) { - std::cout << ex.what() << std::endl; - } + try { + auto result = client.call("add", 1, 2); + CHECK_EQ(result, 3); + } catch (const std::exception &ex) { + std::cout << ex.what() << std::endl; + } } TEST_CASE("test_sync_add") { @@ -74,34 +74,33 @@ void hello(rpc_conn conn, const std::string &str) { std::cout << "hello " << str << std::endl; } TEST_CASE("test_client_async_call") { - rpc_server server(9000, std::thread::hardware_concurrency()); - server.register_handler("get_person", get_person); - server.register_handler("hello", hello); - server.async_run(); - std::this_thread::sleep_for(std::chrono::milliseconds(300)); + rpc_server server(9000, std::thread::hardware_concurrency()); + server.register_handler("get_person", get_person); + server.register_handler("hello", hello); + server.async_run(); + std::this_thread::sleep_for(std::chrono::milliseconds(300)); - rpc_client client("127.0.0.1", 9000); - bool r = client.connect(); - CHECK(r); + rpc_client client("127.0.0.1", 9000); + bool r = client.connect(); + CHECK(r); - client.set_error_callback( - [](asio::error_code ec) { std::cout << ec.message() << std::endl; }); + client.set_error_callback( + [](asio::error_code ec) { std::cout << ec.message() << std::endl; }); - auto f = client.async_call("get_person"); - if (f.wait_for(std::chrono::milliseconds(50)) == - std::future_status::timeout) { - std::cout << "timeout" << std::endl; - } else { - auto p = f.get().as(); - CHECK_EQ(p.id, 1); - CHECK_EQ(p.age, 20); - CHECK_EQ(p.name, "tom"); - } - auto fu = client.async_call("hello", "purecpp"); - fu.get().as(); // no return + auto f = client.async_call("get_person"); + if (f.wait_for(std::chrono::milliseconds(50)) == + std::future_status::timeout) { + std::cout << "timeout" << std::endl; + } else { + auto p = f.get().as(); + CHECK_EQ(p.id, 1); + CHECK_EQ(p.age, 20); + CHECK_EQ(p.name, "tom"); + } + auto fu = client.async_call("hello", "purecpp"); + fu.get().as(); // no return } - TEST_CASE("test_client_reconnect_and_heartbeat") { rpc_client client; @@ -117,8 +116,7 @@ TEST_CASE("test_client_reconnect_and_heartbeat") { if (client.has_connected()) { flag = true; break; - } - else { + } else { // std::cout << "connected fail\n"; cnt++; } @@ -126,7 +124,7 @@ TEST_CASE("test_client_reconnect_and_heartbeat") { flag = false; break; } - + std::this_thread::sleep_for(std::chrono::seconds(1)); } CHECK(flag);