implement invitations

Author: pterk

stream_chat/channel.py

@@ -212,11 +212,19 @@ def unban_user(self, target_id, **options):
             target_id, type=self.channel_type, id=self.id, **options
         )
 
-    def accept_invite(self, user_id):
-        raise NotImplementedError
+    def accept_invite(self, user_id, **data):
+        payload = add_user_id(data, user_id)
+        payload['accept_invite'] = True
+        response = self.client.post(self.url, data=payload)
+        self.custom_data = response['channel']
+        return response
 
-    def reject_invite(self, user_id):
-        raise NotImplementedError
+    def reject_invite(self, user_id, **data):
+        payload = add_user_id(data, user_id)
+        payload['reject_invite'] = True
+        response = self.client.post(self.url, data=payload)
+        self.custom_data = response['channel']
+        return response
 
     def send_file(self, url, name, user, content_type=None):
         return self.client.send_file("{}/file".format(self.url), url, name, user, content_type=content_type)

stream_chat/tests/conftest.py

@@ -18,6 +18,10 @@ def pytest_runtest_setup(item):
         if previousfailed is not None:
             pytest.xfail("previous test failed (%s)" % previousfailed.name)
 
+def pytest_configure(config):
+    config.addinivalue_line(
+        "markers", "incremental: mark test incremental"
+    )
 
 @pytest.fixture(scope="module")
 def client():

stream_chat/tests/test_channel.py

@@ -1,5 +1,9 @@
+import uuid
+
 import pytest
 
+from stream_chat.exceptions import StreamAPIException
+
 
 @pytest.mark.incremental
 class TestChannel(object):
@@ -157,7 +161,37 @@ def test_channel_hide_show(self, client, channel, random_users):
         response = client.query_channels({"id": channel.id}, user_id=random_users[0]['id'])
         assert len(response['channels']) == 0
         # send message
-        msg = channel.send_message({"text": "hi"}, random_users[1]["id"])
+        channel.send_message({"text": "hi"}, random_users[1]["id"])
         # channel should be listed now
         response = client.query_channels({"id": channel.id}, user_id=random_users[0]['id'])
         assert len(response['channels']) == 1
+
+    def test_invites(self, client, channel):
+        members = ["john", "paul", "george", "pete", "ringo", "eric"]
+        client.update_users([{"id": m} for m in members])
+        channel = client.channel(
+            "team", "beatles-" + str(uuid.uuid4()), {
+                "members": members,
+                "invites": ["ringo", "eric"]
+            })
+        channel.create("john")
+        # accept the invite when not a member
+        with pytest.raises(StreamAPIException):
+            accept = channel.accept_invite("brian")
+        # accept the invite when a member
+        accept = channel.accept_invite("ringo")
+        for m in accept['members']:
+            if m['user_id'] == 'ringo':
+                assert m['invited'] is True
+                assert "invite_accepted_at" in m
+        # cannot accept again
+        with pytest.raises(StreamAPIException):
+            _ = channel.accept_invite("ringo")
+        reject = channel.reject_invite("eric")
+        for m in reject['members']:
+            if m['user_id'] == 'eric':
+                assert m['invited'] is True
+                assert "invite_rejected_at" in m
+        # cannot accept again
+        with pytest.raises(StreamAPIException):
+            reject = channel.reject_invite("eric")