cf7d9070d8
Summary: In this diff I just wanted to fix tests' flakyness. We can discuss if we want to always pass endpoint as an argument and never pass address:port pair explicitly. However if we decide that, I will do that change in another diff. Reviewers: dgleich, florijan Reviewed By: dgleich, florijan Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1049
83 lines
2.2 KiB
C++
83 lines
2.2 KiB
C++
#include "gmock/gmock.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#include <thread>
|
|
|
|
#include "communication/messaging/distributed.hpp"
|
|
#include "communication/rpc/rpc.hpp"
|
|
#include "gtest/gtest.h"
|
|
|
|
using communication::messaging::System;
|
|
using communication::messaging::Message;
|
|
using namespace communication::rpc;
|
|
using namespace std::literals::chrono_literals;
|
|
|
|
struct SumReq : public Message {
|
|
SumReq() {} // cereal needs this
|
|
SumReq(int x, int y) : x(x), y(y) {}
|
|
int x;
|
|
int y;
|
|
|
|
template <class Archive>
|
|
void serialize(Archive &ar) {
|
|
ar(cereal::virtual_base_class<Message>(this), x, y);
|
|
}
|
|
};
|
|
CEREAL_REGISTER_TYPE(SumReq);
|
|
|
|
struct SumRes : public Message {
|
|
SumRes() {} // cereal needs this
|
|
SumRes(int sum) : sum(sum) {}
|
|
int sum;
|
|
|
|
template <class Archive>
|
|
void serialize(Archive &ar) {
|
|
ar(cereal::virtual_base_class<Message>(this), sum);
|
|
}
|
|
};
|
|
CEREAL_REGISTER_TYPE(SumRes);
|
|
using Sum = RequestResponse<SumReq, SumRes>;
|
|
|
|
TEST(Rpc, Call) {
|
|
System server_system("127.0.0.1", 0);
|
|
Server server(server_system, "main");
|
|
server.Register<Sum>([](const SumReq &request) {
|
|
return std::make_unique<SumRes>(request.x + request.y);
|
|
});
|
|
std::thread server_thread([&] { server.Start(); });
|
|
std::this_thread::sleep_for(100ms);
|
|
|
|
System client_system("127.0.0.1", 0);
|
|
Client client(client_system, "127.0.0.1", server_system.endpoint().port(),
|
|
"main");
|
|
auto sum = client.Call<Sum>(300ms, 10, 20);
|
|
EXPECT_EQ(sum->sum, 30);
|
|
|
|
server.Shutdown();
|
|
server_thread.join();
|
|
server_system.Shutdown();
|
|
client_system.Shutdown();
|
|
}
|
|
|
|
TEST(Rpc, Timeout) {
|
|
System server_system("127.0.0.1", 0);
|
|
Server server(server_system, "main");
|
|
server.Register<Sum>([](const SumReq &request) {
|
|
std::this_thread::sleep_for(300ms);
|
|
return std::make_unique<SumRes>(request.x + request.y);
|
|
});
|
|
std::thread server_thread([&] { server.Start(); });
|
|
std::this_thread::sleep_for(100ms);
|
|
|
|
System client_system("127.0.0.1", 0);
|
|
Client client(client_system, "127.0.0.1", server_system.endpoint().port(),
|
|
"main");
|
|
auto sum = client.Call<Sum>(100ms, 10, 20);
|
|
EXPECT_FALSE(sum);
|
|
|
|
server.Shutdown();
|
|
server_thread.join();
|
|
server_system.Shutdown();
|
|
client_system.Shutdown();
|
|
}
|