external/aurweb
0
0
Fork 0
mirror of https://gitlab.archlinux.org/archlinux/aurweb.git synced 2025-02-03 10:43:03 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

housekeep(fastapi): rewrite test_package with fixtures

Browse source 
Signed-off-by: Kevin Morris <kevr@0cost.org>
This commit is contained in:
Kevin Morris 2021-11-30 21:27:02 -08:00
parent 5b14ad4065
commit eb396813a8
No known key found for this signature in database
GPG key ID: F7E46DED420788F3
1 changed files with 25 additions and 29 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
test/test_package.py
View file

@ -4,7 +4,7 @@ from sqlalchemy import and_
from sqlalchemy.exc import IntegrityError from sqlalchemy.exc import IntegrityError
from aurweb import db from aurweb import db
from aurweb.models.account_type import AccountType from aurweb.models.account_type import USER_ID
from aurweb.models.package import Package from aurweb.models.package import Package
from aurweb.models.package_base import PackageBase from aurweb.models.package_base import PackageBase
from aurweb.models.user import User from aurweb.models.user import User
@ -14,28 +14,30 @@ user = pkgbase = package = None
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
def setup(db_test): def setup(db_test):
global user, pkgbase, package return
account_type = db.query(AccountType,
AccountType.AccountType == "User").first()
@pytest.fixture
def user() -> User:
with db.begin(): with db.begin():
user = db.create(User, Username="test", Email="test@example.org", user = db.create(User, Username="test", Email="test@example.org",
RealName="Test User", Passwd="testPassword", RealName="Test User", Passwd="testPassword",
AccountType=account_type) AccountTypeID=USER_ID)
yield user
pkgbase = db.create(PackageBase,
Name="beautiful-package", @pytest.fixture
def package(user: User) -> Package:
with db.begin():
pkgbase = db.create(PackageBase, Name="beautiful-package",
Maintainer=user) Maintainer=user)
package = db.create(Package, package = db.create(Package, PackageBase=pkgbase, Name=pkgbase.Name,
PackageBase=pkgbase,
Name=pkgbase.Name,
Description="Test description.", Description="Test description.",
URL="https://test.package") URL="https://test.package")
yield package
def test_package(): def test_package(package: Package):
assert pkgbase == package.PackageBase
assert package.Name == "beautiful-package" assert package.Name == "beautiful-package"
assert package.Description == "Test description." assert package.Description == "Test description."
assert package.Version == str() # Default version. assert package.Version == str() # Default version.
@ -46,27 +48,21 @@ def test_package():
package.Version = "1.2.3" package.Version = "1.2.3"
# Make sure it got updated in the database. # Make sure it got updated in the database.
record = db.query(Package, record = db.query(Package).filter(
and_(Package.ID == package.ID, and_(Package.ID == package.ID,
Package.Version == "1.2.3")).first() Package.Version == "1.2.3")
).first()
assert record is not None assert record is not None
def test_package_null_pkgbase_raises_exception(): def test_package_null_pkgbase_raises():
with pytest.raises(IntegrityError): with pytest.raises(IntegrityError):
with db.begin(): Package(Name="some-package", Description="Some description.",
db.create(Package, URL="https://some.package")
Name="some-package",
Description="Some description.",
URL="https://some.package")
db.rollback()
def test_package_null_name_raises_exception(): def test_package_null_name_raises(package: Package):
pkgbase = package.PackageBase
with pytest.raises(IntegrityError): with pytest.raises(IntegrityError):
with db.begin(): Package(PackageBase=pkgbase, Description="Some description.",
db.create(Package, URL="https://some.package")
PackageBase=pkgbase,
Description="Some description.",
URL="https://some.package")
db.rollback()