pybind11/tests/test_class_sh_shared_ptr_copy_move.cpp

#include "pybind11_tests.h"

#include <pybind11/smart_holder.h>

#include <memory>
#include <string>
#include <vector>

namespace pybind11_tests {
namespace {

template <int SerNo>
struct Foo {
    std::string mtxt;
    Foo(const std::string &mtxt_) : mtxt(mtxt_) {}
    Foo(const Foo &other) = delete;
    Foo(Foo &&other) = delete;
    std::string get_text() const {
        std::string res = "Foo";
        if (SerNo == 0)
            res += "ShPtr_";
        else if (SerNo == 1)
            res += "SmHld_";
        return res + mtxt;
    }
};

using FooShPtr = Foo<0>;
using FooSmHld = Foo<1>;

} // namespace
} // namespace pybind11_tests

PYBIND11_TYPE_CASTER_BASE_HOLDER(pybind11_tests::FooShPtr)
PYBIND11_SMART_HOLDER_TYPE_CASTERS(pybind11_tests::FooSmHld)

namespace pybind11_tests {
namespace {

TEST_SUBMODULE(class_sh_shared_ptr_copy_move, m) {
    namespace py = pybind11;

    py::class_<FooShPtr, std::shared_ptr<FooShPtr>>(m, "FooShPtr")
        .def("get_text", &FooShPtr::get_text);
    py::classh<FooSmHld>(m, "FooSmHld")
        .def("get_text", &FooSmHld::get_text);

    m.def("test_ShPtr_copy", []() {
        auto o = std::make_shared<FooShPtr>("copy");
        auto l = py::list();
        l.append(o);
        return l;
    });
    m.def("test_SmHld_copy", []() {
        auto o = std::make_shared<FooSmHld>("copy");
        auto l = py::list();
        l.append(o);
        return l;
    });

    m.def("test_ShPtr_move", []() {
        auto o = std::make_shared<FooShPtr>("move");
        auto l = py::list();
        l.append(std::move(o));
        return l;
    });
    m.def("test_SmHld_move", []() {
        auto o = std::make_shared<FooSmHld>("move");
        auto l = py::list();
        l.append(std::move(o));
        return l;
    });
}

} // namespace
} // namespace pybind11_tests