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