Added backend unit tests

Author: vsdudakov

tests/api/__init__.py

@@ -0,0 +1,8 @@
+from fastadmin.api.helpers import sanitize
+
+
+async def test_sanitize():
+    assert sanitize("true") is True
+    assert sanitize("false") is False
+    assert sanitize("null") is None
+    assert sanitize("foo") == "foo"

tests/api/test_helpers.py

@@ -0,0 +1,46 @@
+import asyncio
+import pytest
+from tortoise import Tortoise
+
+from tests.api.tortoise.models import Tournament, Event, User
+
+
+
+@pytest.fixture(scope="session", autouse=True)
+async def tortoise_db():
+    await Tortoise.init(
+        db_url='sqlite://:memory:',
+        modules={'models': ['tests.api.tortoise.models']}
+    )
+    await Tortoise.generate_schemas()
+    yield Tortoise.get_connection('default')
+    await Tortoise.close_connections()
+
+
+@pytest.fixture
+async def superuser():
+    user = await User.create(username='Test User', password='password', is_superuser=True)
+    yield user
+    await user.delete()
+
+
+@pytest.fixture
+async def user():
+    user = await User.create(username='Test User', password='password')
+    yield user
+    await user.delete()
+
+
+@pytest.fixture
+async def tournament():
+    tournament = await Tournament.create(name='Test Tournament')
+    yield tournament
+    await tournament.delete()
+
+
+@pytest.fixture
+async def event(tournament, user):
+    event = await Event.create(name='Test Event', tournament=tournament)
+    await event.participants.add(user)
+    yield event
+    await event.delete()

tests/api/tortoise/__init__.py

@@ -0,0 +1,23 @@
+from fastadmin import TortoiseModelAdmin, register
+
+
+async def sign_in(client, user):
+
+    @register(user.__class__)
+    class UserAdmin(TortoiseModelAdmin):
+        async def authenticate(self, username, password):
+            obj = await user.__class__.filter(username=username, password=password, is_superuser=True).first()
+            if not obj:
+                return None
+            return obj.id
+
+    r = await client.post(
+        "/api/sign-in",
+        json={
+            "username": user.username,
+            "password": user.password,
+        }
+    )
+    assert r.status_code == 200
+    assert r.json() is None
+    return r.cookies

tests/api/tortoise/conftest.py

@@ -0,0 +1,37 @@
+from tortoise.models import Model
+from tortoise import fields
+
+
+class BaseModel(Model):
+    id = fields.IntField(pk=True)
+    created_at = fields.DatetimeField(auto_now_add=True)
+    updated_at = fields.DatetimeField(auto_now=True)
+
+    class Meta:
+        abstract = True
+
+
+class Tournament(BaseModel):
+    name = fields.CharField(max_length=255)
+
+    def __str__(self):
+        return self.name
+
+
+class Event(BaseModel):
+    name = fields.CharField(max_length=255)
+
+    tournament = fields.ForeignKeyField('models.Tournament', related_name='events')
+    participants = fields.ManyToManyField('models.User', related_name='events')
+
+    def __str__(self):
+        return self.name
+
+
+class User(BaseModel):
+    username = fields.CharField(max_length=255)
+    password = fields.CharField(max_length=255)
+    is_superuser = fields.BooleanField(default=False)
+
+    def __str__(self):
+        return self.username

tests/api/tortoise/helpers.py

@@ -0,0 +1,55 @@
+from fastadmin import TortoiseModelAdmin, register
+
+from tests.api.tortoise.helpers import sign_in
+from fastadmin.models.helpers import unregister_admin_model
+
+
+async def test_add(superuser, tournament, event, client):
+
+    @register(event.__class__)
+    class EventAdmin(TortoiseModelAdmin):
+        pass
+
+    await sign_in(client, superuser)
+    r = await client.post(
+        f"/api/add/{event.__class__.__name__}",
+        json={
+            "name": "new name",
+            "tournament_id": tournament.id,
+            "participants": [superuser.id],
+        }
+    )
+    assert r.status_code == 200
+    item = r.json()
+    event = await event.__class__.get(id=item["id"])
+    assert item["name"] ==  "new name"
+    assert item["tournament_id"] == tournament.id
+    assert item["created_at"] == event.created_at.isoformat()
+    assert item["updated_at"] == event.updated_at.isoformat()
+    assert item["participants"] == [superuser.id]
+
+
+async def test_add_401(superuser, tournament, event, client):
+    r = await client.post(
+        f"/api/add/{event.__class__.__name__}",
+        json={
+            "name": "new name",
+            "tournament_id": tournament.id,
+            "participants": [superuser.id],
+        }
+    )
+    assert r.status_code == 401
+
+
+async def test_add_404(superuser, tournament, event, client):
+    unregister_admin_model([event.__class__])
+    await sign_in(client, superuser)
+    r = await client.post(
+        f"/api/add/{event.__class__.__name__}",
+        json={
+            "name": "new name",
+            "tournament_id": tournament.id,
+            "participants": [superuser.id],
+        }
+    )
+    assert r.status_code == 404

