mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
aurweb.db: add query, create, delete helpers
Takes sqlalchemy kwargs or stanzas: query(Model, Model.Column == value) query(Model, and_(Model.Column == value, Model.Column != "BAD!")) Updated tests to reflect the new utility and a comment about upcoming function deprecation is added to get_account_type(). From here on, phase out the use of get_account_type(). + aurweb.db: Added create utility function + aurweb.db: Added delete utility function The `delete` function can be used to delete a record by search kwargs directly. Example: delete(User, User.ID == 6) All three functions added in this commit are typically useful to perform these operations without having to import aurweb.db.session. Removes a bit of redundancy overall. Signed-off-by: Kevin Morris <kevr@0cost.org>
This commit is contained in:
parent
5185df629e
commit
a836892cde
5 changed files with 84 additions and 9 deletions
|
@ -7,14 +7,26 @@ import pytest
|
|||
from fastapi.testclient import TestClient
|
||||
|
||||
from aurweb.asgi import app
|
||||
from aurweb.db import query
|
||||
from aurweb.models.account_type import AccountType
|
||||
from aurweb.testing import setup_test_db
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
user = None
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def setup():
|
||||
setup_test_db("Users", "Session")
|
||||
global user
|
||||
|
||||
setup_test_db("Users", "Sessions")
|
||||
|
||||
account_type = query(AccountType,
|
||||
AccountType.AccountType == "User").first()
|
||||
user = make_user(Username="test", Email="test@example.org",
|
||||
RealName="Test User", Passwd="testPassword",
|
||||
AccountType=account_type)
|
||||
|
||||
|
||||
def test_index():
|
||||
|
@ -54,6 +66,7 @@ def test_language_invalid_next():
|
|||
response = req.post("/language", data=post_data)
|
||||
assert response.status_code == int(HTTPStatus.BAD_REQUEST)
|
||||
|
||||
|
||||
def test_language_query_params():
|
||||
""" Test the language post route with query params. """
|
||||
next = urllib.parse.quote_plus("/")
|
||||
|
@ -73,4 +86,5 @@ def test_error_messages():
|
|||
response1 = client.get("/thisroutedoesnotexist")
|
||||
response2 = client.get("/raisefivethree")
|
||||
assert response1.status_code == int(HTTPStatus.NOT_FOUND)
|
||||
assert response2.status_code == int(HTTPStatus.SERVICE_UNAVAILABLE)
|
||||
assert response2.status_code == int(HTTPStatus.SERVICE_UNAVAILABLE)
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue