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, source_filename, db_file_path, logger):
filters = [CommentFilter("#"), SpacesFilter()]
standard_key_transformer = StandardKeyTransformer()
ip_key_transformer = lambda key, state: "%(category)s.ip" % {"category": state.state_id}
handlers = [
StandardConfigSectionHandler(),
SimpleKeyValueHandler.create_with_known_key_predicate(
"=", lambda key, state: state.state_id == "services", standard_key_transformer
),
TransformKeyValueHandler.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
),
]
init_state = OrderedDict()
encoding = "utf-8"
self._collect_task = FileSourceCollectTask(
self._source, filters, handlers, init_state, source_filename, encoding, db_file_path, logger
)
def __init__(self, key_value_delimiter, known_key_predicate, key_transformer, value):
self._key_value_handler = SimpleKeyValueHandler.create_with_known_key_predicate(key_value_delimiter, known_key_predicate, key_transformer)
self._single_key_handler = SingleKeyHandler.create_with_known_key_predicate(key_value_delimiter, known_key_predicate, key_transformer, value)
def __init__(self, methodName='runTest'):
super(TestSimpleKeyValueHandler, self).__init__(methodName)
key_transformer = lambda key, value, state: key
self._handler = SimpleKeyValueHandler.create_with_known_key_list('=', ['key13', 'key666'], key_transformer)