def __init__(self, methodName='runTest'): super(TestFileSourceCollector, self).__init__(methodName) key_transformer = StandardKeyTransformer() ip_key_transformer = lambda key, value, state: state.state_id + '_ip' aggregate_fun = lambda old_value, item: old_value + 1 filters = [CommentFilter('#'), SpacesFilter()] handlers = [StandardConfigSectionHandler(), SimpleKeyListHandler.create_with_known_key_list('=', ['key13', 'key666'], key_transformer), AggregateKeyValueHandler.create_with_known_key_list('=', ['key555', 'key999'], key_transformer, aggregate_fun, 0), AggregateKeyValueHandler.create_with_known_key_list('=', ['ip0', 'ip1', 'ip2', 'ip3', 'ip4'], ip_key_transformer, aggregate_fun, 0)] self._collector = FileSourceCollector(filters, handlers, OrderedDict())
def __init__(self, methodName='runTest'): super(TestSimpleKeyListHandler, self).__init__(methodName) key_transformer = lambda key, value, state: key self._handler = SimpleKeyListHandler.create_with_known_key_list('=', ['key13', 'key666'], key_transformer)