def setUp(self):
self._mox = Mox()
self._main_logger = self._mox.CreateMock(Logger)
self._storage_logger = self._mox.CreateMock(Logger)
source_filename = os.path.abspath("tests/functional_tests/test.conf")
self._db_manager.__enter__()
filters = [CommentFilter("#"), SpacesFilter()]
standard_key_transformer = StandardKeyTransformer()
ip_key_transformer = lambda key, value, state: "{category:s}.ip".format(category=state.state_id)
handlers = [
StandardConfigSectionHandler(),
SimpleKeyValueHandler.create_with_known_key_predicate(
"=", lambda key, state: state.state_id == "services", standard_key_transformer
),
TransformKeyListHandler.create_with_known_key_predicate(
"=", lambda key, state: state.state_id == "users", standard_key_transformer, transform_user_fun
),
AggregateKeyValueHandler.create_with_known_key_list(
"=", ["ip0", "ip1", "ip2", "ip3", "ip4"], ip_key_transformer, lambda old_value, item: old_value + 1, 0
),
SimpleKeyValueHandler.create_with_known_key_list(
"=", ["use_local_mail", "use_remote_mail", "use_jabber"], standard_key_transformer
),
SingleKeyHandler.create_with_known_key_list(
"=", ["use_local_mail", "use_remote_mail", "use_jabber"], standard_key_transformer, "0"
),
]
self._collect_task = FileSourceCollectTask(
"some_source",
filters,
handlers,
OrderedDict(),
source_filename,
"utf8",
self._db_manager.db_filename,
self._main_logger,
)
def __init__(self, methodName='runTest'):
super(TestTransformKeyListHandler, self).__init__(methodName)
key_transformer = lambda key, value, state: key
value_transformer = lambda value: '__' + str(value) + '__'
self._handler = TransformKeyListHandler.create_with_known_key_list('=', ['key13', 'key666'], key_transformer, value_transformer)
