PersonalPublic/gpt_chat_bot
1
0
Fork 0
mirror of https://github.com/Balshgit/gpt_chat_bot.git synced 2025-12-16 21:20:39 +03:00
Code Issues Packages Projects Releases Wiki Activity

update dependencies (#86)

Browse Source 
* update dependencies

* format code
This commit is contained in:
Dmitry Afanasyev
2024-02-01 22:29:47 +03:00
committed by GitHub
parent 7c51a775c6
commit a18c94882a
9 changed files with 494 additions and 542 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
bot_microservice/tests/integration/bot/test_bot_updates.py
View File

@@ -381,11 +381,14 @@ async def test_ask_question_action_bot_user_not_exists(
message = BotMessageFactory.create_instance(text="Привет!")
user = message["from"]
with mock.patch.object(
telegram._bot.Bot, "send_message", return_value=lambda *args, **kwargs: (args, kwargs)
) as mocked_send_message, mocked_ask_question_api(
host=test_settings.GPT_BASE_HOST,
return_value=Response(status_code=httpx.codes.OK, text="Привет! Как я могу помочь вам сегодня?"),
with (
mock.patch.object(
telegram._bot.Bot, "send_message", return_value=lambda *args, **kwargs: (args, kwargs)
) as mocked_send_message,
mocked_ask_question_api(
host=test_settings.GPT_BASE_HOST,
return_value=Response(status_code=httpx.codes.OK, text="Привет! Как я могу помочь вам сегодня?"),
),
):
bot_update = BotUpdateFactory(message=message)
bot_update["message"].pop("entities")
@@ -436,11 +439,14 @@ async def test_ask_question_action_bot_user_already_exists(
users = dbsession.query(User).all()
assert len(users) == 1
with mock.patch.object(
telegram._bot.Bot, "send_message", return_value=lambda *args, **kwargs: (args, kwargs)
) as mocked_send_message, mocked_ask_question_api(
host=test_settings.GPT_BASE_HOST,
return_value=Response(status_code=httpx.codes.OK, text="Привет! Как я могу помочь вам сегодня?"),
with (
mock.patch.object(
telegram._bot.Bot, "send_message", return_value=lambda *args, **kwargs: (args, kwargs)
) as mocked_send_message,
mocked_ask_question_api(
host=test_settings.GPT_BASE_HOST,
return_value=Response(status_code=httpx.codes.OK, text="Привет! Как я могу помочь вам сегодня?"),
),
):
bot_update = BotUpdateFactory(message=message)
bot_update["message"].pop("entities")
@@ -496,12 +502,15 @@ async def test_ask_question_action_user_is_banned(
ban_reason="test reason",
)
with mock.patch.object(
telegram._bot.Bot, "send_message", return_value=lambda *args, **kwargs: (args, kwargs)
) as mocked_send_message, mocked_ask_question_api(
host=test_settings.GPT_BASE_HOST,
return_value=Response(status_code=httpx.codes.OK, text="Привет! Как я могу помочь вам сегодня?"),
assert_all_called=False,
with (
mock.patch.object(
telegram._bot.Bot, "send_message", return_value=lambda *args, **kwargs: (args, kwargs)
) as mocked_send_message,
mocked_ask_question_api(
host=test_settings.GPT_BASE_HOST,
return_value=Response(status_code=httpx.codes.OK, text="Привет! Как я могу помочь вам сегодня?"),
assert_all_called=False,
),
):
bot_update = BotUpdateFactory(message=message)
bot_update["message"].pop("entities")
@@ -531,10 +540,13 @@ async def test_ask_question_action_not_success(
test_settings: AppSettings,
) -> None:
ChatGptModelFactory.create_batch(size=3)
with mock.patch.object(
telegram._bot.Bot, "send_message", return_value=lambda *args, **kwargs: (args, kwargs)
) as mocked_send_message, mocked_ask_question_api(
host=test_settings.GPT_BASE_HOST, return_value=Response(status_code=httpx.codes.INTERNAL_SERVER_ERROR)
with (
mock.patch.object(
telegram._bot.Bot, "send_message", return_value=lambda *args, **kwargs: (args, kwargs)
) as mocked_send_message,
mocked_ask_question_api(
host=test_settings.GPT_BASE_HOST, return_value=Response(status_code=httpx.codes.INTERNAL_SERVER_ERROR)
),
):
bot_update = BotUpdateFactory(message=BotMessageFactory.create_instance(text="Привет!"))
bot_update["message"].pop("entities")
@@ -557,11 +569,14 @@ async def test_ask_question_action_critical_error(
test_settings: AppSettings,
) -> None:
ChatGptModelFactory.create_batch(size=3)
with mock.patch.object(
telegram._bot.Bot, "send_message", return_value=lambda *args, **kwargs: (args, kwargs)
) as mocked_send_message, mocked_ask_question_api(
host=test_settings.GPT_BASE_HOST,
side_effect=Exception(),
with (
mock.patch.object(
telegram._bot.Bot, "send_message", return_value=lambda *args, **kwargs: (args, kwargs)
) as mocked_send_message,
mocked_ask_question_api(
host=test_settings.GPT_BASE_HOST,
side_effect=Exception(),
),
):
bot_update = BotUpdateFactory(message=BotMessageFactory.create_instance(text="Привет!"))
bot_update["message"].pop("entities")