33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
|
import unittest
|
||
|
|
from slixmpp import JID
|
||
|
|
from slixmpp.test.integration import SlixIntegration
|
||
|
|
|
||
|
|
|
||
|
|
class TestBlocking(SlixIntegration):
|
||
|
|
async def asyncSetUp(self):
|
||
|
|
await super().asyncSetUp()
|
||
|
|
self.add_client(
|
||
|
|
self.envjid('CI_ACCOUNT1'),
|
||
|
|
self.envstr('CI_ACCOUNT1_PASSWORD'),
|
||
|
|
)
|
||
|
|
self.register_plugins(['xep_0191'])
|
||
|
|
await self.connect_clients()
|
||
|
|
|
||
|
|
async def test_blocking(self):
|
||
|
|
"""Check we can block, unblock, and list blocked"""
|
||
|
|
await self.clients[0]['xep_0191'].block(
|
||
|
|
[JID('toto@example.com'), JID('titi@example.com')]
|
||
|
|
)
|
||
|
|
blocked = {JID('toto@example.com'), JID('titi@example.com')}
|
||
|
|
iq = await self.clients[0]['xep_0191'].get_blocked()
|
||
|
|
self.assertEqual(iq['blocklist']['items'], blocked)
|
||
|
|
|
||
|
|
info = await self.clients[0]['xep_0191'].unblock(
|
||
|
|
blocked,
|
||
|
|
)
|
||
|
|
iq = await self.clients[0]['xep_0191'].get_blocked()
|
||
|
|
self.assertEqual(len(iq['blocklist']['items']), 0)
|
||
|
|
|
||
|
|
|
||
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(TestBlocking)
|