1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
import shutil
import tempfile
import helpers
import pytest
from conftest_async import ( # noqa: F401
aioresponse,
async_client,
async_client_pair,
async_client_pair_same_user,
unauthed_async_client,
)
from olm import Account
from nio import Client, ClientConfig, HttpClient
from nio.crypto import Olm, OlmDevice
from nio.store import SqliteMemoryStore
ALICE_ID = "@alice:example.org"
ALICE_DEVICE_ID = "JLAFKJWSCS"
BOB_DEVICE = "@bob:example.org"
BOB_DEVICE_ID = "JLAFKJWSRS"
@pytest.fixture
def tempdir():
newpath = tempfile.mkdtemp()
yield newpath
shutil.rmtree(newpath)
@pytest.fixture
def client(tempdir):
return Client("ephemeral", "DEVICEID", tempdir)
@pytest.fixture
def client_no_e2e(tempdir):
config = ClientConfig(encryption_enabled=False)
return Client("ephemeral", "DEVICEID", tempdir, config)
@pytest.fixture
def olm_machine():
key_pair = Account().identity_keys
bob_device = OlmDevice(BOB_DEVICE, BOB_DEVICE_ID, key_pair)
store = SqliteMemoryStore(ALICE_ID, ALICE_DEVICE_ID)
client = Olm(ALICE_ID, ALICE_DEVICE_ID, store)
client.device_store.add(bob_device)
store.save_device_keys(client.device_store)
return client
@pytest.fixture
def alice_client(tempdir):
client = Client(ALICE_ID, ALICE_DEVICE_ID, tempdir)
client.user_id = ALICE_ID
return client
@pytest.fixture
def http_client(tempdir):
return HttpClient("example.org", "ephemeral", "DEVICEID", tempdir)
@pytest.fixture
def frame_factory():
return helpers.FrameFactory()
|
