itests: add another bunch of tests

1 files changed, 33 insertions, 0 deletions

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