pybind11/tests/test_class_sh_shared_from_this.py

114 lines
3.4 KiB
Python

# -*- coding: utf-8 -*-
import pytest
from pybind11_tests import class_sh_shared_from_this as m
from pybind11_tests import ConstructorStats
def test_smart_ptr(capture):
# Object3
for i, o in zip(
[9, 8, 9], [m.MyObject3(9), m.make_myobject3_1(), m.make_myobject3_2()]
):
print(o)
with capture:
m.print_myobject3_1(o)
m.print_myobject3_2(o)
m.print_myobject3_3(o)
m.print_myobject3_3(o) # XXX XXX XXX print_myobject3_4
assert capture == "MyObject3[{i}]\n".format(i=i) * 4
cstats = ConstructorStats.get(m.MyObject3)
assert cstats.alive() == 1
o = None
assert cstats.alive() == 0
assert cstats.values() == ["MyObject3[9]", "MyObject3[8]", "MyObject3[9]"]
assert cstats.default_constructions == 0
assert cstats.copy_constructions == 0
# assert cstats.move_constructions >= 0 # Doesn't invoke any
assert cstats.copy_assignments == 0
assert cstats.move_assignments == 0
def test_shared_from_this_ref():
s = m.SharedFromThisRef()
stats = ConstructorStats.get(m.B)
assert stats.alive() == 2
ref = s.ref # init_holder_helper(holder_ptr=false, owned=false, bad_wp=false)
assert stats.alive() == 2
assert s.set_ref(ref)
assert s.set_holder(
ref
) # std::enable_shared_from_this can create a holder from a reference
del ref, s
assert stats.alive() == 0
def test_shared_from_this_bad_wp():
s = m.SharedFromThisRef()
stats = ConstructorStats.get(m.B)
assert stats.alive() == 2
bad_wp = s.bad_wp # init_holder_helper(holder_ptr=false, owned=false, bad_wp=true)
assert stats.alive() == 2
assert s.set_ref(bad_wp)
# with pytest.raises(RuntimeError) as excinfo:
if 1:
assert s.set_holder(bad_wp)
# assert "Unable to cast from non-held to held instance" in str(excinfo.value)
del bad_wp, s
assert stats.alive() == 0
def test_shared_from_this_copy():
s = m.SharedFromThisRef()
stats = ConstructorStats.get(m.B)
assert stats.alive() == 2
copy = s.copy # init_holder_helper(holder_ptr=false, owned=true, bad_wp=false)
# RuntimeError: Invalid return_value_policy for shared_ptr.
assert stats.alive() == 3
assert s.set_ref(copy)
assert s.set_holder(copy)
del copy, s
assert stats.alive() == 0
def test_shared_from_this_holder_ref():
s = m.SharedFromThisRef()
stats = ConstructorStats.get(m.B)
assert stats.alive() == 2
holder_ref = (
s.holder_ref
) # init_holder_helper(holder_ptr=true, owned=false, bad_wp=false)
assert stats.alive() == 2
assert s.set_ref(holder_ref)
assert s.set_holder(holder_ref)
del holder_ref, s
assert stats.alive() == 0
def test_shared_from_this_holder_copy():
s = m.SharedFromThisRef()
stats = ConstructorStats.get(m.B)
assert stats.alive() == 2
holder_copy = (
# RuntimeError: Invalid return_value_policy for shared_ptr.
s.holder_copy
) # init_holder_helper(holder_ptr=true, owned=true, bad_wp=false)
assert stats.alive() == 2
assert s.set_ref(holder_copy)
assert s.set_holder(holder_copy)
del holder_copy, s
assert stats.alive() == 0
def test_shared_from_this_virt():
pytest.skip("Pre-empting ASAN heap-use-after-free in next line.")
z = m.SharedFromThisVirt.get()
y = m.SharedFromThisVirt.get()
assert y is z