1 files changed, 63 insertions, 0 deletions
diff --git a/sources/shiboken6/tests/samplebinding/keep_reference_test.py b/sources/shiboken6/tests/samplebinding/keep_reference_test.py
new file mode 100644
index 000000000..10591fec6
--- /dev/null
+++ b/sources/shiboken6/tests/samplebinding/keep_reference_test.py
@@ -0,0 +1,63 @@
+#!/usr/bin/env python
+# Copyright (C) 2022 The Qt Company Ltd.
+# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0
+
+import os
+import sys
+import unittest
+
+from pathlib import Path
+sys.path.append(os.fspath(Path(__file__).resolve().parents[1]))
+from shiboken_paths import init_paths
+init_paths()
+
+from sample import ObjectModel, ObjectView
+
+
+class TestKeepReference(unittest.TestCase):
+    '''Test case for objects that keep references to other object without
+       owning them (e.g. model/view relationships).'''
+
+    @unittest.skipUnless(hasattr(sys, "getrefcount"), f"{sys.implementation.name} has no refcount")
+    def testReferenceCounting(self):
+        '''Tests reference count of model-like object referred by view-like objects.'''
+        model1 = ObjectModel()
+        refcount1 = sys.getrefcount(model1)
+        view1 = ObjectView()
+        view1.setModel(model1)
+        self.assertEqual(sys.getrefcount(view1.model()), refcount1 + 1)
+
+        view2 = ObjectView()
+        view2.setModel(model1)
+        self.assertEqual(sys.getrefcount(view2.model()), refcount1 + 2)
+
+        model2 = ObjectModel()
+        view2.setModel(model2)
+        self.assertEqual(sys.getrefcount(view1.model()), refcount1 + 1)
+
+    @unittest.skipUnless(hasattr(sys, "getrefcount"), f"{sys.implementation.name} has no refcount")
+    def testReferenceCountingWhenDeletingReferrer(self):
+        '''Tests reference count of model-like object referred by deceased view-like object.'''
+        model = ObjectModel()
+        refcount1 = sys.getrefcount(model)
+        view = ObjectView()
+        view.setModel(model)
+        self.assertEqual(sys.getrefcount(view.model()), refcount1 + 1)
+
+        del view
+        self.assertEqual(sys.getrefcount(model), refcount1)
+
+    def testReferreedObjectSurvivalAfterContextEnd(self):
+        '''Model-like object assigned to a view-like object must survive
+           after get out of context.'''
+        def createModelAndSetToView(view):
+            model = ObjectModel()
+            model.setObjectName('created model')
+            view.setModel(model)
+        view = ObjectView()
+        createModelAndSetToView(view)
+        model = view.model()    # noqa: F841
+
+
+if __name__ == '__main__':
+    unittest.main()