1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
from typing import Any, Dict
import pytest
from pytest import raises
from knot_resolver.datamodel.policy_schema import ActionSchema, PolicySchema
from knot_resolver.datamodel.types import PolicyActionEnum
from knot_resolver.utils.modeling.exceptions import DataValidationError
from knot_resolver.utils.modeling.types import get_generic_type_arguments
noconfig_actions = [
"pass",
"drop",
"refuse",
"tc",
"debug-always",
"debug-cache-miss",
"qtrace",
"reqtrace",
]
configurable_actions = ["deny", "reroute", "answer", "mirror", "forward", "stub"]
policy_actions = get_generic_type_arguments(PolicyActionEnum)
@pytest.mark.parametrize("val", [item for item in policy_actions if item not in configurable_actions])
def test_policy_action_valid(val: Any):
PolicySchema({"action": val})
ActionSchema({"action": val})
@pytest.mark.parametrize("val", [{"action": "invalid-action"}])
def test_action_invalid(val: Dict[str, Any]):
with raises(DataValidationError):
PolicySchema(val)
with raises(DataValidationError):
ActionSchema(val)
@pytest.mark.parametrize(
"val",
[
{"action": "deny", "message": "this is deny message"},
{
"action": "reroute",
"reroute": [
{"source": "192.0.2.0/24", "destination": "127.0.0.0"},
{"source": "10.10.10.0/24", "destination": "192.168.1.0"},
],
},
{"action": "answer", "answer": {"rtype": "AAAA", "rdata": "192.0.2.7"}},
{"action": "mirror", "servers": ["192.0.2.1@5353", "2001:148f:ffff::1"]},
{"action": "forward", "servers": ["192.0.2.1@5353", "2001:148f:ffff::1"]},
{"action": "stub", "servers": ["192.0.2.1@5353", "2001:148f:ffff::1"]},
{"action": "forward", "servers": [{"address": ["127.0.0.1@5353"]}]},
],
)
def test_policy_valid(val: Dict[str, Any]):
PolicySchema(val)
ActionSchema(val)
@pytest.mark.parametrize(
"val",
[
{"action": "reroute"},
{"action": "answer"},
{"action": "mirror"},
{"action": "pass", "reroute": [{"source": "192.0.2.0/24", "destination": "127.0.0.0"}]},
{"action": "pass", "answer": {"rtype": "AAAA", "rdata": "::1"}},
{"action": "pass", "servers": ["127.0.0.1@5353"]},
{"action": "mirror", "servers": [{"address": ["127.0.0.1@5353"]}]},
],
)
def test_policy_invalid(val: Dict[str, Any]):
with raises(DataValidationError):
PolicySchema(val)
with raises(DataValidationError):
ActionSchema(val)
@pytest.mark.parametrize(
"val",
noconfig_actions,
)
def test_policy_message_invalid(val: str):
with raises(DataValidationError):
PolicySchema({"action": f"{val}", "message": "this is deny message"})
with raises(DataValidationError):
ActionSchema({"action": f"{val}", "message": "this is deny message"})
|
