Sergey Antropoff
72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
"""UPID examples and round-trip properties."""
|
|
|
|
import pytest
|
|
from hypothesis import given
|
|
from hypothesis import strategies as st
|
|
|
|
from app.tasks.upid import Upid
|
|
|
|
SAFE = st.from_regex(r"[a-z0-9][a-z0-9_-]{0,19}", fullmatch=True)
|
|
|
|
|
|
@given(
|
|
node=SAFE,
|
|
pid=st.integers(min_value=0, max_value=0xFFFFFFFF),
|
|
process_start=st.integers(min_value=0, max_value=0xFFFFFFFF),
|
|
start_time=st.integers(min_value=0, max_value=0xFFFFFFFF),
|
|
task_type=SAFE,
|
|
task_id=st.text(alphabet="abcdefghijklmnopqrstuvwxyz0123456789_-", max_size=20),
|
|
user=SAFE,
|
|
)
|
|
def test_upid_round_trip(
|
|
node: str,
|
|
pid: int,
|
|
process_start: int,
|
|
start_time: int,
|
|
task_type: str,
|
|
task_id: str,
|
|
user: str,
|
|
) -> None:
|
|
upid = Upid(node, pid, process_start, start_time, task_type, task_id, user)
|
|
|
|
assert Upid.parse(str(upid)) == upid
|
|
|
|
|
|
def test_known_upid_shape() -> None:
|
|
value = "UPID:pve1:0000002A:00000010:65A1B2C3:qmstart:100:root@pam:"
|
|
|
|
parsed = Upid.parse(value)
|
|
|
|
assert parsed.pid == 42
|
|
assert parsed.task_id == "100"
|
|
assert str(parsed) == value
|
|
|
|
|
|
@pytest.mark.parametrize("value", ["", "UPID:broken", "UPID:pve:GGGGGGGG:00000000:00000000:x::u:"])
|
|
def test_invalid_upids_are_rejected(value: str) -> None:
|
|
with pytest.raises(ValueError):
|
|
Upid.parse(value)
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"kwargs",
|
|
[
|
|
{"pid": -1},
|
|
{"node": "bad:node"},
|
|
{"task_id": "bad:id"},
|
|
],
|
|
)
|
|
def test_invalid_upid_components_are_rejected(kwargs: dict[str, object]) -> None:
|
|
values: dict[str, object] = {
|
|
"node": "pve1",
|
|
"pid": 1,
|
|
"process_start": 1,
|
|
"start_time": 1,
|
|
"task_type": "test",
|
|
"task_id": "100",
|
|
"user": "root@pam",
|
|
}
|
|
values.update(kwargs)
|
|
with pytest.raises(ValueError):
|
|
Upid(**values) # type: ignore[arg-type]
|