import pytest
from django.contrib.auth.models import AnonymousUser
from django.db.models import Q

from funkwhale_api.common import fields
from funkwhale_api.users.factories import UserFactory


@pytest.mark.parametrize(
    "user,expected",
    [
        (AnonymousUser(), Q(privacy_level="everyone")),
        (
            UserFactory.build(pk=1),
            Q(privacy_level__in=["instance", "everyone"])
            | Q(privacy_level="me", user=UserFactory.build(pk=1)),
        ),
    ],
)
def test_privacy_level_query(user, expected):
    query = fields.privacy_level_query(user)
    assert query == expected