add ApiRateLimit SQLAlchemy ORM model

Signed-off-by: Kevin Morris <kevr@0cost.org>

aurweb/models/api_rate_limit.py

@@ -0,0 +1,15 @@
+from sqlalchemy.orm import mapper
+
+from aurweb.schema import ApiRateLimit as _ApiRateLimit
+
+
+class ApiRateLimit:
+    def __init__(self, IP: str = None,
+                 Requests: int = None,
+                 WindowStart: int = None):
+        self.IP = IP
+        self.Requests = Requests
+        self.WindowStart = WindowStart
+
+
+mapper(ApiRateLimit, _ApiRateLimit, primary_key=[_ApiRateLimit.c.IP])

test/test_api_rate_limit.py

@@ -0,0 +1,40 @@
+import pytest
+
+from sqlalchemy.exc import IntegrityError
+
+from aurweb.db import create
+from aurweb.models.api_rate_limit import ApiRateLimit
+from aurweb.testing import setup_test_db
+
+
+@pytest.fixture(autouse=True)
+def setup():
+    setup_test_db("ApiRateLimit")
+
+
+def test_api_rate_key_creation():
+    rate = create(ApiRateLimit, IP="127.0.0.1", Requests=10, WindowStart=1)
+    assert rate.IP == "127.0.0.1"
+    assert rate.Requests == 10
+    assert rate.WindowStart == 1
+
+
+def test_api_rate_key_null_ip_raises_exception():
+    from aurweb.db import session
+    with pytest.raises(IntegrityError):
+        create(ApiRateLimit, Requests=10, WindowStart=1)
+    session.rollback()
+
+
+def test_api_rate_key_null_requests_raises_exception():
+    from aurweb.db import session
+    with pytest.raises(IntegrityError):
+        create(ApiRateLimit, IP="127.0.0.1", WindowStart=1)
+    session.rollback()
+
+
+def test_api_rate_key_null_window_start_raises_exception():
+    from aurweb.db import session
+    with pytest.raises(IntegrityError):
+        create(ApiRateLimit, IP="127.0.0.1", WindowStart=1)
+    session.rollback()