#include <iostream>
#include <future>

#include "daggy/ThreadPool.hpp"

#include "catch.hpp"

using namespace daggy;

TEST_CASE("Threadpool Construction", "[threadpool]") {
  std::atomic<uint32_t> cnt(0);
  ThreadPool tp(10);

  std::vector<std::future<void>> res;

  SECTION("Simple runs") {
    for (size_t i = 0; i < 100; ++i)
      res.push_back(tp.addTask([&cnt]() { cnt++; return; }));
    for (auto & r : res) r.get();
    REQUIRE(cnt == 100);
  }

  SECTION("Slow runs") {
    using namespace std::chrono_literals;
    for (size_t i = 0; i < 100; ++i)
      res.push_back(tp.addTask([&cnt]() { std::this_thread::sleep_for(20ms); cnt++; return; }));
    for (auto & r : res) r.get();
    REQUIRE(cnt == 100);
  }

}