blob: e6e01d72c92ff80bb95cc2df153f18bfddf03825 [file] [log] [blame]
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")
struct SupportsAsync {};
py::class_<SupportsAsync>(m, "SupportsAsync")
.def("__await__", [](const SupportsAsync& self) -> py::object {
py::object loop = py::module_::import("").attr("get_event_loop")();
py::object f = loop.attr("create_future")();
return f.attr("__await__")();