Merge branch 'updat-typing-and-generic-args' into 'master'

Update typing and generic args for plugins (step 1)

See merge request poezio/slixmpp!120

1 files changed, 49 insertions, 0 deletions

diff --git a/itests/test_vcard.py b/itests/test_vcard.py
new file mode 100644
index 00000000..800d5a5b
--- /dev/null
+++ b/itests/test_vcard.py
@@ -0,0 +1,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)