|
3 | 3 | import pytest |
4 | 4 | from fastapi import Depends, FastAPI, HTTPException |
5 | 5 | from fastapi.testclient import TestClient |
6 | | -from pydantic import ValidationError |
| 6 | +from pydantic import BaseModel, ValidationError |
7 | 7 |
|
8 | 8 | from stac_fastapi.api.models import create_get_request_model, create_post_request_model |
9 | 9 | from stac_fastapi.extensions.core import FieldsExtension, FilterExtension, SortExtension |
10 | | -from stac_fastapi.types.search import BaseSearchGetRequest, BaseSearchPostRequest |
| 10 | +from stac_fastapi.types.search import ( |
| 11 | + APIRequest, |
| 12 | + BaseSearchGetRequest, |
| 13 | + BaseSearchPostRequest, |
| 14 | +) |
11 | 15 |
|
12 | 16 |
|
13 | 17 | def test_create_get_request_model(): |
14 | 18 | request_model = create_get_request_model( |
15 | 19 | extensions=[FilterExtension(), FieldsExtension()], |
16 | 20 | base_model=BaseSearchGetRequest, |
17 | 21 | ) |
| 22 | + assert type(request_model) is type |
| 23 | + assert issubclass(request_model, APIRequest) |
18 | 24 |
|
19 | 25 | model = request_model( |
20 | 26 | collections="test1,test2", |
@@ -96,6 +102,7 @@ def test_create_post_request_model(filter_val, passes): |
96 | 102 | extensions=[FilterExtension(), FieldsExtension()], |
97 | 103 | base_model=BaseSearchPostRequest, |
98 | 104 | ) |
| 105 | + assert issubclass(request_model, BaseModel) |
99 | 106 |
|
100 | 107 | if not passes: |
101 | 108 | with pytest.raises(ValidationError): |
|
0 commit comments