def test_remove_custom_modules(monkeypatch):
mock_modules = {'a': 99, 'b': 300, 'c': 400, 'abrt': 190}
mock_templates = {'base_container': 400, 'net_container': 300}
def consume_SELinuxModules_mocked(*models):
semodule_list = [
SELinuxModule(name=name, priority=priority, content='', removed=[])
for name, priority in mock_modules.items()
]
template_list = [
SELinuxModule(name=name, priority=priority, content='', removed=[])
for name, priority in mock_templates.items()
]
yield SELinuxModules(modules=semodule_list, templates=template_list)
monkeypatch.setattr(api, 'consume', consume_SELinuxModules_mocked)
monkeypatch.setattr(selinuxprepare, 'run', run_mocked())
selinuxprepare.remove_custom_modules()
# 1 call for udica templates and 1 for other custom modules
assert selinuxprepare.run.called == 2
assert selinuxprepare.run.non_semodule_calls == 0
# verify that remove_custom_modules tried to remove all given modules
assert (set(mock_modules).union(set(mock_templates)) -
selinuxprepare.run.removed_modules) == set()
def test_remove_custom_modules(monkeypatch):
mock_modules = {'a': 99, 'b': 300, 'c': 400, 'abrt': 190}
def consume_SELinuxModules_mocked(*models):
semodule_list = [SELinuxModule(name=k, priority=mock_modules[k], content='', removed=[])
for k in mock_modules]
yield SELinuxModules(modules=semodule_list)
monkeypatch.setattr(api, 'consume', consume_SELinuxModules_mocked)
monkeypatch.setattr(selinuxprepare, 'run', run_mocked())
selinuxprepare.remove_custom_modules()
assert selinuxprepare.run.called == len(mock_modules)
assert selinuxprepare.run.non_semodule_calls == 0
# verify that remove_custom_modules tried to remove all given modules
assert (set(mock_modules) - selinuxprepare.run.removed_modules) == set()
def process(self):
selinuxprepare.remove_semanage_customizations()
selinuxprepare.remove_custom_modules()