summaryrefslogtreecommitdiff
path: root/itests/test_blocking.py
blob: 7954c1dc08de57562893ae47aa2a48d221e16793 (plain)
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
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)