29 lines
702 B
Python
29 lines
702 B
Python
from __future__ import annotations
|
|
|
|
from notebook_tools.paperless_client import _document_id_from_task_payload
|
|
|
|
|
|
def test_related_document_string() -> None:
|
|
assert (
|
|
_document_id_from_task_payload(
|
|
{
|
|
"related_document": "10",
|
|
"result": "Success. New document id 10 created",
|
|
}
|
|
)
|
|
== 10
|
|
)
|
|
|
|
|
|
def test_result_string_only() -> None:
|
|
assert (
|
|
_document_id_from_task_payload(
|
|
{"related_document": None, "result": "Success. New document id 42 created"}
|
|
)
|
|
== 42
|
|
)
|
|
|
|
|
|
def test_related_document_int() -> None:
|
|
assert _document_id_from_task_payload({"related_document": 7}) == 7
|