def _test_one_comp_from_one_spec_one_comp_from_both_2(
self,
params_ab=None,
params_a=None,
obj_ab=None,
obj_a=None,
logging_level_ab=None,
logging_level_a=None,
):
specs = [
bt2.AutoSourceComponentSpec(
self._dir_ab,
params=params_ab,
obj=obj_ab,
logging_level=logging_level_ab,
),
bt2.AutoSourceComponentSpec(self._dir_a,
params=params_a,
obj=obj_a,
logging_level=logging_level_a),
]
it = bt2.TraceCollectionMessageIterator(specs)
msgs = [x for x in it if type(x) is bt2._StreamBeginningMessageConst]
self.assertEqual(len(msgs), 2)
return msgs
def test_obj_no_override_with_no_obj(self):
specs = [
bt2.AutoSourceComponentSpec(
self._dir_ab, params={'what': 'python-obj'}, obj='deore'
),
bt2.AutoSourceComponentSpec(self._dir_a, params={'what': 'python-obj'}),
]
it = bt2.TraceCollectionMessageIterator(specs)
msgs = [x for x in it if type(x) is bt2._StreamBeginningMessageConst]
self.assertEqual(len(msgs), 2)
self.assertEqual(msgs[0].stream.name, "TestSourceA: deore")
self.assertEqual(msgs[1].stream.name, "TestSourceB: deore")
def test_iter_auto_source_component_spec(self):
specs = [bt2.AutoSourceComponentSpec(_3EVENTS_INTERSECT_TRACE_PATH)]
msg_iter = bt2.TraceCollectionMessageIterator(specs)
msgs = list(msg_iter)
self.assertEqual(len(msgs), 28)
hist = _count_msgs_by_type(msgs)
self.assertEqual(hist[bt2._EventMessageConst], 8)
def test_grouping(self):
specs = [
bt2.AutoSourceComponentSpec('ABCDE'),
bt2.AutoSourceComponentSpec(_AUTO_SOURCE_DISCOVERY_GROUPING_PATH),
]
it = bt2.TraceCollectionMessageIterator(specs)
msgs = [x for x in it if type(x) is bt2._StreamBeginningMessageConst]
self.assertEqual(len(msgs), 8)
self.assertEqual(msgs[0].stream.name, 'TestSourceABCDE: ABCDE')
self.assertEqual(msgs[1].stream.name, 'TestSourceExt: aaa1, aaa2, aaa3')
self.assertEqual(msgs[2].stream.name, 'TestSourceExt: bbb1, bbb2')
self.assertEqual(msgs[3].stream.name, 'TestSourceExt: ccc1')
self.assertEqual(msgs[4].stream.name, 'TestSourceExt: ccc2')
self.assertEqual(msgs[5].stream.name, 'TestSourceExt: ccc3')
self.assertEqual(msgs[6].stream.name, 'TestSourceExt: ccc4')
self.assertEqual(msgs[7].stream.name, 'TestSourceSomeDir: some-dir')
def test_iter_mixed_inputs(self):
msg_iter = bt2.TraceCollectionMessageIterator([
_3EVENTS_INTERSECT_TRACE_PATH,
bt2.AutoSourceComponentSpec(_SEQUENCE_TRACE_PATH),
bt2.ComponentSpec.from_named_plugin_and_component_class(
'ctf', 'fs', _NOINTERSECT_TRACE_PATH),
])
msgs = list(msg_iter)
self.assertEqual(len(msgs), 76)
hist = _count_msgs_by_type(msgs)
self.assertEqual(hist[bt2._EventMessageConst], 24)
def _test_two_comps_from_one_spec(self, params, obj=None, logging_level=None):
specs = [
bt2.AutoSourceComponentSpec(
self._dir_ab, params=params, obj=obj, logging_level=logging_level
)
]
it = bt2.TraceCollectionMessageIterator(specs)
msgs = [x for x in it if type(x) is bt2._StreamBeginningMessageConst]
self.assertEqual(len(msgs), 2)
return msgs
