14
14
import json
15
15
from tempfile import NamedTemporaryFile
16
16
from pathlib import Path
17
- from typing import Any , Type , TypeVar
17
+ from typing import Any , TypeVar
18
18
19
19
20
20
@pytest .fixture
@@ -49,7 +49,7 @@ def dev_mode_enabled(monkeypatch: pytest.MonkeyPatch) -> None:
49
49
@pytest .fixture
50
50
def create_server_mock (
51
51
mocker : MockerFixture ,
52
- ) -> Generator [ Callable [[int , BaseModel ], None ], None , None ]:
52
+ ) -> Callable [[int , BaseModel ], None ]:
53
53
def _mock_server (_status : int , _body : BaseModel ) -> None :
54
54
def _sideeffect (
55
55
method : str ,
@@ -67,7 +67,7 @@ def _sideeffect(
67
67
68
68
mocker .patch ("urllib3.PoolManager.request" , side_effect = _sideeffect )
69
69
70
- yield _mock_server
70
+ return _mock_server
71
71
72
72
73
73
T = TypeVar ("T" , bound = BaseModel )
@@ -76,13 +76,13 @@ def _sideeffect(
76
76
@pytest .fixture
77
77
def create_osparc_response_model (
78
78
osparc_openapi_specs : dict [str , Any ],
79
- ) -> Generator [ Callable [[Type [T ]], T ], None , None ]:
80
- def _create_model (model_type : Type [T ]) -> T :
79
+ ) -> Callable [[type [T ]], T ]:
80
+ def _create_model (model_type : type [T ]) -> T :
81
81
schemas = osparc_openapi_specs .get ("components" , {}).get ("schemas" , {})
82
82
example_data = schemas .get (model_type .__name__ , {}).get ("example" , {})
83
83
error_msg = "Could not extract example data for"
84
84
error_msg += f" '{ model_type .__name__ } ' from openapi specs"
85
85
assert example_data , error_msg
86
86
return model_type .model_validate (example_data )
87
87
88
- yield _create_model
88
+ return _create_model
0 commit comments