amanuensis/tests/backend/test_article.py

import pytest
import time

from amanuensis.backend import artiq
from amanuensis.db import DbContext, Character, Lexicon, User

from amanuensis.errors import ArgumentError
from tests.conftest import ObjectFactory


def test_create_article(db: DbContext, make: ObjectFactory):
    """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)


def test_article_update_ts(db: DbContext, make: ObjectFactory):
    """Test the update timestamp."""
    user: User = make.user()
    lexicon: Lexicon = make.lexicon()
    make.membership(user_id=user.id, lexicon_id=lexicon.id)
    char: Character = make.character(lexicon_id=lexicon.id, user_id=user.id)
    article = artiq.create(db, lexicon.id, user.id, char.id)
    created = article.last_updated
    time.sleep(1)  # The update timestamp has only second-level precision
    article.title = "New title, who dis"
    db.session.commit()
    assert created != article.last_updated