1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
# -*- coding: utf-8 -*-
# @Author : llc
# @Time : 2025/1/15 9:58
import pytest
from flask_openapi3 import APIBlueprint, OpenAPI, APIView
app = OpenAPI(__name__, )
app.config["TESTING"] = True
@pytest.fixture
def client():
client = app.test_client()
return client
api1 = APIBlueprint("/book1", __name__, url_prefix="/api")
api2 = APIBlueprint("/book2", __name__)
@api1.get("/book")
def create_book1():
return "ok"
@api2.get("/book")
def create_book2():
return "ok"
app.register_api(api1, url_prefix="/api1")
app.register_api(api2, url_prefix="/api2")
api_view1 = APIView(url_prefix="/api")
api_view2 = APIView()
@api_view1.route("/book")
class BookAPIView:
@api_view1.doc(summary="get book")
def get(self):
return "ok"
@api_view2.route("/book")
class BookAPIView2:
@api_view2.doc(summary="get book")
def get(self, ):
return "ok"
app.register_api_view(api_view1, url_prefix="/api3")
app.register_api_view(api_view2, url_prefix="/api4")
def test_openapi(client):
resp = client.get("/openapi/openapi.json")
_json = resp.json
assert "/api1/book" in _json["paths"].keys()
assert "/api2/book" in _json["paths"].keys()
assert "/api3/book" in _json["paths"].keys()
assert "/api4/book" in _json["paths"].keys()
|
