summaryrefslogtreecommitdiff
path: root/itests/test_disco.py
blob: 95a07b17cbb5a32012bcd16989f59eb56c3f8758 (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
33
import unittest
from slixmpp.test.integration import SlixIntegration


class TestDisco(SlixIntegration):
    async def asyncSetUp(self):
        await super().asyncSetUp()
        self.add_client(
            self.envjid('CI_ACCOUNT1'),
            self.envstr('CI_ACCOUNT1_PASSWORD'),
        )
        self.add_client(
            self.envjid('CI_ACCOUNT2'),
            self.envstr('CI_ACCOUNT2_PASSWORD'),
        )
        self.register_plugins(['xep_0030'])
        await self.connect_clients()

    async def test_features(self):
        """Check we can add, get and delete a feature"""
        self.clients[0]['xep_0030'].add_feature('urn:xmpp:fake:0')
        info = await self.clients[1]['xep_0030'].get_info(
            self.clients[0].boundjid.full
        )
        self.assertIn('urn:xmpp:fake:0', info['disco_info']['features'])
        self.clients[0]['xep_0030'].del_feature(feature='urn:xmpp:fake:0')
        info = await self.clients[1]['xep_0030'].get_info(
            self.clients[0].boundjid.full
        )
        self.assertNotIn('urn:xmpp:fake:0', info['disco_info']['features'])


suite = unittest.TestLoader().loadTestsFromTestCase(TestDisco)