#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();
}