/*

    tests/test_async.cpp -- __await__ support

    Copyright (c) 2019 Google Inc.

    All rights reserved. Use of this source code is governed by a

    BSD-style license that can be found in the LICENSE file.

*/

#include "pybind11_tests.h"

TEST_SUBMODULE(async_module, m) {

    struct DoesNotSupportAsync {};

    py::class_<DoesNotSupportAsync>(m, "DoesNotSupportAsync")

        .def(py::init<>());

    struct SupportsAsync {};

    py::class_<SupportsAsync>(m, "SupportsAsync")

        .def(py::init<>())

        .def("__await__", [](const SupportsAsync& self) -> py::object {

            static_cast<void>(self);

            py::object loop = py::module::import("asyncio.events").attr("get_event_loop")();

            py::object f = loop.attr("create_future")();

            f.attr("set_result")(5);

            return f.attr("__await__")();

        });

}