1 /*
2 tests/test_call_policies.cpp -- keep_alive and call_guard
3
4 Copyright (c) 2016 Wenzel Jakob <wenzel.jakob@epfl.ch>
5
6 All rights reserved. Use of this source code is governed by a
7 BSD-style license that can be found in the LICENSE file.
8 */
9
10 #include "pybind11_tests.h"
11
12 struct CustomGuard {
13 static bool enabled;
14
CustomGuardCustomGuard15 CustomGuard() { enabled = true; }
~CustomGuardCustomGuard16 ~CustomGuard() { enabled = false; }
17
report_statusCustomGuard18 static const char *report_status() { return enabled ? "guarded" : "unguarded"; }
19 };
20 bool CustomGuard::enabled = false;
21
22 struct DependentGuard {
23 static bool enabled;
24
DependentGuardDependentGuard25 DependentGuard() { enabled = CustomGuard::enabled; }
~DependentGuardDependentGuard26 ~DependentGuard() { enabled = false; }
27
report_statusDependentGuard28 static const char *report_status() { return enabled ? "guarded" : "unguarded"; }
29 };
30 bool DependentGuard::enabled = false;
31
TEST_SUBMODULE(call_policies,m)32 TEST_SUBMODULE(call_policies, m) {
33 // Parent/Child are used in:
34 // test_keep_alive_argument, test_keep_alive_return_value, test_alive_gc_derived,
35 // test_alive_gc_multi_derived, test_return_none, test_keep_alive_constructor
36 class Child {
37 public:
38 Child() { py::print("Allocating child."); }
39 Child(const Child &) = default;
40 Child(Child &&) = default;
41 ~Child() { py::print("Releasing child."); }
42 };
43 py::class_<Child>(m, "Child")
44 .def(py::init<>());
45
46 class Parent {
47 public:
48 Parent() { py::print("Allocating parent."); }
49 Parent(const Parent& parent) = default;
50 ~Parent() { py::print("Releasing parent."); }
51 void addChild(Child *) { }
52 Child *returnChild() { return new Child(); }
53 Child *returnNullChild() { return nullptr; }
54 static Child *staticFunction(Parent*) { return new Child(); }
55 };
56 py::class_<Parent>(m, "Parent")
57 .def(py::init<>())
58 .def(py::init([](Child *) { return new Parent(); }), py::keep_alive<1, 2>())
59 .def("addChild", &Parent::addChild)
60 .def("addChildKeepAlive", &Parent::addChild, py::keep_alive<1, 2>())
61 .def("returnChild", &Parent::returnChild)
62 .def("returnChildKeepAlive", &Parent::returnChild, py::keep_alive<1, 0>())
63 .def("returnNullChildKeepAliveChild", &Parent::returnNullChild, py::keep_alive<1, 0>())
64 .def("returnNullChildKeepAliveParent", &Parent::returnNullChild, py::keep_alive<0, 1>())
65 .def_static(
66 "staticFunction", &Parent::staticFunction, py::keep_alive<1, 0>());
67
68 m.def("free_function", [](Parent*, Child*) {}, py::keep_alive<1, 2>());
69 m.def("invalid_arg_index", []{}, py::keep_alive<0, 1>());
70
71 #if !defined(PYPY_VERSION)
72 // test_alive_gc
73 class ParentGC : public Parent {
74 public:
75 using Parent::Parent;
76 };
77 py::class_<ParentGC, Parent>(m, "ParentGC", py::dynamic_attr())
78 .def(py::init<>());
79 #endif
80
81 // test_call_guard
82 m.def("unguarded_call", &CustomGuard::report_status);
83 m.def("guarded_call", &CustomGuard::report_status, py::call_guard<CustomGuard>());
84
85 m.def("multiple_guards_correct_order", []() {
86 return CustomGuard::report_status() + std::string(" & ") + DependentGuard::report_status();
87 }, py::call_guard<CustomGuard, DependentGuard>());
88
89 m.def("multiple_guards_wrong_order", []() {
90 return DependentGuard::report_status() + std::string(" & ") + CustomGuard::report_status();
91 }, py::call_guard<DependentGuard, CustomGuard>());
92
93 #if defined(WITH_THREAD) && !defined(PYPY_VERSION)
94 // `py::call_guard<py::gil_scoped_release>()` should work in PyPy as well,
95 // but it's unclear how to test it without `PyGILState_GetThisThreadState`.
96 auto report_gil_status = []() {
97 auto is_gil_held = false;
98 if (auto tstate = py::detail::get_thread_state_unchecked())
99 is_gil_held = (tstate == PyGILState_GetThisThreadState());
100
101 return is_gil_held ? "GIL held" : "GIL released";
102 };
103
104 m.def("with_gil", report_gil_status);
105 m.def("without_gil", report_gil_status, py::call_guard<py::gil_scoped_release>());
106 #endif
107 }
108