mirror of
https://github.com/matrix-org/pantalaimon.git
synced 2025-02-03 11:00:10 -05:00
tests: Add initial pan client tests.
This commit is contained in:
Damir Jelić
parent
22ee14e7b9
commit
c17a737c42
356
tests/pan_client_test.py
Normal file
356
tests/pan_client_test.py
Normal file
@ -0,0 +1,356 @@
|
||||
import asyncio
|
||||
import re
|
||||
|
||||
import janus
|
||||
import pytest
|
||||
from nio import (LoginResponse, SyncResponse, Timeline, RoomMemberEvent, Rooms,
|
||||
RoomInfo, DeviceOneTimeKeyCount, RoomEncryptionEvent, DeviceList,
|
||||
RoomSummary)
|
||||
|
||||
from pantalaimon.client import PanClient
|
||||
from pantalaimon.config import ServerConfig
|
||||
from pantalaimon.store import PanStore
|
||||
|
||||
TEST_ROOM_ID = "!SVkFJHzfwvuaIEawgC:localhost"
|
||||
TEST_ROOM2 = "!testroom:localhost"
|
||||
|
||||
ALICE_ID = "@alice:example.org"
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
async def client(tmpdir, loop):
|
||||
store = PanStore(tmpdir)
|
||||
queue = janus.Queue(loop=loop)
|
||||
conf = ServerConfig("example", "https://exapmle.org")
|
||||
|
||||
store.save_server_user("example", "@example:example.org")
|
||||
|
||||
pan_client = PanClient(
|
||||
"example",
|
||||
store,
|
||||
conf,
|
||||
"https://example.org",
|
||||
queue.async_q,
|
||||
"@example:example.org",
|
||||
"DEVICEID",
|
||||
tmpdir
|
||||
)
|
||||
|
||||
yield pan_client
|
||||
|
||||
await pan_client.close()
|
||||
|
||||
|
||||
class TestClass(object):
|
||||
@property
|
||||
def login_response(self):
|
||||
return LoginResponse.from_dict(
|
||||
{
|
||||
"access_token": "abc123",
|
||||
"device_id": "DEVICEID",
|
||||
"home_server": "example.org",
|
||||
"user_id": "@example:example.org"
|
||||
}
|
||||
)
|
||||
|
||||
@property
|
||||
def initial_sync_response(self):
|
||||
return {
|
||||
"device_one_time_keys_count": {},
|
||||
"next_batch": "s526_47314_0_7_1_1_1_11444_1",
|
||||
"device_lists": {
|
||||
"changed": [
|
||||
"@example:example.org"
|
||||
],
|
||||
"left": []
|
||||
},
|
||||
|
||||
"rooms": {
|
||||
"invite": {},
|
||||
"join": {
|
||||
"!SVkFJHzfwvuaIEawgC:localhost": {
|
||||
"account_data": {
|
||||
"events": []
|
||||
},
|
||||
"ephemeral": {
|
||||
"events": []
|
||||
},
|
||||
"state": {
|
||||
"events": [
|
||||
{
|
||||
"content": {
|
||||
"avatar_url": None,
|
||||
"displayname": "example",
|
||||
"membership": "join"
|
||||
},
|
||||
"event_id": "$151800140517rfvjc:localhost",
|
||||
"membership": "join",
|
||||
"origin_server_ts": 1518001405556,
|
||||
"sender": "@example:localhost",
|
||||
"state_key": "@example:localhost",
|
||||
"type": "m.room.member",
|
||||
"unsigned": {
|
||||
"age": 2970366338,
|
||||
"replaces_state": "$151800111315tsynI:localhost"
|
||||
}
|
||||
},
|
||||
{
|
||||
"content": {
|
||||
"history_visibility": "shared"
|
||||
},
|
||||
"event_id": "$15139375515VaJEY:localhost",
|
||||
"origin_server_ts": 1513937551613,
|
||||
"sender": "@example:localhost",
|
||||
"state_key": "",
|
||||
"type": "m.room.history_visibility",
|
||||
"unsigned": {
|
||||
"age": 7034220281
|
||||
}
|
||||
},
|
||||
{
|
||||
"content": {
|
||||
"creator": "@example:localhost"
|
||||
},
|
||||
"event_id": "$15139375510KUZHi:localhost",
|
||||
"origin_server_ts": 1513937551203,
|
||||
"sender": "@example:localhost",
|
||||
"state_key": "",
|
||||
"type": "m.room.create",
|
||||
"unsigned": {
|
||||
"age": 7034220691
|
||||
}
|
||||
},
|
||||
{
|
||||
"content": {
|
||||
"aliases": [
|
||||
"#tutorial:localhost"
|
||||
]
|
||||
},
|
||||
"event_id": "$15139375516NUgtD:localhost",
|
||||
"origin_server_ts": 1513937551720,
|
||||
"sender": "@example:localhost",
|
||||
"state_key": "localhost",
|
||||
"type": "m.room.aliases",
|
||||
"unsigned": {
|
||||
"age": 7034220174
|
||||
}
|
||||
},
|
||||
{
|
||||
"content": {
|
||||
"topic": "\ud83d\ude00"
|
||||
},
|
||||
"event_id": "$151957878228ssqrJ:localhost",
|
||||
"origin_server_ts": 1519578782185,
|
||||
"sender": "@example:localhost",
|
||||
"state_key": "",
|
||||
"type": "m.room.topic",
|
||||
"unsigned": {
|
||||
"age": 1392989709,
|
||||
"prev_content": {
|
||||
"topic": "test"
|
||||
},
|
||||
"prev_sender": "@example:localhost",
|
||||
"replaces_state": "$151957069225EVYKm:localhost"
|
||||
}
|
||||
},
|
||||
{
|
||||
"content": {
|
||||
"ban": 50,
|
||||
"events": {
|
||||
"m.room.avatar": 50,
|
||||
"m.room.canonical_alias": 50,
|
||||
"m.room.history_visibility": 100,
|
||||
"m.room.name": 50,
|
||||
"m.room.power_levels": 100
|
||||
},
|
||||
"events_default": 0,
|
||||
"invite": 0,
|
||||
"kick": 50,
|
||||
"redact": 50,
|
||||
"state_default": 50,
|
||||
"users": {
|
||||
"@example:localhost": 100
|
||||
},
|
||||
"users_default": 0
|
||||
},
|
||||
"event_id": "$15139375512JaHAW:localhost",
|
||||
"origin_server_ts": 1513937551359,
|
||||
"sender": "@example:localhost",
|
||||
"state_key": "",
|
||||
"type": "m.room.power_levels",
|
||||
"unsigned": {
|
||||
"age": 7034220535
|
||||
}
|
||||
},
|
||||
{
|
||||
"content": {
|
||||
"alias": "#tutorial:localhost"
|
||||
},
|
||||
"event_id": "$15139375513VdeRF:localhost",
|
||||
"origin_server_ts": 1513937551461,
|
||||
"sender": "@example:localhost",
|
||||
"state_key": "",
|
||||
"type": "m.room.canonical_alias",
|
||||
"unsigned": {
|
||||
"age": 7034220433
|
||||
}
|
||||
},
|
||||
{
|
||||
"content": {
|
||||
"avatar_url": None,
|
||||
"displayname": "example2",
|
||||
"membership": "join"
|
||||
},
|
||||
"event_id": "$152034824468gOeNB:localhost",
|
||||
"membership": "join",
|
||||
"origin_server_ts": 1520348244605,
|
||||
"sender": "@example2:localhost",
|
||||
"state_key": "@example2:localhost",
|
||||
"type": "m.room.member",
|
||||
"unsigned": {
|
||||
"age": 623527289,
|
||||
"prev_content": {
|
||||
"membership": "leave"
|
||||
},
|
||||
"prev_sender": "@example:localhost",
|
||||
"replaces_state": "$152034819067QWJxM:localhost"
|
||||
}
|
||||
},
|
||||
{
|
||||
"content": {
|
||||
"algorithm": "m.megolm.v1.aes-sha2",
|
||||
"rotation_period_ms": 604800000,
|
||||
"rotation_period_msgs": 100
|
||||
},
|
||||
"event_id": "$143273582443PhrSn:example.org",
|
||||
"origin_server_ts": 1432735824653,
|
||||
"room_id": "!jEsUZKDJdhlrceRyVU:example.org",
|
||||
"sender": "@example:example.org",
|
||||
"state_key": "",
|
||||
"type": "m.room.encryption",
|
||||
"unsigned": {
|
||||
"age": 1234
|
||||
}
|
||||
}
|
||||
]
|
||||
},
|
||||
"timeline": {
|
||||
"events": [
|
||||
{
|
||||
"content": {
|
||||
"body": "baba",
|
||||
"format": "org.matrix.custom.html",
|
||||
"formatted_body": "<strong>baba</strong>",
|
||||
"msgtype": "m.text"
|
||||
},
|
||||
"event_id": "$152037280074GZeOm:localhost",
|
||||
"origin_server_ts": 1520372800469,
|
||||
"sender": "@example:localhost",
|
||||
"type": "m.room.message",
|
||||
"unsigned": {
|
||||
"age": 598971425
|
||||
}
|
||||
}
|
||||
],
|
||||
"limited": True,
|
||||
"prev_batch": "t392-516_47314_0_7_1_1_1_11444_1"
|
||||
},
|
||||
"unread_notifications": {
|
||||
"highlight_count": 0,
|
||||
"notification_count": 11
|
||||
}
|
||||
}
|
||||
},
|
||||
"leave": {}
|
||||
},
|
||||
"to_device": {
|
||||
"events": []
|
||||
}
|
||||
}
|
||||
|
||||
@property
|
||||
def keys_upload_response(self):
|
||||
return {
|
||||
"one_time_key_counts": {
|
||||
"curve25519": 10,
|
||||
"signed_curve25519": 20
|
||||
}
|
||||
}
|
||||
|
||||
@property
|
||||
def keys_query_response(self):
|
||||
return {
|
||||
"device_keys": {
|
||||
"@alice:example.org": {
|
||||
"JLAFKJWSCS": {
|
||||
"algorithms": [
|
||||
"m.olm.v1.curve25519-aes-sha2",
|
||||
"m.megolm.v1.aes-sha2"
|
||||
],
|
||||
"device_id": "JLAFKJWSCS",
|
||||
"user_id": "@alice:example.org",
|
||||
"keys": {
|
||||
"curve25519:JLAFKJWSCS": "wjLpTLRqbqBzLs63aYaEv2Boi6cFEbbM/sSRQ2oAKk4",
|
||||
"ed25519:JLAFKJWSCS": "nE6W2fCblxDcOFmeEtCHNl8/l8bXcu7GKyAswA4r3mM"
|
||||
},
|
||||
"signatures": {
|
||||
"@alice:example.org": {
|
||||
"ed25519:JLAFKJWSCS": "m53Wkbh2HXkc3vFApZvCrfXcX3AI51GsDHustMhKwlv3TuOJMj4wistcOTM8q2+e/Ro7rWFUb9ZfnNbwptSUBA"
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
},
|
||||
"failures": {}
|
||||
}
|
||||
|
||||
@property
|
||||
def empty_sync(self):
|
||||
return SyncResponse(
|
||||
"tokenf#a#inf",
|
||||
Rooms(),
|
||||
DeviceOneTimeKeyCount(50, 50),
|
||||
DeviceList([], []),
|
||||
[]
|
||||
)
|
||||
|
||||
async def test_login(self, client):
|
||||
await client.receive_response(self.login_response)
|
||||
assert client.logged_in
|
||||
|
||||
async def test_start_loop(self, client, aioresponse):
|
||||
sync_url = re.compile(
|
||||
r'^https://example\.org/_matrix/client/r0/sync\?access_token=.*'
|
||||
)
|
||||
|
||||
aioresponse.get(
|
||||
sync_url,
|
||||
status=200,
|
||||
payload=self.initial_sync_response,
|
||||
repeat=True
|
||||
)
|
||||
|
||||
aioresponse.post(
|
||||
"https://example.org/_matrix/client/r0/keys/upload?access_token=abc123",
|
||||
status=200,
|
||||
payload=self.keys_upload_response,
|
||||
repeat=True
|
||||
)
|
||||
|
||||
aioresponse.post(
|
||||
"https://example.org/_matrix/client/r0/keys/query?access_token=abc123",
|
||||
status=200,
|
||||
payload=self.keys_query_response,
|
||||
repeat=True
|
||||
)
|
||||
|
||||
await client.receive_response(self.login_response)
|
||||
|
||||
client.start_loop(100)
|
||||
|
||||
await client.synced.wait()
|
||||
await client.synced.wait()
|
||||
|
||||
assert not client.history_fetch_queue.empty()
|
||||
|
||||
await client.loop_stop()
|
||||
Loading…
x
Reference in New Issue
Block a user