Blame tests/test_rule_handling.py

Packit 792a06
import pytest
Packit 792a06
import mock
Packit 792a06
from collections import defaultdict
Packit 792a06
Packit 792a06
from pyanaconda.modules.common.constants.objects import FIREWALL, DEVICE_TREE, BOOTLOADER
Packit 792a06
from pyanaconda.modules.common.constants.services import NETWORK, STORAGE, USERS
Packit 792a06
Packit 792a06
try:
Packit 792a06
    from org_fedora_oscap import rule_handling, common
Packit 792a06
except ImportError as exc:
Packit 792a06
    pytestmark = pytest.mark.skip(
Packit 792a06
        "Unable to import modules, possibly due to bad version of Anaconda: {error}"
Packit 792a06
        .format(error=str(exc)))
Packit 792a06
Packit 792a06
Packit 792a06
@pytest.fixture()
Packit 792a06
def part_rules():
Packit 792a06
    rules = rule_handling.PartRules()
Packit 792a06
    rules.ensure_mount_point("/tmp")
Packit 792a06
    return rules
Packit 792a06
Packit 792a06
Packit 792a06
# simple tests, shouldn't raise exceptions
Packit 792a06
def test_part_rules_getitem(part_rules):
Packit 792a06
    part_rules["/tmp"]
Packit 792a06
Packit 792a06
Packit 792a06
def test_part_rules_setitem(part_rules):
Packit 792a06
    rule = rule_handling.PartRule("/var/log")
Packit 792a06
    part_rules["/var/log"] = rule
Packit 792a06
Packit 792a06
Packit 792a06
def test_part_rules_len(part_rules):
Packit 792a06
    assert len(part_rules) == 1
Packit 792a06
Packit 792a06
Packit 792a06
def test_part_rules_contains(part_rules):
Packit 792a06
    assert "/tmp" in part_rules
Packit 792a06
Packit 792a06
Packit 792a06
def test_part_rules_delitem(part_rules):
Packit 792a06
    del(part_rules["/tmp"])
Packit 792a06
    assert "/tmp" not in part_rules
Packit 792a06
Packit 792a06
Packit 792a06
@pytest.fixture()
Packit 792a06
def rule_data():
Packit 792a06
    return rule_handling.RuleData()
Packit 792a06
Packit 792a06
Packit 792a06
def test_rule_data_artificial(rule_data):
Packit 792a06
    rule_data.new_rule("  part /tmp --mountoptions=nodev,noauto")
Packit 792a06
    rule_data.new_rule("part /var/log  ")
Packit 792a06
    rule_data.new_rule(" passwd   --minlen=14 ")
Packit 792a06
    rule_data.new_rule("package --add=iptables")
Packit 792a06
    rule_data.new_rule(" package --add=firewalld --remove=telnet")
Packit 792a06
    rule_data.new_rule("package --remove=rlogin --remove=sshd")
Packit 792a06
    rule_data.new_rule("bootloader --passwd")
Packit 792a06
Packit 792a06
    # both partitions should appear in rule_data._part_rules
Packit 792a06
    assert "/tmp" in rule_data._part_rules
Packit 792a06
    assert "/var/log" in rule_data._part_rules
Packit 792a06
Packit 792a06
    # mount options should be parsed
Packit 792a06
    assert "nodev" in rule_data._part_rules["/tmp"]._mount_options
Packit 792a06
    assert "noauto" in rule_data._part_rules["/tmp"]._mount_options
Packit 792a06
Packit 792a06
    # no mount options for /var/log
Packit 792a06
    assert not rule_data._part_rules["/var/log"]._mount_options
Packit 792a06
Packit 792a06
    # minimal password length should be parsed and stored correctly
Packit 792a06
    assert rule_data._passwd_rules._minlen == 14
Packit 792a06
Packit 792a06
    # packages should be parsed correctly
Packit 792a06
    assert "iptables" in rule_data._package_rules._add_pkgs
Packit 792a06
    assert "firewalld" in rule_data._package_rules._add_pkgs
Packit 792a06
    assert "telnet" in rule_data._package_rules._remove_pkgs
Packit 792a06
    assert "rlogin" in rule_data._package_rules._remove_pkgs
Packit 792a06
    assert "sshd" in rule_data._package_rules._remove_pkgs
Packit 792a06
Packit 792a06
    # bootloader should require password
Packit 792a06
    assert rule_data._bootloader_rules._require_password
