dify/api/tests/unit_tests/controllers/console/test_fastopenapi_system.py

import builtins
from unittest.mock import patch

import pytest
from flask.views import MethodView

from configs import dify_config
from dify_app import DifyApp
from extensions import ext_fastopenapi

if not hasattr(builtins, "MethodView"):
    builtins.MethodView = MethodView  # type: ignore[attr-defined]


@pytest.fixture
def app() -> DifyApp:
    app = DifyApp(__name__)
    app.config["TESTING"] = True
    return app


def test_console_ping_fastopenapi_returns_pong(app: DifyApp) -> None:
    ext_fastopenapi.init_app(app)

    response = app.test_client().get("/console/api/ping")

    assert response.status_code == 200
    assert response.get_json() == {"result": "pong"}


def test_console_version_fastopenapi_returns_current_version(app: DifyApp) -> None:
    ext_fastopenapi.init_app(app)

    with patch("controllers.console.system.dify_config.CHECK_UPDATE_URL", None):
        response = app.test_client().get("/console/api/version", query_string={"current_version": "0.0.0"})

    assert response.status_code == 200
    assert response.get_json() == {
        "version": dify_config.project.version,
        "release_notes": "",
    }