mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
aecb649473
First off: This commit changes the default development database backend to mysql. sqlite, however, is still completely supported with the caveat that a user must now modify config.dev to use the sqlite backend. While looking into this, it was discovered that our SQLAlchemy backend for mysql (mysql-connector) completely broke model attributes when we switched to utf8mb4_bin (binary) -- it does not correct the correct conversion to and from binary utf8mb4. The new, replacement dependency mysqlclient does. mysqlclient is also recommended in SQLAlchemy documentation as the "best" one available. The mysqlclient backend uses a different exception flow then sqlite, and so tests expecting IntegrityError has to be modified to expect OperationalError from sqlalchemy.exc. So, for each model that we define, check keys that can't be NULL and raise sqlalchemy.exc.IntegrityError if we have to. This way we keep our exceptions uniform. Signed-off-by: Kevin Morris <kevr@0cost.org>
38 lines
1 KiB
Python
38 lines
1 KiB
Python
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_ip_default():
|
|
api_rate_limit = create(ApiRateLimit, Requests=10, WindowStart=1)
|
|
assert api_rate_limit.IP == str()
|
|
|
|
|
|
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", Requests=1)
|
|
session.rollback()
|