Packit 792a06
Packit 792a06
Packit 792a06
def test_rule_data_quoted_opt_values(rule_data):
Packit 792a06
    rule_data.new_rule('part /tmp --mountoptions="nodev,noauto"')
Packit 792a06
Packit 792a06
    assert "nodev" in rule_data._part_rules["/tmp"]._mount_options
Packit 792a06
    assert "noauto" in rule_data._part_rules["/tmp"]._mount_options
Packit 792a06
    assert '"' not in rule_data._part_rules["/tmp"]._mount_options
Packit 792a06
Packit 792a06
Packit 792a06
def test_rule_data_real_output(rule_data):
Packit 792a06
    output = """
Packit 792a06
    part /tmp
Packit 792a06
Packit 792a06
    part /tmp --mountoptions=nodev
Packit 792a06
    """
Packit 792a06
    for line in output.splitlines():
Packit 792a06
        rule_data.new_rule(line)
Packit 792a06
Packit 792a06
    assert "/tmp" in rule_data._part_rules
Packit 792a06
    assert "nodev" in rule_data._part_rules["/tmp"]._mount_options
Packit 792a06
Packit 792a06
    # should be stripped and merged
Packit 792a06
    assert str(rule_data._part_rules) == "part /tmp --mountoptions=nodev"
Packit 792a06
Packit 792a06
Packit 792a06
@pytest.fixture()
Packit 792a06
def ksdata_mock():
Packit 792a06
    return mock.Mock()
Packit 792a06
Packit 792a06
Packit 792a06
@pytest.fixture()
Packit 792a06
def storage_mock():
Packit 792a06
    return mock.Mock()
Packit 792a06
Packit 792a06
Packit 792a06
# monkeypatch is a predefined fixture that is used to perform per-test case changes in the test env.
Packit 792a06
# Here, it mocks the Anaconda interface on the module level. For more info, see:
Packit 792a06
# https://docs.pytest.org/en/latest/monkeypatch.html#monkeypatching-mocking-modules-and-environments
Packit 792a06
@pytest.fixture()
Packit 792a06
def proxy_getter(monkeypatch):
Packit 792a06
    proxies = defaultdict(mock.Mock)
Packit 792a06
Packit 792a06
    def mock_get(service_name, object_path):
Packit 792a06
        initialize = not proxies
Packit 792a06
        proxy = proxies[(service_name, object_path)]
Packit 792a06
Packit 792a06
        if initialize:
Packit 792a06
            set_dbus_defaults()
Packit 792a06
Packit 792a06
        return proxy
Packit 792a06
Packit 792a06
    monkeypatch.setattr("pyanaconda.core.dbus.DBus.get_proxy", mock_get)
Packit 792a06
    return mock_get
Packit 792a06
Packit 792a06
Packit 792a06
def set_dbus_defaults():
Packit 792a06
    network = NETWORK.get_proxy()
Packit 792a06
    network.Connected.return_value = True
Packit 792a06
Packit 792a06
    firewall = NETWORK.get_proxy(FIREWALL)
Packit 792a06
    firewall.EnabledServices = []
Packit 792a06
    firewall.DisabledServices = []
Packit 792a06
    firewall.EnabledPorts = []
Packit 792a06
    firewall.Trusts = []
Packit 792a06
Packit 792a06
    device_tree = STORAGE.get_proxy(DEVICE_TREE)
Packit 792a06
    device_tree.GetDeviceMountOptions.return_value = "defaults"
Packit 792a06
    device_tree.GetMountPoints.return_value = {}
Packit 792a06
Packit 792a06
    bootloader = STORAGE.get_proxy(BOOTLOADER)
Packit 792a06
    bootloader.IsPasswordSet = False
Packit 792a06
Packit 792a06
    users = USERS.get_proxy()
Packit 792a06
    users.IsRootPasswordSet = True
Packit 792a06
    users.IsRootPasswordCrypted = False
