funkwhale/api/tests/federation/test_models.py

import pytest
from django import db


def test_cannot_duplicate_actor(factories):
    actor = factories["federation.Actor"]()

    with pytest.raises(db.IntegrityError):
        factories["federation.Actor"](
            domain=actor.domain, preferred_username=actor.preferred_username
        )


def test_cannot_duplicate_follow(factories):
    follow = factories["federation.Follow"]()

    with pytest.raises(db.IntegrityError):
        factories["federation.Follow"](target=follow.target, actor=follow.actor)


def test_follow_federation_url(factories):
    follow = factories["federation.Follow"](local=True)
    expected = "{}#follows/{}".format(follow.actor.url, follow.uuid)

    assert follow.get_federation_url() == expected


def test_library_model_unique_per_actor(factories):
    library = factories["federation.Library"]()
    with pytest.raises(db.IntegrityError):
        factories["federation.Library"](actor=library.actor)