tests/api/tortoise/models.py

@@ -0,0 +1,72 @@
+from tests.api.tortoise.helpers import sign_in
+from fastadmin import TortoiseModelAdmin, register
+
+from fastadmin.models.helpers import unregister_admin_model
+
+
+async def test_api_sign_in_401_invalid_password(superuser, client):
+    class UserAdmin(TortoiseModelAdmin):
+        async def authenticate(self, username, password):
+            obj = await superuser.__class__.filter(username=username, password=password, is_superuser=True).first()
+            if not obj:
+                return None
+            return obj.id
+
+    r = await client.post(
+        "/api/sign-in",
+        json={
+            "username": superuser.username,
+            "password": "invalid",
+        }
+    )
+    assert r.status_code == 401
+
+
+async def test_api_sign_in_401(superuser, client):
+    class UserAdmin(TortoiseModelAdmin):
+        async def authenticate(self, username, password):
+            obj = await superuser.__class__.filter(username=username, password=password, is_superuser=True).first()
+            if not obj:
+                return None
+            return obj.id
+
+    unregister_admin_model([superuser.__class__])
+
+    r = await client.post(
+        "/api/sign-in",
+        json={
+            "username": superuser.username,
+            "password": superuser.password,
+        }
+    )
+    assert r.status_code == 401
+
+
+async def test_api_sign_in(superuser, client):
+    assert await sign_in(client, superuser)
+
+
+async def test_api_me(superuser, client):
+    await sign_in(client, superuser)
+
+    r = await client.get(
+        "/api/me",
+    )
+    assert r.status_code == 200
+    me = r.json()
+    assert me["id"] == superuser.id
+    assert me["username"] == superuser.username
+
+
+async def test_api_sign_out(superuser, client):
+    await sign_in(client, superuser)
+
+    r = await client.post(
+        "/api/sign-out",
+    )
+    assert r.status_code == 200
+
+    r = await client.get(
+        "/api/me",
+    )
+    assert r.status_code == 401

tests/api/tortoise/test_add.py

@@ -0,0 +1,54 @@
+from fastadmin import TortoiseModelAdmin, register
+
+from tests.api.tortoise.helpers import sign_in
+from fastadmin.models.helpers import unregister_admin_model
+
+
+async def test_change(superuser, event, client):
+
+    @register(event.__class__)
+    class EventAdmin(TortoiseModelAdmin):
+        pass
+
+    await sign_in(client, superuser)
+    r = await client.patch(
+        f"/api/change/{event.__class__.__name__}/{event.id}",
+        json={
+            "name": "new name",
+            "participants": [superuser.id],
+        }
+    )
+    assert r.status_code == 200
+
+    event = await event.__class__.get(id=event.id)
+    item = r.json()
+    assert item["id"] == event.id
+    assert item["name"] == event.name
+    assert item["tournament_id"] == event.tournament_id
+    assert item["created_at"] == event.created_at.isoformat()
+    assert item["updated_at"] == event.updated_at.isoformat()
+    assert item["participants"] == [superuser.id]
+
+
+async def test_change_401(superuser, tournament, event, client):
+    r = await client.patch(
+        f"/api/change/{event.__class__.__name__}/{event.id}",
+        json={
+            "name": "new name",
+            "participants": [superuser.id],
+        }
+    )
+    assert r.status_code == 401
+
+
+async def test_change_404(superuser, event, client):
+    unregister_admin_model([event.__class__])
+    await sign_in(client, superuser)
+    r = await client.patch(
+        f"/api/change/{event.__class__.__name__}/{event.id}",
+        json={
+            "name": "new name",
+            "participants": [superuser.id],
+        }
+    )
+    assert r.status_code == 404

tests/api/tortoise/test_auth.py

@@ -0,0 +1,18 @@
+from fastadmin import TortoiseModelAdmin, register
+
+from tests.api.tortoise.helpers import sign_in
+
+
+async def test_configuration(superuser, event, client):
+
+    @register(event.__class__)
+    class EventAdmin(TortoiseModelAdmin):
+        pass
+
+    await sign_in(client, superuser)
+    r = await client.get(
+        f"/api/configuration",
+    )
+    assert r.status_code == 200
+    data = r.json()
+    assert data