#include "catch.hpp"

#include <coro/coro.hpp>

TEST_CASE("sync_wait simple integer return")
{
    auto func = []() -> coro::task<int> {
        co_return 11;
    };

    auto result = coro::sync_wait(func());
    REQUIRE(result == 11);
}

TEST_CASE("sync_wait void")
{
    std::string output;

    auto func = [&]() -> coro::task<void> {
        output = "hello from sync_wait<void>\n";
        co_return;
    };

    coro::sync_wait(func());
    REQUIRE(output == "hello from sync_wait<void>\n");
}

TEST_CASE("sync_wait task co_await single")
{
    auto answer = []() -> coro::task<int> {
        std::cerr << "\tThinking deep thoughts...\n";
        co_return 42;
    };

    auto await_answer = [&]() -> coro::task<int> {
        std::cerr << "\tStarting to wait for answer.\n";
        auto a = answer();
        std::cerr << "\tGot the coroutine, getting the value.\n";
        auto v = co_await a;
        std::cerr << "\tCoroutine value is " << v << "\n";
        REQUIRE(v == 42);
        v = co_await a;
        std::cerr << "\tValue is still " << v << "\n";
        REQUIRE(v == 42);
        co_return 1337;
    };

    auto output = coro::sync_wait(await_answer());
    REQUIRE(output == 1337);
}