Add test case for async Python stateful callbacks

Author: uentity

tests/test_callbacks.cpp

@@ -10,6 +10,7 @@
 #include "pybind11_tests.h"
 #include "constructor_stats.h"
 #include <pybind11/functional.h>
+#include <thread>
 
 
 int dummy_function(int i) { return i + 1; }
@@ -146,4 +147,21 @@ TEST_SUBMODULE(callbacks, m) {
     py::class_<CppBoundMethodTest>(m, "CppBoundMethodTest")
         .def(py::init<>())
         .def("triple", [](CppBoundMethodTest &, int val) { return 3 * val; });
+
+    // test async Python callbacks
+    using callback_f = std::function<void(int)>;
+    m.def("test_async_callback", [](callback_f f, py::list work) {
+        // make detached thread that calls `f` with piece of work after a little delay
+        auto start_f = [f](int j){
+            auto t = std::thread([f = std::move(f), j] {
+                std::this_thread::sleep_for(std::chrono::milliseconds(50));
+                f(j);
+            });
+            t.detach();
+        };
+
+        // spawn worker threads
+        for(auto i : work)
+            start_f(py::cast<int>(i));
+    });
 }

tests/test_callbacks.py

@@ -105,3 +105,24 @@ def test_function_signatures(doc):
 
 def test_movable_object():
     assert m.callback_with_movable(lambda _: None) is True
+
+
+def test_async_callbacks():
+    # serves as state for async callback
+    class item:
+        def __init__(self, value):
+            self.value = value
+
+    # generate stateful lambda that will store result in `res`
+    res = []
+    def gen_f():
+        I = item(3)
+        return lambda j: res.append(I.value + j)
+
+    # do some work async
+    work = [1, 2, 3, 4]
+    m.test_async_callback(gen_f(), work)
+    # wait until work is done
+    from time import sleep
+    sleep(0.5)
+    assert sum(res) == sum([x + 3 for x in work])