ade2593b51
Summary: To clean the working directory after this diff you should execute: ``` rm src/database/counters_rpc_messages.capnp rm src/database/counters_rpc_messages.hpp rm src/database/serialization.capnp rm src/database/serialization.hpp ``` Reviewers: teon.banek, msantl Reviewed By: msantl Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1636
31 lines
861 B
C++
31 lines
861 B
C++
#include <gtest/gtest.h>
|
|
|
|
#include "communication/rpc/client_pool.hpp"
|
|
#include "communication/rpc/server.hpp"
|
|
#include "database/distributed/distributed_counters.hpp"
|
|
|
|
const std::string kLocal = "127.0.0.1";
|
|
|
|
TEST(CountersDistributed, All) {
|
|
communication::rpc::Server master_server({kLocal, 0});
|
|
database::MasterCounters master(&master_server);
|
|
communication::rpc::ClientPool master_client_pool(master_server.endpoint());
|
|
|
|
database::WorkerCounters w1(&master_client_pool);
|
|
database::WorkerCounters w2(&master_client_pool);
|
|
|
|
EXPECT_EQ(w1.Get("a"), 0);
|
|
EXPECT_EQ(w1.Get("a"), 1);
|
|
EXPECT_EQ(w2.Get("a"), 2);
|
|
EXPECT_EQ(w1.Get("a"), 3);
|
|
EXPECT_EQ(master.Get("a"), 4);
|
|
|
|
EXPECT_EQ(master.Get("b"), 0);
|
|
EXPECT_EQ(w2.Get("b"), 1);
|
|
w1.Set("b", 42);
|
|
EXPECT_EQ(w2.Get("b"), 42);
|
|
|
|
master_server.Shutdown();
|
|
master_server.AwaitShutdown();
|
|
}
|