Custom command crud (#38)

Author: gumuz

.gitignore

@@ -59,6 +59,6 @@ secrets.*sh
 .venv
 pip-selfcheck.json
 .idea
-
+.vscode
 *,cover
 .eggs

stream_chat/client.py

@@ -239,6 +239,21 @@ def channel(self, channel_type, channel_id=None, data=None):
         """
         return Channel(self, channel_type, channel_id, data)
 
+    def list_commands(self):
+        return self.get("commands")
+
+    def create_command(self, data):
+        return self.post("commands", data=data)
+
+    def delete_command(self, name):
+        return self.delete("commands/{}".format(name))
+
+    def get_command(self, name):
+        return self.get("commands/{}".format(name))
+
+    def update_command(self, name, **settings):
+        return self.put("commands/{}".format(name), data=settings)
+
     def add_device(self, device_id, push_provider, user_id):
         """
         Add a device to a user

stream_chat/tests/conftest.py

@@ -25,10 +25,13 @@ def pytest_configure(config):
 
 @pytest.fixture(scope="module")
 def client():
+    base_url = os.environ.get("STREAM_HOST")
+    options = {"base_url": base_url} if base_url else {}
     return StreamChat(
         api_key=os.environ["STREAM_KEY"],
         api_secret=os.environ["STREAM_SECRET"],
         timeout=10,
+        **options,
     )
 
 
@@ -67,6 +70,15 @@ def channel(client, random_user):
     return channel
 
 
+@pytest.fixture(scope="function")
+def command(client):
+    response = client.create_command(
+        dict(name=str(uuid.uuid4()), description="My command")
+    )
+
+    return response["command"]
+
+
 @pytest.fixture(scope="module")
 def fellowship_of_the_ring(client):
     members = [

stream_chat/tests/test_client.py

@@ -52,6 +52,24 @@ def test_update_channel_type(self, client):
         assert "commands" in response
         assert response["commands"] == ["ban", "unban"]
 
+    def test_get_command(self, client, command):
+        response = client.get_command(command["name"])
+        assert command["name"] == response["name"]
+
+    def test_update_command(self, client, command):
+        response = client.update_command(command["name"], description="My new command")
+        assert "command" in response
+        assert "My new command" == response["command"]["description"]
+
+    def test_delete_command(self, client, command):
+        response = client.delete_command(command["name"])
+        with pytest.raises(StreamAPIException):
+            client.get_command(command["name"])
+
+    def test_list_commands(self, client):
+        response = client.list_commands()
+        assert "commands" in response
+
     def test_create_token(self, client):
         token = client.create_token("tommaso")
         assert type(token) is str