__init__.py

from pathlib import Path

from fastapi import APIRouter

from ..base_classes import AbstractIntegrationTestHelper, TestCase


class VersionUpdate(TestCase):
    def get_router(self) -> APIRouter:
        router = APIRouter()

        @router.get("/pypi.org/pypi/pydantic/json")
        def handle_pypi_json_api():
            return {
                # rest omitted
                "info": {"name": "pydantic", "version": "2.10.2"}
            }

        return router

    def prepare(self, helper: AbstractIntegrationTestHelper):
        feedstock_dir = Path(__file__).parent / "resources" / "feedstock"
        helper.overwrite_feedstock_contents("pydantic", feedstock_dir)

        feedstock_v1_dir = Path(__file__).parent / "resources" / "feedstock_v1"
        helper.overwrite_feedstock_contents("pydantic", feedstock_v1_dir, branch="1.x")

    def validate(self, helper: AbstractIntegrationTestHelper):
        helper.assert_version_pr_present_v0(
            "pydantic",
            new_version="2.10.2",
            new_hash="2bc2d7f17232e0841cbba4641e65ba1eb6fafb3a08de3a091ff3ce14a197c4fa",
            old_version="2.10.1",
            old_hash="a4daca2dc0aa429555e0656d6bf94873a7dc5f54ee42b1f5873d666fb3f35560",
        )


ALL_TEST_CASES: list[TestCase] = [VersionUpdate()]