tests/plugins/src.ctf.fs/query/test_query_support_info.py

   1 # SPDX-License-Identifier: GPL-2.0-only
   2 #
   3 # Copyright (C) 2019 Simon Marchi <simon.marchi@efficios.com>
   4 #
   5
   6 import unittest
   7 import bt2
   8 import os
   9
  10
  11 session_rotation_trace_path = os.path.join(
  12     os.environ["BT_CTF_TRACES_PATH"], "succeed", "session-rotation"
  13 )
  14
  15
  16 trace_10352_1 = os.path.join(
  17     session_rotation_trace_path,
  18     "a",
  19     "1",
  20     "ust",
  21     "pid",
  22     "10352",
  23 )
  24 trace_10353_1 = os.path.join(
  25     session_rotation_trace_path,
  26     "a",
  27     "1",
  28     "ust",
  29     "pid",
  30     "10353",
  31 )
  32 trace_10352_2 = os.path.join(
  33     session_rotation_trace_path,
  34     "a",
  35     "2",
  36     "ust",
  37     "pid",
  38     "10352",
  39 )
  40 trace_10353_2 = os.path.join(
  41     session_rotation_trace_path,
  42     "a",
  43     "2",
  44     "ust",
  45     "pid",
  46     "10353",
  47 )
  48 trace_10352_3 = os.path.join(
  49     session_rotation_trace_path,
  50     "3",
  51     "ust",
  52     "pid",
  53     "10352",
  54 )
  55 trace_10353_3 = os.path.join(
  56     session_rotation_trace_path,
  57     "3",
  58     "ust",
  59     "pid",
  60     "10353",
  61 )
  62
  63
  64 class QuerySupportInfoTestCase(unittest.TestCase):
  65     def test_support_info_with_uuid(self):
  66         # Test that the right group is reported for each trace.
  67
  68         def do_one_query(input, expected_group):
  69             qe = bt2.QueryExecutor(
  70                 fs, "babeltrace.support-info", {"input": input, "type": "directory"}
  71             )
  72
  73             result = qe.query()
  74             self.assertEqual(result["group"], expected_group)
  75
  76         ctf = bt2.find_plugin("ctf")
  77         fs = ctf.source_component_classes["fs"]
  78
  79         do_one_query(trace_10352_1, "21cdfa5e-9a64-490a-832c-53aca6c101ba")
  80         do_one_query(trace_10352_2, "21cdfa5e-9a64-490a-832c-53aca6c101ba")
  81         do_one_query(trace_10352_3, "21cdfa5e-9a64-490a-832c-53aca6c101ba")
  82         do_one_query(trace_10353_1, "83656eb1-b131-40e7-9666-c04ae279b58c")
  83         do_one_query(trace_10353_2, "83656eb1-b131-40e7-9666-c04ae279b58c")
  84         do_one_query(trace_10353_3, "83656eb1-b131-40e7-9666-c04ae279b58c")
  85
  86
  87 if __name__ == "__main__":
  88     unittest.main()