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
|
# Slixmpp: The Slick XMPP Library
# Copyright (C) 2020 Mathieu Pasquet
# This file is part of Slixmpp.
# See the file LICENSE for copying permission.
import asyncio
import os
try:
from unittest import IsolatedAsyncioTestCase
except ImportError:
# Python < 3.8
# just to make sure the imports do not break, but
# not usable.
from unittest import TestCase as IsolatedAsyncioTestCase
from typing import (
Dict,
List,
Optional,
)
from slixmpp import JID
from slixmpp.clientxmpp import ClientXMPP
class SlixIntegration(IsolatedAsyncioTestCase):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.clients = []
self.addAsyncCleanup(self._destroy)
def envjid(self, name):
"""Get a JID from an env var"""
value = os.getenv(name)
return JID(value)
def envstr(self, name):
"""get a str from an env var"""
return os.getenv(name)
def register_plugins(self, plugins: List[str], configs: Optional[List[Dict]] = None):
"""Register plugins on all known clients"""
for index, plugin in enumerate(plugins):
for client in self.clients:
if configs is not None:
client.register_plugin(plugin, pconfig=configs[index])
else:
client.register_plugin(plugin)
def add_client(self, jid: JID, password: str):
"""Register a new client"""
self.clients.append(ClientXMPP(jid, password))
async def connect_clients(self):
"""Connect all clients"""
for client in self.clients:
client.connect()
wait = [client.wait_until('session_start') for client in self.clients]
await asyncio.gather(*wait)
async def _destroy(self):
"""Kill all clients"""
for client in self.clients:
client.abort()
|
