mirror of
https://github.com/Balshgit/gpt_chat_bot.git
synced 2025-09-11 22:30:41 +03:00
23 lines
668 B
Python
23 lines
668 B
Python
from contextlib import contextmanager
|
|
from typing import Any, Iterator
|
|
|
|
import respx
|
|
from httpx import Response
|
|
|
|
from constants import CHAT_GPT_BASE_URI
|
|
|
|
|
|
@contextmanager
|
|
def mocked_ask_question_api(
|
|
host: str, return_value: Response | None = None, side_effect: Any | None = None
|
|
) -> Iterator[respx.MockRouter]:
|
|
with respx.mock(
|
|
assert_all_mocked=True,
|
|
assert_all_called=True,
|
|
base_url=host,
|
|
) as respx_mock:
|
|
ask_question_route = respx_mock.post(url=CHAT_GPT_BASE_URI, name="ask_question")
|
|
ask_question_route.return_value = return_value
|
|
ask_question_route.side_effect = side_effect
|
|
yield respx_mock
|