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
|
from __future__ import annotations
from inline_snapshot import snapshot
from zabbix_cli.bulk import BulkRunnerMode
from zabbix_cli.commands.cli_configwizard import get_enum_attr_docs
from zabbix_cli.commands.cli_configwizard import is_enum_type
from zabbix_cli.config.constants import OutputFormat
def test_get_enum_attr_docs():
"""Test docstring retrieval for enum attributes."""
assert get_enum_attr_docs(BulkRunnerMode) == snapshot(
{
BulkRunnerMode.STRICT: "Stop on first error.",
BulkRunnerMode.CONTINUE: "Continue on errors, report at end.",
BulkRunnerMode.SKIP: "Skip lines with errors. No reporting.",
}
)
assert get_enum_attr_docs(OutputFormat) == snapshot(
{
OutputFormat.JSON: "JSON-serialized output.",
OutputFormat.TABLE: "Rich terminal table output.",
}
)
def test_is_enum():
"""Test if an object is an enum type."""
assert is_enum_type(BulkRunnerMode)
assert is_enum_type(OutputFormat)
assert not is_enum_type(str)
assert not is_enum_type(int)
assert not is_enum_type(list)
assert not is_enum_type(dict)
assert not is_enum_type(tuple)
# Instances
assert not is_enum_type(BulkRunnerMode.STRICT)
assert not is_enum_type(OutputFormat.JSON)
assert not is_enum_type("test")
assert not is_enum_type(123)
assert not is_enum_type(["test"])
assert not is_enum_type({"test": "test"})
assert not is_enum_type((1, 2, 3))
|
