tap-driver.sh: flush stdout after each test result

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

diff --git a/bindings/python/bt2/bt2/query_executor.py b/bindings/python/bt2/bt2/query_executor.py
index f90bb902651cf5fb98c2b43bcab052c35068f250..a87713ff1c8a4f2c540fb35fabd81592cd910692 100644 (file)
--- a/bindings/python/bt2/bt2/query_executor.py
+++ b/bindings/python/bt2/bt2/query_executor.py
@@ -26,8 +26,8 @@ import bt2
 
 
 class QueryExecutor(object._SharedObject):
-    _get_ref = native_bt.query_executor_get_ref
-    _put_ref = native_bt.query_executor_put_ref
+    _get_ref = staticmethod(native_bt.query_executor_get_ref)
+    _put_ref = staticmethod(native_bt.query_executor_put_ref)
 
     def _handle_status(self, status, gen_error_msg):
         if status == native_bt.QUERY_EXECUTOR_STATUS_AGAIN:
@@ -60,6 +60,9 @@ class QueryExecutor(object._SharedObject):
         return is_canceled > 0
 
     def query(self, component_class, object, params=None):
+        if self.is_canceled:
+            raise bt2.QueryExecutorCanceled
+
         if not isinstance(component_class, bt2.component._GenericComponentClass):
             err = False
 
@@ -88,9 +91,3 @@ class QueryExecutor(object._SharedObject):
         self._handle_status(status, 'cannot query component class')
         assert(result_ptr)
         return bt2.value._create_from_ptr(result_ptr)
-
-    def __eq__(self, other):
-        if type(other) is not type(self):
-            return False
-
-        return self.addr == other.addr