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