bt2: Add `_Clock*Const` classes and adapt tests

[babeltrace.git] / src / bindings / python / bt2 / bt2 / clock_snapshot.py

diff --git a/src/bindings/python/bt2/bt2/clock_snapshot.py b/src/bindings/python/bt2/bt2/clock_snapshot.py
index 89d7a11b5847413fac7209625a767b9cfa6dff5b..d2da481b9136f10abfd1d89e6779fbafa2b7a972 100644 (file)
--- a/src/bindings/python/bt2/bt2/clock_snapshot.py
+++ b/src/bindings/python/bt2/bt2/clock_snapshot.py
@@ -22,17 +22,17 @@
 
 from bt2 import native_bt, object, utils
 import numbers
-import bt2
+from bt2 import clock_class as bt2_clock_class
 import functools
 
 
 @functools.total_ordering
-class _ClockSnapshot(object._UniqueObject):
+class _ClockSnapshotConst(object._UniqueObject):
     @property
     def clock_class(self):
         cc_ptr = native_bt.clock_snapshot_borrow_clock_class_const(self._ptr)
         assert cc_ptr is not None
-        return bt2.clock_class._ClockClass._create_from_ptr_and_get_ref(cc_ptr)
+        return bt2_clock_class._ClockClassConst._create_from_ptr_and_get_ref(cc_ptr)
 
     @property
     def value(self):
@@ -40,11 +40,10 @@ class _ClockSnapshot(object._UniqueObject):
 
     @property
     def ns_from_origin(self):
-        ret, ns = native_bt.clock_snapshot_get_ns_from_origin(self._ptr)
-
-        if ret == native_bt.CLOCK_SNAPSHOT_STATUS_OVERFLOW:
-            raise OverflowError("cannot get clock snapshot's nanoseconds from origin")
-
+        status, ns = native_bt.clock_snapshot_get_ns_from_origin(self._ptr)
+        utils._handle_func_status(
+            status, "cannot get clock snapshot's nanoseconds from origin"
+        )
         return ns
 
     def __eq__(self, other):
@@ -62,7 +61,3 @@ class _ClockSnapshot(object._UniqueObject):
 
 class _UnknownClockSnapshot:
     pass
-
-
-class _InfiniteClockSnapshot:
-    pass