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

TOOLS = {
    "read_register": {
        "required": ["addr"],
        "types": {"addr": int},
        "bounds": {"addr": (0, 15)},
    },
    "write_register": {
        "required": ["addr", "data"],
        "types": {"addr": int, "data": int},
        "bounds": {"addr": (0, 15), "data": (0, 255)},
    },
}

def validate(call):
    if not isinstance(call, dict):
        return "not-an-object"
    if "tool" not in call:
        return "missing-tool"
    if call["tool"] not in TOOLS:
        return "unknown-tool"
    args = call.get("arguments")
    if not isinstance(args, dict):
        return "bad-arguments"
    spec = TOOLS[call["tool"]]
    for key in spec["required"]:
        if key not in args:
            return "missing-" + key
    for key in spec["required"]:
        if type(args[key]) is not int:
            return "type-" + key
    for key in spec["required"]:
        lo, hi = spec["bounds"][key]
        if args[key] < lo or args[key] > hi:
            return "range-" + key
    return "ok"

CASES = [
    ({"tool": "read_register", "arguments": {"addr": 5}}, "ok"),
    ({"tool": "write_register", "arguments": {"addr": 2, "data": 255}}, "ok"),
    ({"tool": "write_register", "arguments": {"addr": 2, "data": 1, "tag": "x"}}, "ok"),
    ({"tool": "read_register", "arguments": {"addr": 15}}, "ok"),
    ({"tool": "zap_register", "arguments": {"addr": 1}}, "unknown-tool"),
    ({"tool": "read_register", "arguments": {}}, "missing-addr"),
    ({"tool": "read_register", "arguments": {"addr": True}}, "type-addr"),
    ({"tool": "write_register", "arguments": {"addr": 2, "data": "12"}}, "type-data"),
    ({"tool": "read_register", "arguments": {"addr": 16}}, "range-addr"),
    ({"tool": "write_register", "arguments": {"addr": 2, "data": 256}}, "range-data"),
    ({"tool": "read_register", "arguments": [5]}, "bad-arguments"),
]

bad = 0
accepted = 0
rejected = 0
for call, want in CASES:
    got = validate(call)
    if got != want:
        bad += 1
    if got == "ok":
        accepted += 1
    else:
        rejected += 1
if accepted != 4:
    bad += 1
if rejected != 7:
    bad += 1

if bad == 0:
    print("PASS: 13 validator checks accept 4 calls and reject 7 across 11 cases")
else:
    print("FAIL: " + str(bad) + " of 13 validator checks went wrong")
    raise SystemExit(1)