55 lines
2.3 KiB
Python
55 lines
2.3 KiB
Python
import pytest
|
|
|
|
from amanuensis.db import DbContext
|
|
from amanuensis.db.models import Character, Lexicon, User
|
|
import amanuensis.backend.article as artiq
|
|
|
|
from amanuensis.errors import ArgumentError
|
|
|
|
|
|
def test_create_article(db: DbContext, make):
|
|
"""Test new article creation"""
|
|
# Create two users in a shared lexicon
|
|
user1: User = make.user()
|
|
user2: User = make.user()
|
|
lexicon1: Lexicon = make.lexicon()
|
|
make.membership(user_id=user1.id, lexicon_id=lexicon1.id)
|
|
make.membership(user_id=user2.id, lexicon_id=lexicon1.id)
|
|
char1_1: Character = make.character(lexicon_id=lexicon1.id, user_id=user1.id)
|
|
char1_2: Character = make.character(lexicon_id=lexicon1.id, user_id=user2.id)
|
|
|
|
# Create a lexicon that only one user is in
|
|
lexicon2: Lexicon = make.lexicon()
|
|
make.membership(user_id=user2.id, lexicon_id=lexicon2.id)
|
|
char2_2: Character = make.character(lexicon_id=lexicon2.id, user_id=user2.id)
|
|
|
|
# User cannot create article for another user's character
|
|
with pytest.raises(ArgumentError):
|
|
artiq.create(db, lexicon1.id, user1.id, char1_2.id)
|
|
with pytest.raises(ArgumentError):
|
|
artiq.create(db, lexicon1.id, user2.id, char1_1.id)
|
|
|
|
# User cannot create article for their character in the wrong lexicon
|
|
with pytest.raises(ArgumentError):
|
|
artiq.create(db, lexicon1.id, user2.id, char2_2.id)
|
|
with pytest.raises(ArgumentError):
|
|
artiq.create(db, lexicon2.id, user2.id, char1_2.id)
|
|
|
|
# User cannot create article in a lexicon they aren't in
|
|
with pytest.raises(ArgumentError):
|
|
artiq.create(db, lexicon2.id, user1.id, char1_1.id)
|
|
|
|
# User cannot create anonymous articles in a lexicon they aren't in
|
|
with pytest.raises(ArgumentError):
|
|
artiq.create(db, lexicon2.id, user1.id, character_id=None)
|
|
|
|
# Users can create character-owned articles
|
|
assert artiq.create(db, lexicon1.id, user1.id, char1_1.id)
|
|
assert artiq.create(db, lexicon1.id, user2.id, char1_2.id)
|
|
assert artiq.create(db, lexicon2.id, user2.id, char2_2.id)
|
|
|
|
# Users can create anonymous articles
|
|
assert artiq.create(db, lexicon1.id, user1.id, character_id=None)
|
|
assert artiq.create(db, lexicon1.id, user2.id, character_id=None)
|
|
assert artiq.create(db, lexicon2.id, user2.id, character_id=None)
|