housekeep(fastapi): rewrite test_package_comaintainer with fixtures

Signed-off-by: Kevin Morris <kevr@0cost.org>
This commit is contained in:
Kevin Morris 2021-11-30 18:18:48 -08:00
parent ff3931e435
commit 14d80d756f
No known key found for this signature in database
GPG key ID: F7E46DED420788F3

View file

@ -3,24 +3,34 @@ import pytest
from sqlalchemy.exc import IntegrityError from sqlalchemy.exc import IntegrityError
from aurweb import db from aurweb import db
from aurweb.models.account_type import USER_ID
from aurweb.models.package_base import PackageBase from aurweb.models.package_base import PackageBase
from aurweb.models.package_comaintainer import PackageComaintainer from aurweb.models.package_comaintainer import PackageComaintainer
from aurweb.models.user import User from aurweb.models.user import User
user = pkgbase = None
@pytest.fixture(autouse=True) @pytest.fixture(autouse=True)
def setup(db_test): def setup(db_test):
global user, pkgbase return
@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",
AccountTypeID=USER_ID)
yield user
@pytest.fixture
def pkgbase(user: User) -> PackageBase:
with db.begin():
pkgbase = db.create(PackageBase, Name="test-package", Maintainer=user) pkgbase = db.create(PackageBase, Name="test-package", Maintainer=user)
yield pkgbase
def test_package_comaintainer_creation(): def test_package_comaintainer_creation(user: User, pkgbase: PackageBase):
with db.begin(): with db.begin():
package_comaintainer = db.create(PackageComaintainer, User=user, package_comaintainer = db.create(PackageComaintainer, User=user,
PackageBase=pkgbase, Priority=5) PackageBase=pkgbase, Priority=5)
@ -30,16 +40,17 @@ def test_package_comaintainer_creation():
assert package_comaintainer.Priority == 5 assert package_comaintainer.Priority == 5
def test_package_comaintainer_null_user_raises_exception(): def test_package_comaintainer_null_user_raises(pkgbase: PackageBase):
with pytest.raises(IntegrityError): with pytest.raises(IntegrityError):
PackageComaintainer(PackageBase=pkgbase, Priority=1) PackageComaintainer(PackageBase=pkgbase, Priority=1)
def test_package_comaintainer_null_pkgbase_raises_exception(): def test_package_comaintainer_null_pkgbase_raises(user: User):
with pytest.raises(IntegrityError): with pytest.raises(IntegrityError):
PackageComaintainer(User=user, Priority=1) PackageComaintainer(User=user, Priority=1)
def test_package_comaintainer_null_priority_raises_exception(): def test_package_comaintainer_null_priority_raises(user: User,
pkgbase: PackageBase):
with pytest.raises(IntegrityError): with pytest.raises(IntegrityError):
PackageComaintainer(User=user, PackageBase=pkgbase) PackageComaintainer(User=user, PackageBase=pkgbase)