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