from importlib.metadata import PackageNotFoundError

import pytest

import arm_statements.version as version_module
from arm_statements.version import statements_version


def test_version_from_distribution(monkeypatch: pytest.MonkeyPatch) -> None:
    def distribution_version(name: str) -> str:
        assert name == "pylib-ai-arm-statements"
        return "0.1.2"

    monkeypatch.setattr(version_module, "version", distribution_version)
    assert statements_version() == "0.1.2"


def test_version_without_metadata_is_none(monkeypatch: pytest.MonkeyPatch) -> None:
    def missing_distribution(name: str) -> str:
        raise PackageNotFoundError(name)

    monkeypatch.setattr(version_module, "version", missing_distribution)
    assert statements_version() is None