aurweb/test/test_accepted_term.py
Kevin Morris a36cc0c00a eradicate Term records after testing them
Otherwise, Terms can leak out into other tests causing /tos
redirects unexpectedly.

Signed-off-by: Kevin Morris <kevr@0cost.org>
2021-06-24 19:52:17 -07:00

55 lines
1.6 KiB
Python

import pytest
from sqlalchemy.exc import IntegrityError
from aurweb.db import create, query
from aurweb.models.accepted_term import AcceptedTerm
from aurweb.models.account_type import AccountType
from aurweb.models.term import Term
from aurweb.models.user import User
from aurweb.testing import setup_test_db
user = term = accepted_term = None
@pytest.fixture(autouse=True)
def setup():
global user, term, accepted_term
setup_test_db("Users", "AcceptedTerms", "Terms")
account_type = query(AccountType,
AccountType.AccountType == "User").first()
user = create(User, Username="test", Email="test@example.org",
RealName="Test User", Passwd="testPassword",
AccountType=account_type)
term = create(Term, Description="Test term", URL="https://test.term")
yield term
# Eradicate any terms we created.
setup_test_db("AcceptedTerms", "Terms")
def test_accepted_term():
accepted_term = create(AcceptedTerm, User=user, Term=term)
# Make sure our AcceptedTerm relationships got initialized properly.
assert accepted_term.User == user
assert accepted_term in user.accepted_terms
assert accepted_term in term.accepted_terms
def test_accepted_term_null_user_raises_exception():
from aurweb.db import session
with pytest.raises(IntegrityError):
create(AcceptedTerm, Term=term)
session.rollback()
def test_accepted_term_null_term_raises_exception():
from aurweb.db import session
with pytest.raises(IntegrityError):
create(AcceptedTerm, User=user)
session.rollback()