diff --git a/services_backend/routes/models/button.py b/services_backend/routes/models/button.py index 59cca46..1eca9de 100644 --- a/services_backend/routes/models/button.py +++ b/services_backend/routes/models/button.py @@ -17,7 +17,7 @@ class ButtonUpdate(Base): name: str | None = Field(description='Название кнопки') order: int | None = Field(description='Порядок, в котором отображаются кнопки') link: str | None = Field(description='Ссылка, на которую перенаправляет кнопка') - type: str | None = Field(description='Тип открываемой ссылки (Ссылка приложения/Браузер в приложении/Браузер') + type: Type | None = Field(description='Тип открываемой ссылки (Ссылка приложения/Браузер в приложении/Браузер') class ButtonGet(Base): @@ -26,4 +26,4 @@ class ButtonGet(Base): icon: str | None = Field(description='Иконка кнопки') name: str | None = Field(description='Название кнопки') link: str | None = Field(description='Ссылка, на которую перенаправляет кнопка') - type: str | None = Field(description='Тип открываемой ссылки (Ссылка приложения/Браузер в приложении/Браузер') + type: Type | None = Field(description='Тип открываемой ссылки (Ссылка приложения/Браузер в приложении/Браузер') diff --git a/tests/api/button.py b/tests/api/button.py index b841b23..e2d667d 100644 --- a/tests/api/button.py +++ b/tests/api/button.py @@ -153,3 +153,13 @@ def test_delete_order(self, db_button, client, db_category): res = client.get(f"/category/{db_category.id}/button/{db_button.id}") assert res.json()['order'] == 1 + + def test_type_not_enum(self, client, dbsession, db_category): + body = { + "icon": "test", + "name": "new", + "link": "test", + "type": "lmao", + } + res = client.post(f"/category/{db_category.id}/button/", data=json.dumps(body)) + assert res.status_code == status.HTTP_422_UNPROCESSABLE_ENTITY