forked from missytake/team-bot
218 lines
8.2 KiB
Python
218 lines
8.2 KiB
Python
import os.path
|
|
|
|
import deltachat
|
|
import pytest
|
|
from deltachat.capi import lib as dclib
|
|
|
|
|
|
@pytest.mark.timeout(60)
|
|
def test_not_relay_groups(relaycrew, outsider):
|
|
bot = relaycrew.bot
|
|
user = relaycrew.user
|
|
|
|
# bot <-> outsider 1:1 chat
|
|
outsider_botcontact = outsider.create_contact(bot.get_config("addr"))
|
|
outsider_outside_chat = outsider.create_chat(outsider_botcontact)
|
|
outsider_outside_chat.send_text("test 1:1 message to bot")
|
|
|
|
bot_message_from_outsider = bot.wait_next_incoming_message()
|
|
bot_outside_chat = bot_message_from_outsider.chat
|
|
assert not bot.relayplugin.is_relay_group(bot_outside_chat)
|
|
|
|
# bot <-> outsider group chat
|
|
outsider_bot_group = outsider.create_group_chat(
|
|
"test with outsider", contacts=[outsider_botcontact]
|
|
)
|
|
outsider_bot_group.send_text("test message to outsider group")
|
|
bot_message_from_outsider = bot.wait_next_incoming_message()
|
|
assert not bot.relayplugin.is_relay_group(bot_message_from_outsider.chat)
|
|
|
|
# bot <-> user 1:1 chat
|
|
user_botcontact = user.create_contact(bot.get_config("addr"))
|
|
user_to_bot = user.create_chat(user_botcontact)
|
|
user_to_bot.send_text("test message to bot")
|
|
bot_message_from_user = bot.wait_next_incoming_message()
|
|
assert not bot.relayplugin.is_relay_group(bot_message_from_user.chat)
|
|
|
|
# bot <-> user group chat
|
|
user_group = user.create_group_chat("test with user", contacts=[user_botcontact])
|
|
user_group.send_text("testing message to user group")
|
|
bot_message_from_user = bot.wait_next_incoming_message()
|
|
assert not bot.relayplugin.is_relay_group(bot_message_from_user.chat)
|
|
|
|
|
|
@pytest.mark.timeout(60)
|
|
def test_relay_group_forwarding(relaycrew, outsider):
|
|
bot = relaycrew.bot
|
|
user = relaycrew.user
|
|
|
|
# create outside chat
|
|
outsider_botcontact = outsider.create_contact(bot.get_config("addr"))
|
|
outsider_outside_chat = outsider.create_chat(outsider_botcontact)
|
|
outsider_outside_chat.send_text("test 1:1 message to bot")
|
|
|
|
# get outside chat
|
|
message_from_outsider = bot.wait_next_incoming_message()
|
|
bot_outside_chat = message_from_outsider.chat
|
|
assert not bot.relayplugin.is_relay_group(bot_outside_chat)
|
|
|
|
# get relay group
|
|
user.wait_next_incoming_message() # group added message
|
|
user_forwarded_message_from_outsider = user.wait_next_incoming_message()
|
|
user_relay_group = user_forwarded_message_from_outsider.create_chat()
|
|
user_relay_group.send_text(
|
|
"Chatter in relay group"
|
|
) # send normal reply, not forwarded
|
|
bot_chatter_in_relay_group = bot.wait_next_incoming_message()
|
|
bot_relay_group = bot_chatter_in_relay_group.chat
|
|
|
|
# check if relay group has relay group properties
|
|
assert bot_relay_group.get_name().startswith(
|
|
"[%s] " % (bot.get_config("addr").split("@")[0],)
|
|
)
|
|
assert (
|
|
bot_relay_group.get_messages()[0].get_sender_contact() == bot.get_self_contact()
|
|
)
|
|
assert not bot_relay_group.is_protected()
|
|
assert relaycrew.get_contacts() == bot_relay_group.get_contacts()
|
|
assert bot.relayplugin.is_relay_group(bot_relay_group)
|
|
|
|
# send direct reply, should be forwarded
|
|
user_direct_reply = deltachat.Message.new_empty(user, view_type="text")
|
|
user_direct_reply.set_text("This should be forwarded to the outsider")
|
|
user_direct_reply.quote = user_forwarded_message_from_outsider
|
|
sent_id = dclib.dc_send_msg(
|
|
user._dc_context, user_relay_group.id, user_direct_reply._dc_msg
|
|
)
|
|
assert sent_id == user_direct_reply.id
|
|
|
|
# check that direct reply was forwarded to outsider
|
|
outsider_direct_reply = outsider.wait_next_incoming_message()
|
|
assert outsider_direct_reply.text == "This should be forwarded to the outsider"
|
|
assert outsider_direct_reply.chat == outsider_outside_chat
|
|
assert outsider_direct_reply.get_sender_contact() == outsider_botcontact
|
|
|
|
# check that normal reply was not forwarded to outsider
|
|
assert bot_chatter_in_relay_group.text not in [
|
|
msg.text for msg in bot_outside_chat.get_messages()
|
|
]
|
|
|
|
# reply with outsider
|
|
outsider_outside_chat.send_text("Second message by outsider")
|
|
|
|
# check that outsider's reply ends up in the same chat
|
|
user_second_message_from_outsider = user.wait_next_incoming_message()
|
|
assert user_second_message_from_outsider.chat == user_relay_group
|
|
|
|
# check that relay group explanation is not forwarded to outsider
|
|
for chat in outsider.get_chats():
|
|
for msg in chat.get_messages():
|
|
assert "This is the relay group for" not in msg.text
|
|
|
|
|
|
def test_default_outside_help(relaycrew, outsider):
|
|
bot = relaycrew.bot
|
|
user = relaycrew.user
|
|
|
|
# create outside chat
|
|
outsider_botcontact = outsider.create_contact(bot.get_config("addr"))
|
|
outsider_outside_chat = outsider.create_chat(outsider_botcontact)
|
|
outsider_outside_chat.send_text("/help")
|
|
|
|
# get response
|
|
outside_help_message = outsider.wait_next_incoming_message()
|
|
assert "I forward messages to the " in outside_help_message.text
|
|
|
|
# assert no relay group was created
|
|
assert len(bot.get_chats()) == 2
|
|
assert len(user.get_chats()) == 1
|
|
|
|
|
|
def test_empty_outside_help(relaycrew, outsider):
|
|
bot = relaycrew.bot
|
|
user = relaycrew.user
|
|
|
|
# set outside_help_message empty
|
|
for chat in user.get_chats():
|
|
print(chat.id, chat.get_name())
|
|
user_crew = user.get_chat_by_id(11)
|
|
assert user_crew.get_name().startswith("Team")
|
|
user_crew.send_text("/set_outside_help")
|
|
# ensure /set_outside_help arrives before sending /help
|
|
bot.wait_next_incoming_message()
|
|
|
|
# create outside chat
|
|
outsider_botcontact = outsider.create_contact(bot.get_config("addr"))
|
|
outsider_outside_chat = outsider.create_chat(outsider_botcontact)
|
|
outsider_outside_chat.send_text("/help")
|
|
|
|
# get forwarded /help message
|
|
user.wait_next_incoming_message() # group added message
|
|
user.wait_next_incoming_message() # explanation message
|
|
user_forwarded_message_from_outsider = user.wait_next_incoming_message()
|
|
assert user_forwarded_message_from_outsider.text == "/help"
|
|
|
|
|
|
def test_changed_outside_help(relaycrew, outsider):
|
|
bot = relaycrew.bot
|
|
user = relaycrew.user
|
|
|
|
# set outside_help_message empty
|
|
for chat in user.get_chats():
|
|
print(chat.id, chat.get_name())
|
|
user_crew = user.get_chat_by_id(11)
|
|
assert user_crew.get_name().startswith("Team")
|
|
outside_help_text = "Hi friend :) send me messages to chat with the team"
|
|
user_crew.send_text("/set_outside_help " + outside_help_text)
|
|
# ensure /set_outside_help arrives before sending /help
|
|
bot.wait_next_incoming_message()
|
|
|
|
# create outside chat
|
|
outsider_botcontact = outsider.create_contact(bot.get_config("addr"))
|
|
outsider_outside_chat = outsider.create_chat(outsider_botcontact)
|
|
outsider_outside_chat.send_text("/help")
|
|
|
|
# get response
|
|
outside_help_message = outsider.wait_next_incoming_message()
|
|
assert outside_help_message.text == outside_help_text
|
|
|
|
# assert no relay group was created
|
|
assert len(bot.get_chats()) == 2
|
|
assert len(user.get_chats()) == 1
|
|
|
|
|
|
def test_change_avatar(relaycrew):
|
|
bot = relaycrew.bot
|
|
user = relaycrew.user
|
|
|
|
for contact in user.get_contacts():
|
|
if contact.addr == bot.get_config("addr"):
|
|
assert not contact.get_profile_image()
|
|
botcontact = contact
|
|
break
|
|
else:
|
|
pytest.fail("bot contact not found")
|
|
|
|
example_png_path = "/usr/share/pixmaps/debian-logo.png"
|
|
if not os.path.exists(example_png_path):
|
|
pytest.skip(f"example image not available: {example_png_path}")
|
|
|
|
# set avatar to example image
|
|
for chat in user.get_chats():
|
|
print(chat.id, chat.get_name())
|
|
user_crew = user.get_chat_by_id(11)
|
|
assert user_crew.get_name().startswith("Team")
|
|
msg = deltachat.Message.new_empty(user, "image")
|
|
msg.set_text("/set_avatar")
|
|
msg.set_file(example_png_path)
|
|
sent_id = dclib.dc_send_msg(user._dc_context, user_crew.id, msg._dc_msg)
|
|
assert sent_id == msg.id
|
|
|
|
group_avatar_changed_msg = user.wait_next_incoming_message()
|
|
assert "Group image changed" in group_avatar_changed_msg.text
|
|
assert user_crew.get_profile_image()
|
|
|
|
confirmation_msg = user.wait_next_incoming_message()
|
|
assert confirmation_msg.text == "Avatar changed to this image."
|
|
assert botcontact.get_profile_image()
|