aurweb/test/test_package_request.py

import pytest
from sqlalchemy.exc import IntegrityError

from aurweb import db, time
from aurweb.models.account_type import USER_ID
from aurweb.models.package_base import PackageBase
from aurweb.models.package_request import (
    ACCEPTED,
    ACCEPTED_ID,
    CLOSED,
    CLOSED_ID,
    PENDING,
    PENDING_ID,
    REJECTED,
    REJECTED_ID,
    PackageRequest,
)
from aurweb.models.request_type import MERGE_ID
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="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)
    yield pkgbase


def test_package_request_creation(user: User, pkgbase: PackageBase):
    with db.begin():
        package_request = db.create(
            PackageRequest,
            ReqTypeID=MERGE_ID,
            User=user,
            PackageBase=pkgbase,
            PackageBaseName=pkgbase.Name,
            Comments=str(),
            ClosureComment=str(),
        )

    assert bool(package_request.ID)
    assert package_request.User == user
    assert package_request.PackageBase == pkgbase
    assert package_request.PackageBaseName == pkgbase.Name
    assert package_request.Comments == str()
    assert package_request.ClosureComment == str()

    # Make sure that everything is cross-referenced with relationships.
    assert package_request in user.package_requests
    assert package_request in pkgbase.requests


def test_package_request_closed(user: User, pkgbase: PackageBase):
    ts = time.utcnow()
    with db.begin():
        package_request = db.create(
            PackageRequest,
            ReqTypeID=MERGE_ID,
            User=user,
            PackageBase=pkgbase,
            PackageBaseName=pkgbase.Name,
            Closer=user,
            ClosedTS=ts,
            Comments=str(),
            ClosureComment=str(),
        )

    assert package_request.Closer == user
    assert package_request.ClosedTS == ts

    # Test relationships.
    assert package_request in user.closed_requests


def test_package_request_null_request_type_raises(user: User, pkgbase: PackageBase):
    with pytest.raises(IntegrityError):
        PackageRequest(
            User=user,
            PackageBase=pkgbase,
            PackageBaseName=pkgbase.Name,
            Comments=str(),
            ClosureComment=str(),
        )


def test_package_request_null_user_raises(pkgbase: PackageBase):
    with pytest.raises(IntegrityError):
        PackageRequest(
            ReqTypeID=MERGE_ID,
            PackageBase=pkgbase,
            PackageBaseName=pkgbase.Name,
            Comments=str(),
            ClosureComment=str(),
        )


def test_package_request_null_package_base_raises(user: User, pkgbase: PackageBase):
    with pytest.raises(IntegrityError):
        PackageRequest(
            ReqTypeID=MERGE_ID,
            User=user,
            PackageBaseName=pkgbase.Name,
            Comments=str(),
            ClosureComment=str(),
        )


def test_package_request_null_package_base_name_raises(
    user: User, pkgbase: PackageBase
):
    with pytest.raises(IntegrityError):
        PackageRequest(
            ReqTypeID=MERGE_ID,
            User=user,
            PackageBase=pkgbase,
            Comments=str(),
            ClosureComment=str(),
        )


def test_package_request_null_comments_raises(user: User, pkgbase: PackageBase):
    with pytest.raises(IntegrityError):
        PackageRequest(
            ReqTypeID=MERGE_ID,
            User=user,
            PackageBase=pkgbase,
            PackageBaseName=pkgbase.Name,
            ClosureComment=str(),
        )


def test_package_request_null_closure_comment_raises(user: User, pkgbase: PackageBase):
    with pytest.raises(IntegrityError):
        PackageRequest(
            ReqTypeID=MERGE_ID,
            User=user,
            PackageBase=pkgbase,
            PackageBaseName=pkgbase.Name,
            Comments=str(),
        )


def test_package_request_status_display(user: User, pkgbase: PackageBase):
    """Test status_display() based on the Status column value."""
    with db.begin():
        pkgreq = db.create(
            PackageRequest,
            ReqTypeID=MERGE_ID,
            User=user,
            PackageBase=pkgbase,
            PackageBaseName=pkgbase.Name,
            Comments=str(),
            ClosureComment=str(),
            Status=PENDING_ID,
        )
    assert pkgreq.status_display() == PENDING

    with db.begin():
        pkgreq.Status = CLOSED_ID
    assert pkgreq.status_display() == CLOSED

    with db.begin():
        pkgreq.Status = ACCEPTED_ID
    assert pkgreq.status_display() == ACCEPTED

    with db.begin():
        pkgreq.Status = REJECTED_ID
    assert pkgreq.status_display() == REJECTED

    with db.begin():
        pkgreq.Status = 124
    with pytest.raises(KeyError):
        pkgreq.status_display()