Add `babeltrace.` prefix to `trace-info` and `support-info` query objects
[babeltrace.git] / tests / data / cli / auto-source-discovery / bt_plugin_test.py
index 8d65b3d00882b1a044e155531cd60a3a796bf0c9..a4177240ed03b674fe32afd39dc893cd3b3a000a 100644 (file)
@@ -26,8 +26,8 @@ class TestSourceExt(Base, bt2._UserSourceComponent, message_iterator_class=TestI
         self._print_params(params)
 
     @staticmethod
-    def _query(query_exec, obj, params, log_level):
-        if obj == 'support-info':
+    def _user_query(query_exec, obj, params, log_level):
+        if obj == 'babeltrace.support-info':
             if params['type'] == 'file':
                 name = os.path.basename(str(params['input']))
 
@@ -49,7 +49,7 @@ class TestSourceExt(Base, bt2._UserSourceComponent, message_iterator_class=TestI
             else:
                 return 0
         else:
-            raise bt2.InvalidObject
+            raise bt2.UnknownObject
 
 
 @bt2.plugin_component_class
@@ -64,15 +64,15 @@ class TestSourceSomeDir(
         self._print_params(params)
 
     @staticmethod
-    def _query(query_exec, obj, params, log_level):
-        if obj == 'support-info':
+    def _user_query(query_exec, obj, params, log_level):
+        if obj == 'babeltrace.support-info':
             if params['type'] == 'directory':
                 name = os.path.basename(str(params['input']))
                 return 1 if name == 'some-dir' else 0
             else:
                 return 0
         else:
-            raise bt2.InvalidObject
+            raise bt2.UnknownObject
 
 
 @bt2.plugin_component_class
@@ -83,15 +83,15 @@ class TestSourceABCDE(Base, bt2._UserSourceComponent, message_iterator_class=Tes
         self._print_params(params)
 
     @staticmethod
-    def _query(query_exec, obj, params, log_level):
-        if obj == 'support-info':
+    def _user_query(query_exec, obj, params, log_level):
+        if obj == 'babeltrace.support-info':
             return (
                 1.0
                 if params['type'] == 'string' and params['input'] == 'ABCDE'
                 else 0.0
             )
         else:
-            raise bt2.InvalidObject
+            raise bt2.UnknownObject
 
 
 class TestSourceNoQuery(bt2._UserSourceComponent, message_iterator_class=TestIter):
This page took 0.025081 seconds and 4 git commands to generate.