mirror of
https://github.com/langgenius/dify.git
synced 2026-11-13 23:06:40 +08:00
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""GET /openapi/v1/oauth/device/lookup is the canonical user-code lookup."""
|
|
|
|
import builtins
|
|
|
|
import pytest
|
|
from flask import Flask
|
|
from flask.views import MethodView
|
|
|
|
from controllers.openapi import bp as openapi_bp
|
|
from controllers.openapi.oauth_device import OAuthDeviceLookupApi
|
|
|
|
if not hasattr(builtins, "MethodView"):
|
|
builtins.MethodView = MethodView # type: ignore[attr-defined]
|
|
|
|
|
|
@pytest.fixture
|
|
def openapi_app() -> Flask:
|
|
app = Flask(__name__)
|
|
app.config["TESTING"] = True
|
|
app.register_blueprint(openapi_bp)
|
|
return app
|
|
|
|
|
|
def test_openapi_route_registered(openapi_app: Flask) -> None:
|
|
rules = {r.rule for r in openapi_app.url_map.iter_rules()}
|
|
assert "/openapi/v1/oauth/device/lookup" in rules
|
|
|
|
|
|
def test_route_dispatches_to_class(openapi_app: Flask) -> None:
|
|
rule = next(r for r in openapi_app.url_map.iter_rules() if r.rule == "/openapi/v1/oauth/device/lookup")
|
|
view = openapi_app.view_functions[rule.endpoint]
|
|
assert hasattr(view, "view_class")
|
|
assert view.view_class is OAuthDeviceLookupApi
|
|
|
|
|
|
def test_route_accepts_get(openapi_app: Flask) -> None:
|
|
rule = next(r for r in openapi_app.url_map.iter_rules() if r.rule == "/openapi/v1/oauth/device/lookup")
|
|
assert rule.methods is not None
|
|
assert "GET" in rule.methods
|