From 2cb2fcefbf3aaed2596d4c49423ad3285fbb277d Mon Sep 17 00:00:00 2001
From: mathieui <mathieui@mathieui.net>
Date: Fri, 4 Dec 2020 23:04:49 +0100
Subject: Add some very basic integration tests

---
 itests/__init__.py                       |  0
 itests/test_basic_connect_and_message.py | 28 ++++++++++++
 itests/test_muc.py                       | 78 ++++++++++++++++++++++++++++++++
 3 files changed, 106 insertions(+)
 create mode 100644 itests/__init__.py
 create mode 100644 itests/test_basic_connect_and_message.py
 create mode 100644 itests/test_muc.py

(limited to 'itests')

diff --git a/itests/__init__.py b/itests/__init__.py
new file mode 100644
index 00000000..e69de29b
diff --git a/itests/test_basic_connect_and_message.py b/itests/test_basic_connect_and_message.py
new file mode 100644
index 00000000..79ad37be
--- /dev/null
+++ b/itests/test_basic_connect_and_message.py
@@ -0,0 +1,28 @@
+import unittest
+from slixmpp.test.integration import SlixIntegration
+
+
+class TestConnect(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'),
+        )
+        await self.connect_clients()
+
+    async def test_send_message(self):
+        """Make sure we can send and receive messages"""
+        msg = self.clients[0].make_message(
+            mto=self.clients[1].boundjid, mbody='Msg body',
+        )
+        msg.send()
+        message = await self.clients[1].wait_until('message')
+        self.assertEqual(message['body'], msg['body'])
+
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestConnect)
diff --git a/itests/test_muc.py b/itests/test_muc.py
new file mode 100644
index 00000000..3dc91955
--- /dev/null
+++ b/itests/test_muc.py
@@ -0,0 +1,78 @@
+import asyncio
+import unittest
+from uuid import uuid4
+from slixmpp import JID
+from slixmpp.test.integration import SlixIntegration
+
+UNIQUE = uuid4().hex
+
+
+class TestConnect(SlixIntegration):
+
+    async def asyncSetUp(self):
+        self.mucserver = self.envjid('CI_MUC_SERVER')
+        self.muc = JID('%s@%s' % (UNIQUE, self.mucserver))
+        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_0045'])
+        await self.connect_clients()
+
+    async def test_initial_join(self):
+        """Check that we can connect to a new muc"""
+        self.clients[0]['xep_0045'].join_muc(self.muc, 'client1')
+        presence = await self.clients[0].wait_until('muc::%s::got_online' % self.muc)
+        self.assertEqual(presence['muc']['affiliation'], 'owner')
+
+    async def test_setup_muc(self):
+        """Check that sending the initial room config and affiliation list works"""
+        self.clients[0]['xep_0045'].join_muc(self.muc, 'client1')
+        presence = await self.clients[0].wait_until('muc::%s::got_online' % self.muc)
+        self.assertEqual(presence['muc']['affiliation'], 'owner')
+        # Send initial configuration
+        config = await self.clients[0]['xep_0045'].get_room_config(self.muc)
+        values = config.get_values()
+        values['muc#roomconfig_persistentroom'] = False
+        values['muc#roomconfig_membersonly'] = True
+        config['values'] = values
+        config.reply()
+        config = await self.clients[0]['xep_0045'].set_room_config(self.muc, config)
+
+        # Send affiliation list including client 2
+        await self.clients[0]['xep_0045'].send_affiliation_list(
+            self.muc,
+            [
+                (self.clients[1].boundjid.bare, 'member'),
+            ],
+        )
+
+    async def test_join_after_config(self):
+        """Join a room after being added to the affiliation list"""
+        await self.test_setup_muc()
+        self.clients[1]['xep_0045'].join_muc(self.muc, 'client2')
+        await self.clients[1].wait_until('muc::%s::got_online' % self.muc)
+
+    async def test_leave(self):
+        """Check that we leave properly"""
+        await self.test_join_after_config()
+        self.clients[0]['xep_0045'].leave_muc(self.muc, 'client1', 'boooring')
+        pres = await self.clients[1].wait_until('muc::%s::got_offline' % self.muc)
+        self.assertEqual(pres['status'], 'boooring')
+        self.assertEqual(pres['type'], 'unavailable')
+
+
+    async def test_kick(self):
+        """Test kicking a user"""
+        await self.test_join_after_config()
+        await asyncio.gather(
+            self.clients[0].wait_until('muc::%s::got_offline' % self.muc),
+            self.clients[0]['xep_0045'].set_role(self.muc, 'client2', 'none')
+        )
+
+
+suite = unittest.TestLoader().loadTestsFromTestCase(TestConnect)
-- 
cgit v1.2.3