diff --git a/pyproject.toml b/pyproject.toml index d86cb9d..6484a11 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -23,7 +23,7 @@ classifiers = [ "Operating System :: POSIX :: Linux", ] dependencies = [ - "tmt~=1.36", + "tmt~=1.65", "fastapi~=0.115", "httpx~=0.27", "uvicorn~=0.30", diff --git a/src/tmt_web/formatters.py b/src/tmt_web/formatters.py index e9509a9..57ada1f 100644 --- a/src/tmt_web/formatters.py +++ b/src/tmt_web/formatters.py @@ -3,7 +3,7 @@ import json from tmt import Logger -from tmt.utils import dict_to_yaml +from tmt.utils import to_yaml from tmt_web.generators import html_generator from tmt_web.models import PlanData, TestData, TestPlanData @@ -20,7 +20,7 @@ def _format_yaml(data: TestData | PlanData | TestPlanData) -> str: data_dict = data.model_dump(by_alias=True) # Then use tmt's yaml formatter - return dict_to_yaml(data_dict) + return to_yaml(data_dict) def _format_html(data: TestData | PlanData | TestPlanData, logger: Logger) -> str: diff --git a/src/tmt_web/generators/yaml_generator.py b/src/tmt_web/generators/yaml_generator.py index c2979d8..fb66e15 100644 --- a/src/tmt_web/generators/yaml_generator.py +++ b/src/tmt_web/generators/yaml_generator.py @@ -1,7 +1,7 @@ from typing import Any from tmt import Logger, Plan, Test -from tmt.utils import GeneralError, dict_to_yaml +from tmt.utils import GeneralError, to_yaml from tmt_web.generators.json_generator import CombinedTestPlanModel, ObjectModel @@ -17,7 +17,7 @@ def _serialize_yaml(data: dict[str, Any], logger: Logger) -> str: """ try: logger.debug("Serializing data to YAML") - return dict_to_yaml(data) + return to_yaml(data) except Exception as err: logger.fail("Failed to serialize data to YAML") raise GeneralError("Failed to generate YAML output") from err diff --git a/tests/unit/test_yaml_generator.py b/tests/unit/test_yaml_generator.py index 8aedcf6..34d698a 100644 --- a/tests/unit/test_yaml_generator.py +++ b/tests/unit/test_yaml_generator.py @@ -78,10 +78,10 @@ def test_generate_testplan_yaml(self, test_obj, plan_obj, logger): def test_yaml_serialization_error(self, test_obj, logger, monkeypatch): """Test error handling when YAML serialization fails.""" - def mock_dict_to_yaml(*args, **kwargs): + def mock_to_yaml(*args, **kwargs): raise ValueError("YAML serialization failed") - monkeypatch.setattr(yaml_generator, "dict_to_yaml", mock_dict_to_yaml) + monkeypatch.setattr(yaml_generator, "to_yaml", mock_to_yaml) with pytest.raises(GeneralError) as exc: yaml_generator.generate_test_yaml(test_obj, logger)