Packit 792a06
    users.RootPassword = "anaconda"
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_existing_part_must_exist_rules(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    rules = [
Packit 792a06
        "part /tmp",
Packit 792a06
        "part /",
Packit 792a06
    ]
Packit 792a06
    for rule in rules:
Packit 792a06
        rule_data.new_rule(rule)
Packit 792a06
Packit 792a06
    device_tree_mock = STORAGE.get_proxy(DEVICE_TREE)
Packit 792a06
    device_tree_mock.GetDeviceMountOptions.return_value = "defaults"
Packit 792a06
    device_tree_mock.GetMountPoints.return_value = {
Packit 792a06
        "/tmp": "/dev/sda1",
Packit 792a06
        "/": "/dev/sda2",
Packit 792a06
    }
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # partitions exist --> no errors, warnings or additional info
Packit 792a06
    assert not messages
Packit 792a06
Packit 792a06
    # no additional mount options specified
Packit 792a06
    device_tree_mock.SetDeviceMountOptions.assert_has_calls([
Packit 792a06
        mock.call("/dev/sda1", "defaults"),
Packit 792a06
        mock.call("/dev/sda2", "defaults"),
Packit 792a06
    ])
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_nonexisting_part_must_exist(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    rules = [
Packit 792a06
        "part /tmp",
Packit 792a06
        "part /",
Packit 792a06
    ]
Packit 792a06
    for rule in rules:
Packit 792a06
        rule_data.new_rule(rule)
Packit 792a06
Packit 792a06
    device_tree_mock = STORAGE.get_proxy(DEVICE_TREE)
Packit 792a06
    device_tree_mock.GetDeviceMountOptions.return_value = "defaults"
Packit 792a06
    device_tree_mock.GetMountPoints.return_value = {
Packit 792a06
        "/tmp": "/dev/sda1",
Packit 792a06
    }
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # / mount point missing --> one error
Packit 792a06
    assert len(messages) == 1
Packit 792a06
    assert messages[0].type == common.MESSAGE_TYPE_FATAL
Packit 792a06
Packit 792a06
    # error has to mention the mount point
Packit 792a06
    assert "/" in messages[0].text
Packit 792a06
Packit 792a06
Packit 792a06
def get_messages_for_partition_rules(
Packit 792a06
        rule_data, ksdata_mock, storage_mock,
Packit 792a06
        rules,
Packit 792a06
        messages_evaluation_count=1,
Packit 792a06
        actual_mountpoints=("/tmp", "/"),
Packit 792a06
        mount_options=None,
Packit 792a06
        report_only=False,
Packit 792a06
        ):
Packit 792a06
Packit 792a06
    assert len(rules) == 2, \
Packit 792a06
        "We need rules for temp partition and root."
Packit 792a06
Packit 792a06
    if mount_options is None:
Packit 792a06
        mount_options = {
Packit 792a06
            "/": "defaults",
Packit 792a06
            "/tmp": "defaults",
Packit 792a06
        }
Packit 792a06
Packit 792a06
    for rule in rules:
Packit 792a06
        rule_data.new_rule(rule)
Packit 792a06
Packit 792a06
    # Map mount points to device names.
Packit 792a06
    mount_points = {}
Packit 792a06
Packit 792a06
    if "/tmp" in actual_mountpoints:
Packit 792a06
        mount_points["/tmp"] = "/dev/sda1"
Packit 792a06
Packit 792a06
    if "/" in actual_mountpoints:
Packit 792a06
        mount_points["/"] = "/dev/sda2"
Packit 792a06
Packit 792a06
    # Map device names to mount options.
Packit 792a06
    mount_options = {
Packit 792a06
        mount_points[mount_point]: mount_options[mount_point]
Packit 792a06
        for mount_point in actual_mountpoints
Packit 792a06
    }
Packit 792a06
Packit 792a06
    # Mock the device tree proxy.
Packit 792a06
    def get_device_mount_options(device_name):
Packit 792a06
        return mount_options[device_name]
Packit 792a06
Packit 792a06
    def set_device_mount_options(device_name, options):
Packit 792a06
        mount_options[device_name] = options
Packit 792a06
Packit 792a06
    device_tree_mock = STORAGE.get_proxy(DEVICE_TREE)
Packit 792a06
    device_tree_mock.GetMountPoints.return_value = mount_points
Packit 792a06
    device_tree_mock.GetDeviceMountOptions.side_effect = get_device_mount_options
Packit 792a06
    device_tree_mock.SetDeviceMountOptions.side_effect = set_device_mount_options
Packit 792a06
Packit 792a06
    messages = []
Packit 792a06
    for _ in range(messages_evaluation_count):
Packit 792a06
        messages = rule_data.eval_rules(ksdata_mock, storage_mock, report_only)
Packit 792a06
Packit 792a06
    return messages
Packit 792a06
Packit 792a06
Packit 792a06
def evaluation_add_mount_options(
Packit 792a06
        rule_data, ksdata_mock, storage_mock,
Packit 792a06
        messages_evaluation_count):
Packit 792a06
    rules = [
Packit 792a06
        "part /tmp --mountoptions=defaults,nodev",
Packit 792a06
        "part / --mountoptions=noauto",
Packit 792a06
    ]
Packit 792a06
Packit 792a06
    messages = get_messages_for_partition_rules(
Packit 792a06
        rule_data, ksdata_mock, storage_mock,
Packit 792a06
        rules, messages_evaluation_count)
Packit 792a06
Packit 792a06
    # two mount options added --> two info messages
Packit 792a06
    assert len(messages) == 2
Packit 792a06
    assert all(message.type == common.MESSAGE_TYPE_INFO for message in messages)
Packit 792a06
Packit 792a06
    # newly added mount options should be mentioned in the messages
Packit 792a06
    # together with their mount points
Packit 792a06
    nodev_found = False
Packit 792a06
    noauto_found = False
Packit 792a06
Packit 792a06
    for message in messages:
Packit 792a06
        if "'nodev'" in message.text:
Packit 792a06
            assert "/tmp" in message.text
Packit 792a06
            nodev_found = True
Packit 792a06
        elif "'noauto'" in message.text:
Packit 792a06
            assert "/" in message.text
Packit 792a06
            noauto_found = True
Packit 792a06
Packit 792a06
    assert all([nodev_found, noauto_found])
Packit 792a06
Packit 792a06
    device_tree_mock = STORAGE.get_proxy(DEVICE_TREE)
Packit 792a06
    device_tree_mock.SetDeviceMountOptions.assert_has_calls([
Packit 792a06
        mock.call("/dev/sda1", "defaults,nodev"),
Packit 792a06
        mock.call("/dev/sda2", "defaults,noauto"),
Packit 792a06
    ])
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_add_mount_options(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    evaluation_add_mount_options(rule_data, ksdata_mock, storage_mock, 1)
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_add_mount_options_no_duplicates(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    evaluation_add_mount_options(rule_data, ksdata_mock, storage_mock, 2)
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_add_mount_options_report_only(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    rules = [
Packit 792a06
        "part /tmp --mountoptions=nodev",
Packit 792a06
        "part / --mountoptions=noauto",
Packit 792a06
    ]
Packit 792a06
    messages = get_messages_for_partition_rules(
Packit 792a06
        rule_data, ksdata_mock, storage_mock,
Packit 792a06
        rules, 1, report_only=True)
Packit 792a06
Packit 792a06
    # two mount options added --> two info messages
Packit 792a06
    assert len(messages) == 2
Packit 792a06
    assert messages[0].type == common.MESSAGE_TYPE_INFO
Packit 792a06
    assert messages[1].type == common.MESSAGE_TYPE_INFO
Packit 792a06
Packit 792a06
    # newly added mount options should be mentioned in the messages
Packit 792a06
    # together with their mount points
Packit 792a06
    nodev_found = False
Packit 792a06
    noauto_found = False
Packit 792a06
Packit 792a06
    for message in messages:
Packit 792a06
        if "'nodev'" in message.text:
Packit 792a06
            assert "/tmp" in message.text
Packit 792a06
            nodev_found = True
Packit 792a06
        elif "'noauto'" in message.text:
Packit 792a06
            assert "/" in message.text
Packit 792a06
            noauto_found = True
Packit 792a06
Packit 792a06
    assert all([nodev_found, noauto_found])
Packit 792a06
Packit 792a06
    # no changes should be made
Packit 792a06
    device_tree_mock = STORAGE.get_proxy(DEVICE_TREE)
Packit 792a06
    device_tree_mock.SetDeviceMountOptions.assert_not_called()
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_add_mount_option_prefix(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    rules = [
Packit 792a06
        "part /tmp --mountoptions=nodev",
Packit 792a06
        "part / --mountoptions=noauto",
Packit 792a06
    ]
Packit 792a06
    mount_options = {
Packit 792a06
        "/": "defaults",
Packit 792a06
        "/tmp": "defaults,nodevice",
Packit 792a06
    }
Packit 792a06
    messages = get_messages_for_partition_rules(
Packit 792a06
        rule_data, ksdata_mock, storage_mock,
Packit 792a06
        rules, mount_options=mount_options)
Packit 792a06
Packit 792a06
    # two mount options added (even though it is a prefix of another one)
Packit 792a06
    #   --> two info messages
Packit 792a06
    assert len(messages) == 2
Packit 792a06
    assert messages[0].type == common.MESSAGE_TYPE_INFO
Packit 792a06
    assert messages[1].type == common.MESSAGE_TYPE_INFO
Packit 792a06
Packit 792a06
    # the option should be added even though it is a prefix of another one
Packit 792a06
    device_tree_mock = STORAGE.get_proxy(DEVICE_TREE)
Packit 792a06
    device_tree_mock.SetDeviceMountOptions.assert_has_calls([
Packit 792a06
        mock.call("/dev/sda1", "defaults,nodevice,nodev"),
Packit 792a06
    ])
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_add_mount_options_nonexisting_part(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    rules = [
Packit 792a06
        "part /tmp --mountoptions=nodev",
Packit 792a06
        "part / --mountoptions=noauto",
Packit 792a06
    ]
Packit 792a06
    messages = get_messages_for_partition_rules(
Packit 792a06
        rule_data, ksdata_mock, storage_mock,
Packit 792a06
        rules, actual_mountpoints=["/"])
Packit 792a06
Packit 792a06
    # one mount option added, one mount point missing (mount options
Packit 792a06
    # cannot be added) --> one info, one error
Packit 792a06
    assert len(messages) == 2
Packit 792a06
    assert any(message.type == common.MESSAGE_TYPE_INFO for message in messages)
Packit 792a06
    assert any(message.type == common.MESSAGE_TYPE_FATAL for message in messages)
Packit 792a06
Packit 792a06
    # the info message should report mount options added to the existing
Packit 792a06
    # mount point, the error message shoud contain the missing mount point
Packit 792a06
    # and not the mount option
Packit 792a06
    for message in messages:
Packit 792a06
        if message.type == common.MESSAGE_TYPE_INFO:
Packit 792a06
            assert "/" in message.text
Packit 792a06
            assert "'noauto'" in message.text
Packit 792a06
        elif message.type == common.MESSAGE_TYPE_FATAL:
Packit 792a06
            assert "/tmp" in message.text
Packit 792a06
            assert "'nodev'" not in message.text
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_passwd_minlen_no_passwd(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    password_proxy_mock = USERS.get_proxy()
Packit 792a06
    password_proxy_mock.IsRootPasswordSet = False
Packit 792a06
    evaluation_passwd_minlen_no_passwd(rule_data, ksdata_mock, storage_mock, 8, (10, 11))
Packit 792a06
    evaluation_passwd_minlen_no_passwd(rule_data, ksdata_mock, storage_mock, 10, (8, 11))
Packit 792a06
    evaluation_passwd_minlen_no_passwd(rule_data, ksdata_mock, storage_mock, 11, (8, 10))
Packit 792a06
Packit 792a06
Packit 792a06
def evaluation_passwd_minlen_no_passwd(
Packit 792a06
        rule_data, ksdata_mock, storage_mock, min_password_length, check_against=tuple()):
Packit 792a06
    rule_data.new_rule("passwd --minlen={0}".format(min_password_length))
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # minimal password length required --> one warning
Packit 792a06
    assert len(messages) == 1
Packit 792a06
    assert messages[0].type == common.MESSAGE_TYPE_WARNING
Packit 792a06
Packit 792a06
    # warning has to mention the length
Packit 792a06
    assert str(min_password_length) in messages[0].text
Packit 792a06
Packit 792a06
    for not_wanted in check_against:
Packit 792a06
        assert str(not_wanted) not in messages[0].text
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_passwd_minlen_short_passwd(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    password_proxy_mock = USERS.get_proxy()
Packit 792a06
    password_proxy_mock.IsRootPasswordCrypted = False
Packit 792a06
    password_proxy_mock.RootPassword = "aaaa"
Packit 792a06
Packit 792a06
    rule_data.new_rule("passwd --minlen=8")
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock, report_only=False)
Packit 792a06
Packit 792a06
    # minimal password length greater than actual length --> one warning
Packit 792a06
    assert len(messages) == 1
Packit 792a06
    assert messages[0].type == common.MESSAGE_TYPE_FATAL
Packit 792a06
Packit 792a06
    # warning has to mention the length
Packit 792a06
    assert "8" in messages[0].text
Packit 792a06
Packit 792a06
    # warning should mention that something is wrong with the old password
Packit 792a06
    assert "is" in messages[0].text
Packit 792a06
Packit 792a06
    # doing changes --> password should not be cleared
Packit 792a06
    assert password_proxy_mock.RootPassword == "aaaa"
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_passwd_minlen_short_passwd_report_only(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    password_proxy_mock = USERS.get_proxy()
Packit 792a06
    password_proxy_mock.IsRootPasswordCrypted = False
Packit 792a06
    password_proxy_mock.RootPassword = "aaaa"
Packit 792a06
Packit 792a06
    rule_data.new_rule("passwd --minlen=8")
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock, report_only=True)
Packit 792a06
Packit 792a06
    # minimal password length greater than actual length --> one warning
Packit 792a06
    assert len(messages) == 1
Packit 792a06
    assert messages[0].type == common.MESSAGE_TYPE_FATAL
Packit 792a06
Packit 792a06
    # warning has to mention the length
Packit 792a06
    assert "8" in messages[0].text
Packit 792a06
Packit 792a06
    # warning should mention that something is wrong with the old password
Packit 792a06
    assert "is" in messages[0].text
Packit 792a06
Packit 792a06
    # doing changes --> password should not be cleared
Packit 792a06
    assert password_proxy_mock.RootPassword == "aaaa"
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_passwd_minlen_crypted_passwd(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    password_proxy_mock = USERS.get_proxy()
Packit 792a06
    password_proxy_mock.IsRootPasswordCrypted = True
Packit 792a06
    password_proxy_mock.RootPassword = "aaaa"
Packit 792a06
Packit 792a06
    rule_data.new_rule("passwd --minlen=8")
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock, report_only=False)
Packit 792a06
Packit 792a06
    # minimal password length greater than actual length --> one warning
Packit 792a06
    assert len(messages) == 1
Packit 792a06
    assert messages[0].type == common.MESSAGE_TYPE_WARNING
Packit 792a06
Packit 792a06
    # warning has to mention that the password cannot be checked
Packit 792a06
    assert "cannot check" in messages[0].text
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_passwd_minlen_good_passwd(proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    password_proxy_mock = USERS.get_proxy()
Packit 792a06
    password_proxy_mock.IsRootPasswordCrypted = False
Packit 792a06
    password_proxy_mock.RootPassword = "aaaaaaaaaaaaaaaaa"
Packit 792a06
Packit 792a06
    rule_data.new_rule("passwd --minlen=8")
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock, report_only=False)
Packit 792a06
Packit 792a06
    # minimal password length less than actual length --> no warning
Packit 792a06
    assert not messages
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_passwd_minlen_report_only_not_ignored(
Packit 792a06
        proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    password_proxy_mock = USERS.get_proxy()
Packit 792a06
    password_proxy_mock.IsRootPasswordCrypted = False
Packit 792a06
    password_proxy_mock.RootPassword = "aaaaaaaaaaaaaaaaa"
Packit 792a06
Packit 792a06
    rule_data.new_rule("passwd --minlen=8")
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock, report_only=False)
Packit 792a06
Packit 792a06
    # Mock pw_policy returned by anaconda.pwpolicy.get_policy()
Packit 792a06
    pw_policy_mock = mock.Mock()
Packit 792a06
    pw_policy_mock.minlen = 6
Packit 792a06
    pw_policy_mock.strict = False
Packit 792a06
    ksdata_mock.anaconda.pwpolicy.get_policy.return_value = pw_policy_mock
Packit 792a06
Packit 792a06
    # call eval_rules with report_only=False
Packit 792a06
    # should set password minimal length to 8
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock, report_only=False)
Packit 792a06
Packit 792a06
    # Password Policy changed --> no warnings
Packit 792a06
    assert not messages
Packit 792a06
    assert rule_data._passwd_rules._orig_minlen == 6
Packit 792a06
    assert not rule_data._passwd_rules._orig_strict
Packit 792a06
    assert pw_policy_mock.minlen == 8
Packit 792a06
    assert pw_policy_mock.strict
Packit 792a06
    assert rule_data._passwd_rules._minlen == 8
Packit 792a06
Packit 792a06
    # call of eval_rules with report_only=True
Packit 792a06
    # should not change anything
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock, report_only=True)
Packit 792a06
    # Password Policy stayed the same --> no warnings
Packit 792a06
    assert not messages
Packit 792a06
Packit 792a06
    assert rule_data._passwd_rules._orig_minlen == 6
Packit 792a06
    assert not rule_data._passwd_rules._orig_strict
Packit 792a06
    assert pw_policy_mock.minlen == 8
Packit 792a06
    assert pw_policy_mock.strict
Packit 792a06
    assert rule_data._passwd_rules._minlen == 8
Packit 792a06
Packit 792a06
Packit 792a06
def _occurences_not_seen_in_strings(seeked, strings):
Packit 792a06
    found = set(seeked)
Packit 792a06
    for string in strings:
Packit 792a06
        for might_have_seen in seeked:
Packit 792a06
            if might_have_seen in string:
Packit 792a06
                found.add(string)
Packit 792a06
                break
Packit 792a06
    return set(seeked).difference(found)
Packit 792a06
Packit 792a06
Packit 792a06
def _quoted_keywords_not_seen_in_messages(keywords, messages):
Packit 792a06
    return _occurences_not_seen_in_strings(
Packit 792a06
        {"'{}'".format(kw) for kw in keywords},
Packit 792a06
        [m.text for m in messages],
Packit 792a06
    )
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_package_rules(proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    rule_data.new_rule("package --add=firewalld --remove=telnet --add=iptables --add=vim")
Packit 792a06
Packit 792a06
    ksdata_mock.packages.packageList = ["vim"]
Packit 792a06
    ksdata_mock.packages.excludedList = []
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # one info message for each (really) added/removed package
Packit 792a06
    assert len(messages) == 3
Packit 792a06
    assert all(message.type == common.MESSAGE_TYPE_INFO for message in messages)
Packit 792a06
Packit 792a06
    # all packages should appear in the messages
Packit 792a06
    not_seen = _quoted_keywords_not_seen_in_messages(
Packit 792a06
        {"firewalld", "telnet", "iptables"},
Packit 792a06
        messages,
Packit 792a06
    )
Packit 792a06
Packit 792a06
    assert not not_seen
Packit 792a06
    assert set(ksdata_mock.packages.packageList) == {"firewalld", "iptables", "vim"}
Packit 792a06
    assert set(ksdata_mock.packages.excludedList) == {"telnet"}
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_package_rules_report_only(proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    rule_data.new_rule("package --add=firewalld --remove=telnet --add=iptables")
Packit 792a06
Packit 792a06
    ksdata_mock.packages.packageList = []
Packit 792a06
    ksdata_mock.packages.excludedList = []
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock, report_only=True)
Packit 792a06
Packit 792a06
    # one info message for each added/removed package
Packit 792a06
    assert len(messages) == 3
Packit 792a06
    assert all(message.type == common.MESSAGE_TYPE_INFO for message in messages)
Packit 792a06
Packit 792a06
    not_seen = _quoted_keywords_not_seen_in_messages(
Packit 792a06
        {"firewalld", "telnet", "iptables"},
Packit 792a06
        messages,
Packit 792a06
    )
Packit 792a06
Packit 792a06
    assert not not_seen
Packit 792a06
Packit 792a06
    # report_only --> no packages should be added or excluded
Packit 792a06
    assert not ksdata_mock.packages.packageList
Packit 792a06
    assert not ksdata_mock.packages.excludedList
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_bootloader_passwd_not_set(proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    bootloader_proxy_mock = STORAGE.get_proxy(BOOTLOADER)
Packit 792a06
    bootloader_proxy_mock.IsPasswordSet = False
Packit 792a06
Packit 792a06
    rule_data.new_rule("bootloader --passwd")
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # bootloader password not set --> one warning
Packit 792a06
    assert len(messages) == 1
Packit 792a06
    assert messages[0].type == common.MESSAGE_TYPE_WARNING
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_bootloader_passwd_set(proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    bootloader_proxy_mock = STORAGE.get_proxy(BOOTLOADER)
Packit 792a06
    bootloader_proxy_mock.IsPasswordSet = True
Packit 792a06
Packit 792a06
    rule_data.new_rule("bootloader --passwd")
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # bootloader password set --> no warnings
Packit 792a06
    assert messages == []
Packit 792a06
Packit 792a06
Packit 792a06
def test_evaluation_various_rules(proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    for rule in ["part /tmp", "part /", "passwd --minlen=14",
Packit 792a06
                 "package --add=firewalld", ]:
Packit 792a06
        rule_data.new_rule(rule)
Packit 792a06
Packit 792a06
    storage_mock.mountpoints = dict()
Packit 792a06
    ksdata_mock.packages.packageList = []
Packit 792a06
    ksdata_mock.packages.excludedList = []
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # four rules, all fail --> four messages
Packit 792a06
    assert len(messages) == 4
Packit 792a06
Packit 792a06
Packit 792a06
def test_revert_mount_options_nonexistent(proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    rule_data.new_rule("part /tmp --mountoptions=nodev")
Packit 792a06
    storage_mock.mountpoints = dict()
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # mount point doesn't exist -> one message, nothing done
Packit 792a06
    assert len(messages) == 1
Packit 792a06
    assert storage_mock.mountpoints == dict()
Packit 792a06
Packit 792a06
    # mount point doesn't exist -> shouldn't do anything
Packit 792a06
    rule_data.revert_changes(ksdata_mock, storage_mock)
Packit 792a06
    assert storage_mock.mountpoints == dict()
Packit 792a06
Packit 792a06
Packit 792a06
def test_revert_mount_options(proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    rule_data.new_rule("part /tmp --mountoptions=nodev")
Packit 792a06
    storage_mock.mountpoints = dict()
Packit 792a06
    storage_mock.mountpoints["/tmp"] = mock.Mock()
Packit 792a06
    storage_mock.mountpoints["/tmp"].format.options = "defaults"
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # mount option added --> one message
Packit 792a06
    assert len(messages) == 1
Packit 792a06
Packit 792a06
    # "nodev" option should be added
Packit 792a06
    assert storage_mock.mountpoints["/tmp"].format.options, "defaults == nodev"
Packit 792a06
Packit 792a06
    rule_data.revert_changes(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # should be reverted to the original value
Packit 792a06
    assert storage_mock.mountpoints["/tmp"].format.options == "defaults"
Packit 792a06
Packit 792a06
    # another cycle of the same #
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # mount option added --> one message
Packit 792a06
    assert len(messages) == 1
Packit 792a06
Packit 792a06
    # "nodev" option should be added
Packit 792a06
    assert storage_mock.mountpoints["/tmp"].format.options, "defaults == nodev"
Packit 792a06
Packit 792a06
    rule_data.revert_changes(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # should be reverted to the original value
Packit 792a06
    assert storage_mock.mountpoints["/tmp"].format.options == "defaults"
Packit 792a06
Packit 792a06
Packit 792a06
def test_revert_password_policy_changes(proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    password_proxy_mock = USERS.get_proxy()
Packit 792a06
    password_proxy_mock.IsRootPasswordCrypted = False
Packit 792a06
    password_proxy_mock.RootPassword = "aaaa"
Packit 792a06
Packit 792a06
    # FIXME: Add password policy changes to this test. It only checks
Packit 792a06
    # password length right now outside of policy changes.
Packit 792a06
    rule_data.new_rule("passwd --minlen=8")
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # password error --> one message
Packit 792a06
    assert len(messages) == 1
Packit 792a06
Packit 792a06
    rule_data.revert_changes(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # with long enough password this time #
Packit 792a06
    password_proxy_mock.RootPassword = "aaaaaaaaaaaaa"
Packit 792a06
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # long enough password
Packit 792a06
    # entered --> no message
Packit 792a06
    assert messages == []
Packit 792a06
Packit 792a06
Packit 792a06
def test_revert_package_rules(proxy_getter, rule_data, ksdata_mock, storage_mock):
Packit 792a06
    rule_data.new_rule("package --add=firewalld --remove=telnet --add=iptables --add=vim")
Packit 792a06
Packit 792a06
    ksdata_mock.packages.packageList = ["vim"]
Packit 792a06
    ksdata_mock.packages.excludedList = []
Packit 792a06
Packit 792a06
    # run twice --> nothing should be different in the second run
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # one info message for each added/removed package
Packit 792a06
    assert len(messages) == 3
Packit 792a06
Packit 792a06
    rule_data.revert_changes(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # (only) added and excluded packages should have been removed from the
Packit 792a06
    # list
Packit 792a06
    assert ksdata_mock.packages.packageList == ["vim"]
Packit 792a06
    assert ksdata_mock.packages.excludedList == []
Packit 792a06
Packit 792a06
    # now do the same again #
Packit 792a06
    messages = rule_data.eval_rules(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # one info message for each added/removed package
Packit 792a06
    assert len(messages) == 3
Packit 792a06
Packit 792a06
    rule_data.revert_changes(ksdata_mock, storage_mock)
Packit 792a06
Packit 792a06
    # (only) added and excluded packages should have been removed from the
Packit 792a06
    # list
Packit 792a06
    assert ksdata_mock.packages.packageList == ["vim"]
Packit 792a06
    assert ksdata_mock.packages.excludedList == []