| // -*- C++ -*- |
| //===----------------------------------------------------------------------===// |
| // |
| // The LLVM Compiler Infrastructure |
| // |
| // This file is dual licensed under the MIT and the University of Illinois Open |
| // Source Licenses. See LICENSE.TXT for details. |
| // |
| //===----------------------------------------------------------------------===// |
| |
| // UNSUPPORTED: c++98, c++03, c++11 |
| |
| // <experimental/coroutine> |
| |
| // template <class Promise> |
| // struct coroutine_handle<Promise>; |
| |
| // Promise& promise() const |
| |
| #include <experimental/coroutine> |
| #include <type_traits> |
| #include <memory> |
| #include <utility> |
| #include <cstdint> |
| #include <cassert> |
| |
| #include "test_macros.h" |
| |
| namespace coro = std::experimental; |
| |
| struct MyCoro { |
| struct promise_type { |
| void unhandled_exception() {} |
| void return_void() {} |
| coro::suspend_never initial_suspend() { return {}; } |
| coro::suspend_never final_suspend() { return {}; } |
| MyCoro get_return_object() { |
| do_runtime_test(); |
| return {}; |
| } |
| void do_runtime_test() { |
| // Test that a coroutine_handle<const T> can be created from a const |
| // promise_type and that it represents the same coroutine as |
| // coroutine_handle<T> |
| using CH = coro::coroutine_handle<promise_type>; |
| using CCH = coro::coroutine_handle<const promise_type>; |
| const auto &cthis = *this; |
| CH h = CH::from_promise(*this); |
| CCH h2 = CCH::from_promise(*this); |
| CCH h3 = CCH::from_promise(cthis); |
| assert(&h.promise() == this); |
| assert(&h2.promise() == this); |
| assert(&h3.promise() == this); |
| assert(h.address() == h2.address()); |
| assert(h2.address() == h3.address()); |
| } |
| }; |
| }; |
| |
| MyCoro do_runtime_test() { |
| co_await coro::suspend_never{}; |
| } |
| |
| template <class Promise> |
| void do_test(coro::coroutine_handle<Promise>&& H) { |
| |
| // FIXME Add a runtime test |
| { |
| ASSERT_SAME_TYPE(decltype(H.promise()), Promise&); |
| LIBCPP_ASSERT_NOT_NOEXCEPT(H.promise()); |
| } |
| { |
| auto const& CH = H; |
| ASSERT_SAME_TYPE(decltype(CH.promise()), Promise&); |
| LIBCPP_ASSERT_NOT_NOEXCEPT(CH.promise()); |
| } |
| } |
| |
| int main() |
| { |
| do_test(coro::coroutine_handle<int>{}); |
| do_test(coro::coroutine_handle<const int>{}); |
| do_runtime_test(); |
| } |