add tests.test_admin_api.test_instance_invites

- admin_api.instance_invites: fixes

litecord/blueprints/admin_api/instance_invites.py

@@ -65,7 +65,7 @@ async def _all_instance_invites():
     rows = [dict(row) for row in rows]
 
     for row in rows:
-        row['timestamp'] = timestamp_(row['timestamp'])
+        row['created_at'] = timestamp_(row['created_at'])
 
     return jsonify(rows)
 
@@ -75,7 +75,7 @@ async def _create_invite():
     await admin_check()
 
     code = await gen_inv(app)
-    if not code:
+    if code is None:
         return 'failed to make invite', 500
 
     j = validate(await request.get_json(), INSTANCE_INVITE)
@@ -103,7 +103,7 @@ async def _del_invite(invite: str):
     WHERE code = $1
     """, invite)
 
-    if res.lower() == 'update 0':
+    if res.lower() == 'delete 0':
         return 'invite not found', 404
 
     return '', 204

tests/test_admin_api/test_instance_invites.py

@@ -0,0 +1,87 @@
+"""
+
+Litecord
+Copyright (C) 2018-2019  Luna Mendes
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, version 3 of the License.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+"""
+
+import pytest
+
+from tests.common import login
+
+async def _get_invs(test_cli, token=None):
+    if token is None:
+        token = await login('admin', test_cli)
+
+    resp = await test_cli.get('/api/v6/admin/instance/invites', headers={
+        'Authorization': token
+    })
+
+    assert resp.status_code == 200
+    rjson = await resp.json
+    assert isinstance(rjson, list)
+    return rjson
+
+
+@pytest.mark.asyncio
+async def test_get_invites(test_cli):
+    """Test the listing of instance invites."""
+    await _get_invs(test_cli)
+
+
+@pytest.mark.asyncio
+async def test_inv_delete_invalid(test_cli):
+    """Test errors happen when trying to delete a
+    non-existing instance invite."""
+    token = await login('admin', test_cli)
+    resp = await test_cli.delete(
+        '/api/v6/admin/instance/invites/aaaaaaaaaa',
+        headers={
+            'Authorization': token
+        }
+    )
+
+    assert resp.status_code == 404
+
+
+@pytest.mark.asyncio
+async def test_create_invite(test_cli):
+    """Test the creation of an instance invite, then listing it,
+    then deleting it."""
+    token = await login('admin', test_cli)
+    resp = await test_cli.put('/api/v6/admin/instance/invites', headers={
+        'Authorization': token
+    }, json={
+        'max_uses': 1
+    })
+
+    assert resp.status_code == 200
+    rjson = await resp.json
+    assert isinstance(rjson, dict)
+    code = rjson['code']
+
+    # assert that the invite is in the list
+    invites = await _get_invs(test_cli, token)
+    assert any(inv['code'] == code for inv in invites)
+
+    # delete it, and assert it worked
+    resp = await test_cli.delete(
+        f'/api/v6/admin/instance/invites/{code}',
+        headers={
+            'Authorization': token
+        }
+    )
+
+    assert resp.status_code == 204