From 1836b2b887a467afffe6e5e7ab408ffb55aa7c00 Mon Sep 17 00:00:00 2001
From: mathieui <mathieui@mathieui.net>
Date: Sun, 6 Dec 2020 18:58:35 +0100
Subject: itests: add another bunch of tests

---
 itests/test_disco.py | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)
 create mode 100644 itests/test_disco.py

(limited to 'itests/test_disco.py')

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)
-- 
cgit v1.2.3