2021-06-20 01:13:27 +00:00
|
|
|
# -*- coding: utf-8 -*-
|
2021-06-21 23:16:37 +00:00
|
|
|
import pytest
|
2021-06-20 01:13:27 +00:00
|
|
|
|
|
|
|
import pybind11_tests.class_sh_trampoline_shared_from_this as m
|
|
|
|
|
2021-06-20 15:35:51 +00:00
|
|
|
|
|
|
|
class PySft(m.Sft):
|
2021-06-20 01:13:27 +00:00
|
|
|
pass
|
|
|
|
|
|
|
|
|
2021-06-23 14:00:44 +00:00
|
|
|
def test_release_and_shared_from_this():
|
2021-06-20 15:35:51 +00:00
|
|
|
obj = PySft("PySft")
|
|
|
|
assert obj.history == "PySft"
|
2021-06-23 14:00:44 +00:00
|
|
|
assert m.use_count(obj) == 1
|
2021-06-22 18:01:48 +00:00
|
|
|
assert m.pass_shared_ptr(obj) == 2
|
2021-06-20 15:35:51 +00:00
|
|
|
assert obj.history == "PySft_PassSharedPtr"
|
2021-06-23 14:00:44 +00:00
|
|
|
assert m.use_count(obj) == 1
|
2021-06-22 18:01:48 +00:00
|
|
|
assert m.pass_shared_ptr(obj) == 2
|
2021-06-20 15:35:51 +00:00
|
|
|
assert obj.history == "PySft_PassSharedPtr_PassSharedPtr"
|
2021-06-23 14:00:44 +00:00
|
|
|
assert m.use_count(obj) == 1
|
2021-06-22 18:01:48 +00:00
|
|
|
|
2021-06-23 14:00:44 +00:00
|
|
|
|
|
|
|
def test_release_and_shared_from_this_leak():
|
2021-06-22 18:01:48 +00:00
|
|
|
obj = PySft("")
|
|
|
|
while True:
|
|
|
|
m.pass_shared_ptr(obj)
|
|
|
|
assert obj.history == ""
|
2021-06-23 14:00:44 +00:00
|
|
|
assert m.use_count(obj) == 1
|
2021-06-22 18:01:48 +00:00
|
|
|
break # Comment out for manual leak checking (use `top` command).
|
2021-06-20 15:35:51 +00:00
|
|
|
|
|
|
|
|
2021-06-23 14:00:44 +00:00
|
|
|
def test_release_and_stash():
|
2021-06-20 15:35:51 +00:00
|
|
|
obj = PySft("PySft")
|
|
|
|
stash1 = m.SftSharedPtrStash(1)
|
|
|
|
stash1.Add(obj)
|
2021-06-23 14:00:44 +00:00
|
|
|
exp_hist = "PySft_Stash1Add"
|
|
|
|
assert obj.history == exp_hist
|
|
|
|
assert m.use_count(obj) == 2
|
|
|
|
assert stash1.history(0) == exp_hist
|
|
|
|
assert stash1.use_count(0) == 1
|
2021-06-22 18:01:48 +00:00
|
|
|
assert m.pass_shared_ptr(obj) == 3
|
2021-06-23 14:00:44 +00:00
|
|
|
exp_hist += "_PassSharedPtr"
|
|
|
|
assert obj.history == exp_hist
|
|
|
|
assert m.use_count(obj) == 2
|
|
|
|
assert stash1.history(0) == exp_hist
|
2021-06-22 01:09:14 +00:00
|
|
|
assert stash1.use_count(0) == 1
|
2021-06-20 15:35:51 +00:00
|
|
|
stash2 = m.SftSharedPtrStash(2)
|
|
|
|
stash2.Add(obj)
|
2021-06-23 14:00:44 +00:00
|
|
|
exp_hist += "_Stash2Add"
|
|
|
|
assert obj.history == exp_hist
|
|
|
|
assert m.use_count(obj) == 3
|
|
|
|
assert stash2.history(0) == exp_hist
|
2021-06-22 01:09:14 +00:00
|
|
|
assert stash2.use_count(0) == 2
|
2021-06-20 15:35:51 +00:00
|
|
|
stash2.Add(obj)
|
2021-06-23 14:00:44 +00:00
|
|
|
exp_hist += "_Stash2Add"
|
|
|
|
assert obj.history == exp_hist
|
|
|
|
assert m.use_count(obj) == 4
|
|
|
|
assert stash1.history(0) == exp_hist
|
2021-06-22 01:09:14 +00:00
|
|
|
assert stash1.use_count(0) == 3
|
2021-06-23 14:00:44 +00:00
|
|
|
assert stash2.history(0) == exp_hist
|
2021-06-22 01:09:14 +00:00
|
|
|
assert stash2.use_count(0) == 3
|
2021-06-23 14:00:44 +00:00
|
|
|
assert stash2.history(1) == exp_hist
|
2021-06-22 01:09:14 +00:00
|
|
|
assert stash2.use_count(1) == 3
|
2021-06-20 15:35:51 +00:00
|
|
|
del obj
|
2021-06-23 14:00:44 +00:00
|
|
|
assert stash2.history(0) == exp_hist
|
2021-06-22 01:09:14 +00:00
|
|
|
assert stash2.use_count(0) == 3
|
2021-06-23 14:00:44 +00:00
|
|
|
assert stash2.history(1) == exp_hist
|
2021-06-22 01:09:14 +00:00
|
|
|
assert stash2.use_count(1) == 3
|
2021-06-23 14:00:44 +00:00
|
|
|
stash2.Clear()
|
|
|
|
assert stash1.history(0) == exp_hist
|
2021-06-22 18:01:48 +00:00
|
|
|
assert stash1.use_count(0) == 1
|
2021-06-20 15:35:51 +00:00
|
|
|
|
|
|
|
|
2021-06-23 14:00:44 +00:00
|
|
|
def test_release_and_stash_leak():
|
2021-06-22 18:01:48 +00:00
|
|
|
obj = PySft("")
|
|
|
|
while True:
|
|
|
|
stash1 = m.SftSharedPtrStash(1)
|
|
|
|
stash1.Add(obj)
|
|
|
|
assert obj.history == ""
|
2021-06-23 14:00:44 +00:00
|
|
|
assert m.use_count(obj) == 2
|
2021-06-22 18:01:48 +00:00
|
|
|
assert stash1.use_count(0) == 1
|
|
|
|
stash1.Add(obj)
|
|
|
|
assert obj.history == ""
|
2021-06-23 14:00:44 +00:00
|
|
|
assert m.use_count(obj) == 3
|
2021-06-22 18:01:48 +00:00
|
|
|
assert stash1.use_count(0) == 2
|
|
|
|
assert stash1.use_count(1) == 2
|
|
|
|
break # Comment out for manual leak checking (use `top` command).
|
|
|
|
|
|
|
|
|
2021-06-23 14:00:44 +00:00
|
|
|
def test_release_and_stash_via_shared_from_this():
|
2021-06-20 15:35:51 +00:00
|
|
|
obj = PySft("PySft")
|
|
|
|
stash1 = m.SftSharedPtrStash(1)
|
2021-06-23 14:00:44 +00:00
|
|
|
with pytest.raises(RuntimeError) as exc_info:
|
|
|
|
stash1.AddSharedFromThis(obj)
|
|
|
|
assert str(exc_info.value) == "bad_weak_ptr"
|
|
|
|
stash1.Add(obj)
|
|
|
|
assert obj.history == "PySft_Stash1Add"
|
|
|
|
assert stash1.use_count(0) == 1
|
2021-06-20 15:35:51 +00:00
|
|
|
stash1.AddSharedFromThis(obj)
|
2021-06-23 14:00:44 +00:00
|
|
|
assert obj.history == "PySft_Stash1Add_Stash1AddSharedFromThis"
|
2021-06-20 15:35:51 +00:00
|
|
|
assert stash1.use_count(0) == 2
|
2021-06-23 14:00:44 +00:00
|
|
|
assert stash1.use_count(1) == 2
|
2021-06-22 18:01:48 +00:00
|
|
|
|
|
|
|
|
2021-06-23 14:00:44 +00:00
|
|
|
def test_release_and_stash_via_shared_from_this_leak():
|
2021-06-22 18:01:48 +00:00
|
|
|
obj = PySft("")
|
|
|
|
while True:
|
|
|
|
stash1 = m.SftSharedPtrStash(1)
|
2021-06-23 14:00:44 +00:00
|
|
|
with pytest.raises(RuntimeError) as exc_info:
|
|
|
|
stash1.AddSharedFromThis(obj)
|
|
|
|
assert str(exc_info.value) == "bad_weak_ptr"
|
2021-06-22 18:01:48 +00:00
|
|
|
stash1.Add(obj)
|
|
|
|
assert obj.history == ""
|
|
|
|
assert stash1.use_count(0) == 1
|
|
|
|
stash1.AddSharedFromThis(obj)
|
|
|
|
assert obj.history == ""
|
|
|
|
assert stash1.use_count(0) == 2
|
|
|
|
assert stash1.use_count(1) == 2
|
|
|
|
break # Comment out for manual leak checking (use `top` command).
|
2021-06-21 23:16:37 +00:00
|
|
|
|
|
|
|
|
|
|
|
def test_pass_released_shared_ptr_as_unique_ptr():
|
|
|
|
obj = PySft("PySft")
|
|
|
|
stash1 = m.SftSharedPtrStash(1)
|
|
|
|
stash1.Add(obj) # Releases shared_ptr to C++.
|
|
|
|
with pytest.raises(ValueError) as exc_info:
|
|
|
|
m.pass_unique_ptr(obj)
|
|
|
|
assert str(exc_info.value) == (
|
|
|
|
"Python instance is currently owned by a std::shared_ptr."
|
|
|
|
)
|
2021-06-23 23:35:27 +00:00
|
|
|
|
|
|
|
|
|
|
|
def test_pure_cpp_sft_raw_ptr():
|
|
|
|
obj = m.make_pure_cpp_sft_raw_ptr("PureCppSft")
|
|
|
|
with pytest.raises(RuntimeError) as exc_info:
|
|
|
|
assert m.pass_shared_ptr(obj) == 3 # TODO: FIX.
|
|
|
|
assert str(exc_info.value) == "bad_weak_ptr"
|
|
|
|
obj = m.make_pure_cpp_sft_raw_ptr("PureCppSft")
|
|
|
|
stash1 = m.SftSharedPtrStash(1)
|
|
|
|
with pytest.raises(RuntimeError) as exc_info:
|
|
|
|
stash1.AddSharedFromThis(obj) # TODO: FIX.
|
|
|
|
assert str(exc_info.value) == "bad_weak_ptr"
|
|
|
|
stash1.Add(obj)
|
|
|
|
with pytest.raises(RuntimeError) as exc_info:
|
|
|
|
stash1.AddSharedFromThis(obj) # TODO: FIX.
|
|
|
|
assert str(exc_info.value) == "bad_weak_ptr"
|
|
|
|
|
|
|
|
|
|
|
|
def test_pure_cpp_sft_shd_ptr():
|
|
|
|
obj = m.make_pure_cpp_sft_shd_ptr("PureCppSft")
|
|
|
|
assert m.pass_shared_ptr(obj) == 3
|
|
|
|
assert obj.history == "PureCppSft_PassSharedPtr"
|
|
|
|
obj = m.make_pure_cpp_sft_shd_ptr("PureCppSft")
|
|
|
|
stash1 = m.SftSharedPtrStash(1)
|
|
|
|
stash1.AddSharedFromThis(obj)
|
|
|
|
assert obj.history == "PureCppSft_Stash1AddSharedFromThis"
|