mirror of https://github.com/zulip/zulip.git
115 lines
4.5 KiB
Python
115 lines
4.5 KiB
Python
from unittest.mock import patch
|
|
|
|
import orjson
|
|
from typing_extensions import override
|
|
|
|
from zerver.lib.bot_lib import EmbeddedBotQuitError
|
|
from zerver.lib.display_recipient import get_display_recipient
|
|
from zerver.lib.test_classes import ZulipTestCase
|
|
from zerver.models import UserProfile
|
|
from zerver.models.bots import get_service_profile
|
|
from zerver.models.realms import get_realm
|
|
from zerver.models.users import get_user
|
|
|
|
|
|
class TestEmbeddedBotMessaging(ZulipTestCase):
|
|
@override
|
|
def setUp(self) -> None:
|
|
super().setUp()
|
|
self.user_profile = self.example_user("othello")
|
|
self.bot_profile = self.create_test_bot(
|
|
"embedded",
|
|
self.user_profile,
|
|
full_name="Embedded bot",
|
|
bot_type=UserProfile.EMBEDDED_BOT,
|
|
service_name="helloworld",
|
|
config_data=orjson.dumps({"foo": "bar"}).decode(),
|
|
)
|
|
|
|
def test_pm_to_embedded_bot(self) -> None:
|
|
assert self.bot_profile is not None
|
|
self.send_personal_message(self.user_profile, self.bot_profile, content="help")
|
|
last_message = self.get_last_message()
|
|
self.assertEqual(last_message.content, "beep boop")
|
|
self.assertEqual(last_message.sender_id, self.bot_profile.id)
|
|
display_recipient = get_display_recipient(last_message.recipient)
|
|
assert isinstance(display_recipient, list)
|
|
self.assert_length(display_recipient, 1)
|
|
self.assertEqual(display_recipient[0]["email"], self.user_profile.email)
|
|
|
|
def test_stream_message_to_embedded_bot(self) -> None:
|
|
assert self.bot_profile is not None
|
|
self.send_stream_message(
|
|
self.user_profile,
|
|
"Denmark",
|
|
content=f"@**{self.bot_profile.full_name}** foo",
|
|
topic_name="bar",
|
|
)
|
|
last_message = self.get_last_message()
|
|
self.assertEqual(last_message.content, "beep boop")
|
|
self.assertEqual(last_message.sender_id, self.bot_profile.id)
|
|
self.assertEqual(last_message.topic_name(), "bar")
|
|
self.assert_message_stream_name(last_message, "Denmark")
|
|
|
|
def test_stream_message_not_to_embedded_bot(self) -> None:
|
|
self.send_stream_message(self.user_profile, "Denmark", content="foo", topic_name="bar")
|
|
last_message = self.get_last_message()
|
|
self.assertEqual(last_message.content, "foo")
|
|
|
|
def test_message_to_embedded_bot_with_initialize(self) -> None:
|
|
assert self.bot_profile is not None
|
|
self.subscribe(self.user_profile, "Denmark")
|
|
with patch(
|
|
"zulip_bots.bots.helloworld.helloworld.HelloWorldHandler.initialize", create=True
|
|
) as mock_initialize:
|
|
self.send_stream_message(
|
|
self.user_profile,
|
|
"Denmark",
|
|
content=f"@**{self.bot_profile.full_name}** foo",
|
|
topic_name="bar",
|
|
)
|
|
mock_initialize.assert_called_once()
|
|
|
|
def test_embedded_bot_quit_exception(self) -> None:
|
|
assert self.bot_profile is not None
|
|
with (
|
|
patch(
|
|
"zulip_bots.bots.helloworld.helloworld.HelloWorldHandler.handle_message",
|
|
side_effect=EmbeddedBotQuitError("I'm quitting!"),
|
|
),
|
|
self.assertLogs(level="WARNING") as m,
|
|
):
|
|
self.send_stream_message(
|
|
self.user_profile,
|
|
"Denmark",
|
|
content=f"@**{self.bot_profile.full_name}** foo",
|
|
topic_name="bar",
|
|
)
|
|
self.assertEqual(m.output, ["WARNING:root:I'm quitting!"])
|
|
|
|
|
|
class TestEmbeddedBotFailures(ZulipTestCase):
|
|
def test_message_embedded_bot_with_invalid_service(self) -> None:
|
|
user_profile = self.example_user("othello")
|
|
self.create_test_bot(
|
|
short_name="embedded",
|
|
user_profile=user_profile,
|
|
bot_type=UserProfile.EMBEDDED_BOT,
|
|
service_name="helloworld",
|
|
)
|
|
bot_profile = get_user("embedded-bot@zulip.testserver", get_realm("zulip"))
|
|
service_profile = get_service_profile(bot_profile.id, "helloworld")
|
|
service_profile.name = "invalid"
|
|
service_profile.save()
|
|
with self.assertLogs(level="ERROR") as m:
|
|
self.send_stream_message(
|
|
user_profile,
|
|
"Denmark",
|
|
content=f"@**{bot_profile.full_name}** foo",
|
|
topic_name="bar",
|
|
)
|
|
self.assertEqual(
|
|
m.output[0],
|
|
f"ERROR:root:Error: User {bot_profile.id} has bot with invalid embedded bot service invalid",
|
|
)
|