summaryrefslogtreecommitdiff
path: root/itests/test_vcard.py
blob: 800d5a5becb096ee29091eef0bdc3020c5b693e9 (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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
import unittest
from slixmpp.test.integration import SlixIntegration


class TestVcardTemp(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_0054'])
        await self.connect_clients()

    async def _clear_vcard(self):
        # cleanup
        await self.clients[0]['xep_0054'].publish_vcard(
            self.clients[0]['xep_0054'].make_vcard()
        )

    async def test_vcard(self):
        """Check we can set and get a vcard"""
        await self._clear_vcard()

        # Check that vcard is empty
        recv = await self.clients[1]['xep_0054'].get_vcard(
            self.clients[0].boundjid.bare
        )
        self.assertEqual(recv['vcard_temp']['TITLE'], None)

        vcard = self.clients[0]['xep_0054'].make_vcard()
        vcard['TITLE'] = 'Coucou coucou'
        await self.clients[0]['xep_0054'].publish_vcard(
            vcard,
        )
        #
        recv = await self.clients[1]['xep_0054'].get_vcard(
            self.clients[0].boundjid.bare
        )
        self.assertEqual(recv['vcard_temp']['TITLE'], 'Coucou coucou')

        await self._clear_vcard()


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