bt2: Adapt test_event.py and make it pass

[babeltrace.git] / bindings / python / bt2 / bt2 / object.py

diff --git a/bindings/python/bt2/bt2/object.py b/bindings/python/bt2/bt2/object.py
index ac5107d1eead943f44e8c42a41ce45d4dd29eb85..218965de7799762839cae6120696a55ff21023b7 100644 (file)
--- a/bindings/python/bt2/bt2/object.py
+++ b/bindings/python/bt2/bt2/object.py
@@ -72,10 +72,14 @@ class _UniqueObject(_BaseObject):
     def _create_from_ptr_and_get_ref(cls, ptr, owner_ptr,
                                      owner_get_ref, owner_put_ref):
         obj = cls.__new__(cls)
+
         obj._ptr = ptr
         obj._owner_ptr = owner_ptr
+        obj._owner_get_ref = owner_get_ref
         obj._owner_put_ref = owner_put_ref
-        owner_get_ref(obj._owner_ptr)
+
+        obj._owner_get_ref(obj._owner_ptr)
+
         return obj
 
     def __del__(self):