#define CATCH_CONFIG_MAIN

#include <catch.hpp>

#include <chrono>

#include <iostream>

#include <thread>

#include "etcd/Client.hpp"

#include "etcd/KeepAlive.hpp"

static const std::string etcd_url =

etcdv3::detail::resolve_etcd_endpoints("http://127.0.0.1:2379");

TEST_CASE("setup") {

etcd::Client etcd(etcd_url);

etcd.rmdir("/test", true).wait();

}

TEST_CASE("campaign and resign") {

etcd::Client etcd(etcd_url);

auto keepalive = etcd.leasekeepalive(60).get();

auto lease_id = keepalive->Lease();

// campaign

auto resp1 = etcd.campaign("test", lease_id, "xxxx").get();

REQUIRE(0 == resp1.error_code());

// leader

{

auto resp2 = etcd.leader("test").get();

REQUIRE(0 == resp2.error_code());

REQUIRE(resp1.value().key() == resp2.value().key());

REQUIRE("xxxx" == resp2.value().as_string());

}

// proclaim

auto resp3 = etcd.proclaim("test", lease_id, resp1.value().key(),

resp1.value().created_index(), "tttt")

.get();

REQUIRE(0 == resp3.error_code());

// leader

{

auto resp4 = etcd.leader("test").get();

REQUIRE(0 == resp4.error_code());

REQUIRE(resp1.value().key() == resp4.value().key());

REQUIRE("tttt" == resp4.value().as_string());

}

// resign

auto resp5 = etcd.resign("test", lease_id, resp1.value().key(),

resp1.value().created_index())

.get();

REQUIRE(0 == resp5.error_code());

}

TEST_CASE("campaign and observe") {

etcd::Client etcd(etcd_url);

auto keepalive = etcd.leasekeepalive(60).get();

auto lease_id = keepalive->Lease();

auto observer_thread = std::thread([&etcd]() {

std::unique_ptr<etcd::SyncClient::Observer> observer = etcd.observe("test");

// wait many change events, blocked execution

for (size_t i = 0; i < 10; ++i) {

etcd::Response resp = observer->WaitOnce();

std::cout << "observe " << resp.value().key()

<< " as the leader: " << resp.value().as_string() << std::endl;

}

std::cout << "finish the observe" << std::endl;

// cancel the observers

observer.reset(nullptr);

});

std::this_thread::sleep_for(std::chrono::seconds(1));

for (int i = 0; i < 5; ++i) {

// campaign

auto resp1 = etcd.campaign("test", lease_id, "xxxx").get();

REQUIRE(0 == resp1.error_code());

std::cout << "key " << resp1.value().key() << " becomes the leader"

<< std::endl;

// proclaim

auto resp3 = etcd.proclaim("test", lease_id, resp1.value().key(),

resp1.value().created_index(),

"tttt - " + std::to_string(i))

.get();

REQUIRE(0 == resp3.error_code());

// leader

{

auto resp4 = etcd.leader("test").get();

REQUIRE(0 == resp4.error_code());

REQUIRE(resp1.value().key() == resp4.value().key());

REQUIRE("tttt - " + std::to_string(i) == resp4.value().as_string());

}

// resign

auto resp5 = etcd.resign("test", lease_id, resp1.value().key(),

resp1.value().created_index())

.get();

REQUIRE(0 == resp5.error_code());

std::this_thread::sleep_for(std::chrono::seconds(1));

}

observer_thread.join();

}

TEST_CASE("cleanup") {

etcd::Client etcd(etcd_url);

etcd.rmdir("/test", true).get();

}