mirror of
https://gitlab.archlinux.org/archlinux/aurweb.git
synced 2025-02-03 10:43:03 +01:00
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
from datetime import datetime
|
|
|
|
import pytest
|
|
|
|
from sqlalchemy.exc import IntegrityError
|
|
|
|
from aurweb import db
|
|
from aurweb.models.account_type import USER_ID
|
|
from aurweb.models.package_base import PackageBase
|
|
from aurweb.models.package_vote import PackageVote
|
|
from aurweb.models.user import User
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def setup(db_test):
|
|
return
|
|
|
|
|
|
@pytest.fixture
|
|
def user() -> User:
|
|
with db.begin():
|
|
user = db.create(User, Username="test", Email="test@example.org",
|
|
RealName="Test User", Passwd=str(),
|
|
AccountTypeID=USER_ID)
|
|
yield user
|
|
|
|
|
|
@pytest.fixture
|
|
def pkgbase(user: User) -> PackageBase:
|
|
with db.begin():
|
|
pkgbase = db.create(PackageBase, Name="test-package", Maintainer=user)
|
|
yield pkgbase
|
|
|
|
|
|
def test_package_vote_creation(user: User, pkgbase: PackageBase):
|
|
ts = int(datetime.utcnow().timestamp())
|
|
|
|
with db.begin():
|
|
package_vote = db.create(PackageVote, User=user,
|
|
PackageBase=pkgbase, VoteTS=ts)
|
|
assert bool(package_vote)
|
|
assert package_vote.User == user
|
|
assert package_vote.PackageBase == pkgbase
|
|
assert package_vote.VoteTS == ts
|
|
|
|
|
|
def test_package_vote_null_user_raises(pkgbase: PackageBase):
|
|
with pytest.raises(IntegrityError):
|
|
PackageVote(PackageBase=pkgbase, VoteTS=1)
|
|
|
|
|
|
def test_package_vote_null_pkgbase_raises(user: User):
|
|
with pytest.raises(IntegrityError):
|
|
PackageVote(User=user, VoteTS=1)
|
|
|
|
|
|
def test_package_vote_null_votets_raises(user: User, pkgbase: PackageBase):
|
|
with pytest.raises(IntegrityError):
|
|
PackageVote(User=user, PackageBase=pkgbase)
|