Commit a65a6060 authored by Kevin Morris's avatar Kevin Morris
Browse files

add ApiRateLimit SQLAlchemy ORM model



Signed-off-by: Kevin Morris's avatarKevin Morris <kevr@0cost.org>
parent 2b83d2fb
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])
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()
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment