def test__get_service_filter_func_same_lists(monkeypatch, whitelist, result):
monkeypatch.setattr(config, "service_description",
lambda h, c, i: "Test Description")
params = {"inventory_rediscovery": {"service_whitelist": whitelist}}
service_filters = discovery.get_service_filter_funcs(params)
service = discovery.Service(CheckPluginName("check_plugin_name"), "item",
"Test Description", None)
assert service_filters.new is not None
assert service_filters.new("hostname", service) is result
params = {"inventory_rediscovery": {"service_blacklist": whitelist}}
service_filters_inverse = discovery.get_service_filter_funcs(params)
assert service_filters_inverse.new is not None
assert service_filters_inverse.new("hostname", service) is not result
params = {
"inventory_rediscovery": {
"service_whitelist": whitelist,
"service_blacklist": whitelist,
}
}
service_filters_both = discovery.get_service_filter_funcs(params)
assert service_filters_both.new is not None
assert service_filters_both.new("hostname", service) is False
def test__get_post_discovery_services(monkeypatch, grouped_services, mode, parameters_rediscovery,
result_new_item_names, result_counts):
def _get_service_description(_hostname, _check_plugin_name, item):
return "Test Description %s" % item
monkeypatch.setattr(config, "service_description", _get_service_description)
counts = discovery._empty_counts()
params = {"inventory_rediscovery": parameters_rediscovery}
service_filters = discovery.get_service_filter_funcs(params)
new_item_names = [
entry.item for entry in discovery._get_post_discovery_services(
"hostname",
grouped_services,
service_filters,
counts,
mode,
)
]
count_new, count_kept, count_removed = result_counts
assert sorted(new_item_names) == sorted(result_new_item_names)
assert counts["self_new"] == count_new
assert counts["self_kept"] == count_kept
assert counts["self_removed"] == count_removed
def test__get_service_filter_func(monkeypatch, parameters_rediscovery, result):
monkeypatch.setattr(config, "service_description", lambda h, c, i: "Test Description")
params = {"inventory_rediscovery": parameters_rediscovery}
service_filters = discovery.get_service_filter_funcs(params)
service = discovery.Service(CheckPluginName("check_plugin_name"), "item", "Test Description",
None)
assert service_filters.new is not None
assert service_filters.new("hostname", service) is result
def test__get_service_filters_lists(parameters, new_whitelist, new_blacklist, vanished_whitelist,
vanished_blacklist):
service_filter_lists = discovery._get_service_filter_lists(parameters)
assert service_filter_lists.new_whitelist == new_whitelist
assert service_filter_lists.new_blacklist == new_blacklist
assert service_filter_lists.vanished_whitelist == vanished_whitelist
assert service_filter_lists.vanished_blacklist == vanished_blacklist
service_filters = discovery.get_service_filter_funcs(parameters)
assert service_filters.new is not None
assert service_filters.vanished is not None
def test__get_service_filter_func_no_lists(parameters_rediscovery):
params = {"inventory_rediscovery": parameters_rediscovery}
service_filters = discovery.get_service_filter_funcs(params)
assert service_filters.new is discovery._accept_all_services
assert service_filters.vanished is discovery._accept_all_services
