mirror of
https://github.com/SaraSmiseth/prosody.git
synced 2025-05-02 12:37:06 +00:00
81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
import aioxmpp
|
|
import aioxmpp.dispatcher
|
|
import asyncio
|
|
import pytest
|
|
|
|
@pytest.fixture
|
|
def client(client_username):
|
|
|
|
jid = aioxmpp.JID.fromstr(client_username)
|
|
password = "12345678"
|
|
|
|
client = aioxmpp.PresenceManagedClient(
|
|
jid,
|
|
aioxmpp.make_security_layer(
|
|
password,
|
|
no_verify=True
|
|
),
|
|
)
|
|
return client
|
|
|
|
@pytest.fixture
|
|
def client_with_message_dispatcher(client):
|
|
def message_received(msg):
|
|
print(msg)
|
|
print(msg.body)
|
|
assert msg.body == "Hello World!"
|
|
|
|
# obtain an instance of the service
|
|
message_dispatcher = client.summon(
|
|
aioxmpp.dispatcher.SimpleMessageDispatcher
|
|
)
|
|
|
|
# register a message callback here
|
|
message_dispatcher.register_callback(
|
|
aioxmpp.MessageType.CHAT,
|
|
None,
|
|
message_received,
|
|
)
|
|
return client
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize("client_username", ["admin@localhost"])
|
|
async def test_send_message_from_admin_to_user(client):
|
|
recipient_jid = aioxmpp.JID.fromstr("user@localhost")
|
|
async with client.connected() as stream:
|
|
msg = aioxmpp.Message(
|
|
to=recipient_jid,
|
|
type_=aioxmpp.MessageType.CHAT,
|
|
)
|
|
# None is for "default language"
|
|
msg.body[None] = "Hello World!"
|
|
|
|
await client.send(msg)
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize("client_username", ["admin@localhost"])
|
|
async def test_send_message_from_admin_to_user2(client):
|
|
recipient_jid = aioxmpp.JID.fromstr("user2@localhost")
|
|
async with client.connected() as stream:
|
|
msg = aioxmpp.Message(
|
|
to=recipient_jid,
|
|
type_=aioxmpp.MessageType.CHAT,
|
|
)
|
|
# None is for "default language"
|
|
msg.body[None] = "Hello World!"
|
|
|
|
await client.send(msg)
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize("client_username", ["user@localhost"])
|
|
async def test_send_message_from_user_to_user2(client):
|
|
recipient_jid = aioxmpp.JID.fromstr("user2@localhost")
|
|
async with client.connected() as stream:
|
|
msg = aioxmpp.Message(
|
|
to=recipient_jid,
|
|
type_=aioxmpp.MessageType.CHAT,
|
|
)
|
|
# None is for "default language"
|
|
msg.body[None] = "Hello World!"
|
|
|
|
await client.send(msg)
|