#define CATCH_CONFIG_MAIN
#include <catch.hpp>

#include <chrono>
#include <thread>

#include "etcd/Watcher.hpp"
#include "etcd/SyncClient.hpp"

static std::string etcd_uri("http://127.0.0.1:2379");
static int watcher_called = 0;

void printResponse(etcd::Response const & resp)
{
  ++watcher_called;
  std::cout << "print response called" << std::endl;
  if (resp.error_code()) {
    std::cout << resp.error_code() << ": " << resp.error_message() << std::endl;
  }
  else
  {
    std::cout << resp.action() << " " << resp.value().as_string() << std::endl;
    std::cout << "Previous value: " << resp.prev_value().as_string() << std::endl;
  }
}

TEST_CASE("create watcher with cancel")
{
  
  etcd::SyncClient etcd(etcd_uri);
  etcd.rmdir("/test", true);

  watcher_called = 0;
  etcd::Watcher watcher(etcd_uri, "/test", printResponse, true);
  std::this_thread::sleep_for(std::chrono::seconds(1));
  etcd.set("/test/key", "42");
  etcd.set("/test/key", "43");
  etcd.rm("/test/key");
  etcd.set("/test/key", "44");
  std::this_thread::sleep_for(std::chrono::seconds(1));
  CHECK(4 == watcher_called);
  watcher.Cancel();
  etcd.set("/test/key", "50");
  etcd.set("/test/key", "51");
  std::this_thread::sleep_for(std::chrono::seconds(1));
  CHECK(4 == watcher_called);

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

}

TEST_CASE("create watcher")
{
  etcd::SyncClient etcd(etcd_uri);
  etcd.rmdir("/test", true);

  watcher_called = 0;
  {
    etcd::Watcher watcher(etcd_uri, "/test", printResponse, true);
    std::this_thread::sleep_for(std::chrono::seconds(1));
    etcd.set("/test/key", "42");
    etcd.set("/test/key", "43");
  }
  
  CHECK(2 == watcher_called);
  etcd.rmdir("/test", true).error_code();
}

TEST_CASE("watch should exit normally")
{
  // cancal immediately after start watch.
  etcd::Watcher watcher(etcd_uri, "/test", printResponse, true);
  watcher.Cancel();
}

TEST_CASE("watch should can be cancelled repeatedly")
{
  // cancal immediately after start watch.
  etcd::Watcher watcher(etcd_uri, "/test", printResponse, true);
  std::vector<std::thread> threads(10);
  for (size_t i = 0; i < 10; ++i) {
    threads[i] = std::thread([&]() {
      watcher.Cancel();
    });
  }
  for (size_t i = 0; i < 10; ++i) {
    threads[i].join();
  }
}

// TEST_CASE("request cancellation")
// {
//   etcd::Client etcd(etcd_uri);
//   etcd.set("/test/key1", "42").wait();

//   pplx::task<etcd::Response> res = etcd.watch("/test/key1");
//   CHECK(!res.is_done());

//   etcd.cancel_operations();

//   std::this_thread::sleep_for(std::chrono::seconds(1));
//   REQUIRE(res.is_done());
//   try {
//     res.wait();
//   }
//   catch(pplx::task_canceled const & ex) {
//     std::cout << "pplx::task_canceled: " << ex.what() << "\n";
//   }
//   catch(std::exception const & ex) {
//     std::cout << "std::exception: " << ex.what() << "\n";
//   }
// }