mirror of https://github.com/langgenius/dify.git
36 lines
1019 B
Python
36 lines
1019 B
Python
import builtins
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
from flask import Flask
|
|
from flask.views import MethodView
|
|
|
|
from configs import dify_config
|
|
from extensions import ext_fastopenapi
|
|
|
|
if not hasattr(builtins, "MethodView"):
|
|
builtins.MethodView = MethodView # type: ignore[attr-defined]
|
|
|
|
|
|
@pytest.fixture
|
|
def app() -> Flask:
|
|
app = Flask(__name__)
|
|
app.config["TESTING"] = True
|
|
return app
|
|
|
|
|
|
def test_console_version_fastopenapi_returns_current_version(app: Flask):
|
|
ext_fastopenapi.init_app(app)
|
|
|
|
with patch("controllers.console.version.dify_config.CHECK_UPDATE_URL", None):
|
|
client = app.test_client()
|
|
response = client.get("/console/api/version", query_string={"current_version": "0.0.0"})
|
|
|
|
assert response.status_code == 200
|
|
data = response.get_json()
|
|
assert data["version"] == dify_config.project.version
|
|
assert data["release_date"] == ""
|
|
assert data["release_notes"] == ""
|
|
assert data["can_auto_update"] is False
|
|
assert "features" in data
|