Misago/misago/acl/tests/test_acl_algebra.py

from misago.acl import algebra


def test_greatest_value_is_returned():
    assert algebra.greater(1, 3) == 3
    assert algebra.greater(4, 2) == 4
    assert algebra.greater(2, 2) == 2
    assert algebra.greater(True, False) == True


def test_greatest_or_zero_value_is_returned():
    assert algebra.greater_or_zero(1, 3) == 3
    assert algebra.greater_or_zero(4, 2) == 4
    assert algebra.greater_or_zero(2, 2) == 2
    assert algebra.greater_or_zero(True, False) == False
    assert algebra.greater_or_zero(2, 0) == 0
    assert algebra.greater_or_zero(0, 0) == 0
    assert algebra.greater_or_zero(0, 120) == 0


def test_lower_value_is_returned():
    assert algebra.lower(1, 3) == 1
    assert algebra.lower(4, 2) == 2
    assert algebra.lower(2, 2) == 2
    assert algebra.lower(True, False) == False


def test_lowest_non_zero_value_is_returned():
    assert algebra.lower_non_zero(1, 3) == 1
    assert algebra.lower_non_zero(0, 2) == 2
    assert algebra.lower_non_zero(1, 2) == 1
    assert algebra.lower_non_zero(0, 0) == 0


def test_acls_are_be_added_together():
    test_acls = [
        {
            "can_see": 0,
            "can_hear": 0,
            "max_speed": 10,
            "min_age": 16,
            "speed_limit": 50,
        },
        {"can_see": 1, "can_hear": 0, "max_speed": 40, "min_age": 20, "speed_limit": 0},
        {
            "can_see": 0,
            "can_hear": 1,
            "max_speed": 80,
            "min_age": 18,
            "speed_limit": 40,
        },
    ]

    defaults = {
        "can_see": 0,
        "can_hear": 0,
        "max_speed": 30,
        "min_age": 18,
        "speed_limit": 60,
    }

    acl = algebra.sum_acls(
        defaults,
        acls=test_acls,
        can_see=algebra.greater,
        can_hear=algebra.greater,
        max_speed=algebra.greater,
        min_age=algebra.lower,
        speed_limit=algebra.greater_or_zero,
    )

    assert acl["can_see"] == 1
    assert acl["can_hear"] == 1
    assert acl["max_speed"] == 80
    assert acl["min_age"] == 16
    assert acl["speed_limit"] == 0