1 files changed, 85 insertions, 0 deletions
diff --git a/libs/plugin_python/tests/python/test_qt_widgets.cpp b/libs/plugin_python/tests/python/test_qt_widgets.cpp
new file mode 100644
index 0000000..6d8a901
--- /dev/null
+++ b/libs/plugin_python/tests/python/test_qt_widgets.cpp
@@ -0,0 +1,85 @@
+#include "pybind11_qt/pybind11_qt.h"
+
+#include <pybind11/pybind11.h>
+
+#include <QMap>
+#include <QWidget>
+
+namespace py = pybind11;
+using namespace pybind11::literals;
+
+QMap<QString, QWidget*> s_Widgets;
+QWidget* s_Parent;
+
+class CustomWidget : public QWidget {
+public:
+    CustomWidget(QString const& name, QWidget* parent = nullptr) : QWidget(parent)
+    {
+        s_Widgets[name] = this;
+        setProperty("name", name);
+    }
+
+    ~CustomWidget() { s_Widgets.remove(property("name").toString()); }
+};
+
+class PyCustomWidget : public CustomWidget {
+public:
+    using CustomWidget::CustomWidget;
+    int heightForWidth(int value) const
+    {
+        PYBIND11_OVERRIDE(int, CustomWidget, heightForWidth, value);
+    }
+};
+
+PYBIND11_MODULE(qt_widgets, m)
+{
+    s_Parent = new QWidget();
+
+    py::class_<CustomWidget, PyCustomWidget, py::qt::qobject_holder<CustomWidget>>
+        pyCustomWidget(m, "CustomWidget");
+    pyCustomWidget
+        .def(py::init<QString, QWidget*>(), "name"_a, "parent"_a = (QWidget*)nullptr)
+        .def("set_parent_cpp", [](CustomWidget* w) {
+            w->setParent(s_Parent);
+        });
+    py::qt::add_qt_delegate<QWidget>(pyCustomWidget, "_widget");
+
+    m.def("is_alive", [](QString const& name) {
+        return s_Widgets.contains(name);
+    });
+
+    m.def("get", [](QString const& name) {
+        return s_Widgets.contains(name) ? s_Widgets[name] : nullptr;
+    });
+
+    m.def("set_parent", [](QWidget* widget) {
+        widget->setParent(s_Parent);
+    });
+
+    m.def("is_owned_cpp", [](QString const& name) {
+        return s_Widgets.contains(name) && s_Widgets[name]->parent() == s_Parent;
+    });
+
+    m.def("make_widget_own_cpp", [](QString const& name) -> QWidget* {
+        return new CustomWidget(name, s_Parent);
+    });
+
+    m.def(
+        "make_widget_own_py",
+        [](QString const& name) -> QWidget* {
+            return new CustomWidget(name);
+        },
+        py::return_value_policy::take_ownership);
+
+    // simply passing the widget gives the ownership of the Python object to C++
+    m.def("send_to_cpp", [](QString const& name, QWidget* widget) {
+        widget->setProperty("name", name);
+        widget->setParent(s_Parent);
+        s_Widgets[name] = widget;
+    });
+
+    m.def("heightForWidth", [](QString const& name, int value) {
+        return s_Widgets.contains(name) ? s_Widgets[name]->heightForWidth(value)
+                                        : -1024;
+    });
+}