5418dfb19e
Summary: Rename redunant port str Add endpoint << operator Migrate everything to endpoint Reviewers: mferencevic, florijan Reviewed By: mferencevic Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1100
85 lines
2.3 KiB
C++
85 lines
2.3 KiB
C++
#include <thread>
|
|
|
|
#include "boost/archive/binary_iarchive.hpp"
|
|
#include "boost/archive/binary_oarchive.hpp"
|
|
#include "boost/archive/text_iarchive.hpp"
|
|
#include "boost/archive/text_oarchive.hpp"
|
|
#include "boost/serialization/access.hpp"
|
|
#include "boost/serialization/base_object.hpp"
|
|
#include "boost/serialization/export.hpp"
|
|
#include "gmock/gmock.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "communication/messaging/distributed.hpp"
|
|
#include "communication/rpc/rpc.hpp"
|
|
#include "gtest/gtest.h"
|
|
|
|
using communication::messaging::Message;
|
|
using communication::messaging::System;
|
|
using namespace communication::rpc;
|
|
using namespace std::literals::chrono_literals;
|
|
|
|
struct SumReq : public Message {
|
|
SumReq(int x, int y) : x(x), y(y) {}
|
|
int x;
|
|
int y;
|
|
|
|
private:
|
|
friend class boost::serialization::access;
|
|
SumReq() {} // Needed for serialization.
|
|
|
|
template <class TArchive>
|
|
void serialize(TArchive &ar, unsigned int) {
|
|
ar &boost::serialization::base_object<Message>(*this);
|
|
ar &x;
|
|
ar &y;
|
|
}
|
|
};
|
|
BOOST_CLASS_EXPORT(SumReq);
|
|
|
|
struct SumRes : public Message {
|
|
SumRes(int sum) : sum(sum) {}
|
|
int sum;
|
|
|
|
private:
|
|
friend class boost::serialization::access;
|
|
SumRes() {} // Needed for serialization.
|
|
|
|
template <class TArchive>
|
|
void serialize(TArchive &ar, unsigned int) {
|
|
ar &boost::serialization::base_object<Message>(*this);
|
|
ar ∑
|
|
}
|
|
};
|
|
BOOST_CLASS_EXPORT(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::this_thread::sleep_for(100ms);
|
|
|
|
System client_system({"127.0.0.1", 0});
|
|
Client client(client_system, server_system.endpoint(), "main");
|
|
auto sum = client.Call<Sum>(300ms, 10, 20);
|
|
EXPECT_EQ(sum->sum, 30);
|
|
}
|
|
|
|
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::this_thread::sleep_for(100ms);
|
|
|
|
System client_system({"127.0.0.1", 0});
|
|
Client client(client_system, server_system.endpoint(), "main");
|
|
auto sum = client.Call<Sum>(100ms, 10, 20);
|
|
EXPECT_FALSE(sum);
|
|
}
|