ServerVariable.enum should be optional (#194)

Co-authored-by: luolingchun <[email protected]>

Author: luolingchun

flask_openapi3/models/server_variable.py

@@ -3,15 +3,15 @@
 # @Time    : 2023/7/4 9:57
 from typing import List, Optional
 
-from pydantic import BaseModel
+from pydantic import BaseModel, Field
 
 
 class ServerVariable(BaseModel):
     """
     https://spec.openapis.org/oas/v3.1.0#server-variable-object
     """
 
-    enum: List[str]
+    enum: Optional[List[str]] = Field(None, min_length=1)
     default: str
     description: Optional[str] = None
 

tests/test_server.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# @Author  : llc
+# @Time    : 2024/11/10 12:17
+from pydantic import ValidationError
+
+from flask_openapi3 import Server, ServerVariable
+
+
+def test_server_variable():
+    Server(
+        url="http://127.0.0.1:5000",
+        variables=None
+    )
+    try:
+        variables = {"one": ServerVariable(default="one", enum=[])}
+        Server(
+            url="http://127.0.0.1:5000",
+            variables=variables
+        )
+        error = 0
+    except ValidationError:
+        error = 1
+    assert error == 1
+    try:
+        variables = {"one": ServerVariable(default="one")}
+        Server(
+            url="http://127.0.0.1:5000",
+            variables=variables
+        )
+        error = 0
+    except ValidationError:
+        error = 1
+    assert error == 0
+    try:
+        variables = {"one": ServerVariable(default="one", enum=["one", "two"])}
+        Server(
+            url="http://127.0.0.1:5000",
+            variables=variables
+        )
+        error = 0
+    except ValidationError:
+        error = 1
+    assert error == 0