15760070102

Committed 19 Jun 2025 02:18PM UTC coverage: 97.286% (+0.07%) from 97.214%

Build # 15760070102

Build Type

Pull #1944

github

Committed by

web-flow

Commit Message

Merge 1e99a089b into 7565eefd3

Pull Request Pull Request #1944: Replace thread events with updates

Run Details

2828 of 2914 new or added lines in 89 files covered. (97.05%)

13 existing lines in 4 files now uncovered.

74124 of 76192 relevant lines covered (97.29%)

0.97 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

99.27

/misago/threads/tests/test_prefetch_posts_feed_related_objects.py

import pytest

from ...conf.test import override_dynamic_settings
from ...permissions.enums import CategoryPermission
from ...permissions.models import CategoryGroupPermission
from ...permissions.proxy import UserPermissionsProxy
from ..prefetch import (
    PrefetchPostsFeedRelatedObjects,
    prefetch_posts_feed_related_objects,
)


def apple(*args):
    pass


def orange(*args):
    pass


def tomato(*args):
    pass


def test_prefetch_posts_feed_related_objects_add_appends_operation(
    dynamic_settings, cache_versions, user
):
    permissions = UserPermissionsProxy(user, cache_versions)
    prefetch = PrefetchPostsFeedRelatedObjects(dynamic_settings, permissions, posts=[])

    prefetch.add(apple)
    assert prefetch.operations == [apple]


def test_prefetch_posts_feed_related_objects_add_before_inserts_operation(
    dynamic_settings, cache_versions, user
):
    permissions = UserPermissionsProxy(user, cache_versions)
    prefetch = PrefetchPostsFeedRelatedObjects(dynamic_settings, permissions, posts=[])

    prefetch.add(apple)
    prefetch.add(tomato)

    prefetch.add_before(tomato, orange)


def test_prefetch_posts_feed_related_objects_add_before_raises_for_not_found_item(
    dynamic_settings, cache_versions, user
):
    permissions = UserPermissionsProxy(user, cache_versions)
    prefetch = PrefetchPostsFeedRelatedObjects(dynamic_settings, permissions, posts=[])

    prefetch.add(apple)

    with pytest.raises(ValueError):
        prefetch.add_before(tomato, orange)


def test_prefetch_posts_feed_related_objects_add_after_inserts_operation(
    dynamic_settings, cache_versions, user
):
    permissions = UserPermissionsProxy(user, cache_versions)
    prefetch = PrefetchPostsFeedRelatedObjects(dynamic_settings, permissions, posts=[])

    prefetch.add(apple)
    prefetch.add(tomato)

    prefetch.add_after(apple, orange)
    assert prefetch.operations == [apple, orange, tomato]


def test_prefetch_posts_feed_related_objects_add_after_raises_for_not_found_item(
    dynamic_settings, cache_versions, user
):
    permissions = UserPermissionsProxy(user, cache_versions)
    prefetch = PrefetchPostsFeedRelatedObjects(dynamic_settings, permissions, posts=[])

    prefetch.add(apple)

    with pytest.raises(ValueError):
        prefetch.add_after(tomato, orange)


def test_prefetch_posts_feed_related_objects_supports_membership_test(
    dynamic_settings, cache_versions, user
):
    permissions = UserPermissionsProxy(user, cache_versions)
    prefetch = PrefetchPostsFeedRelatedObjects(dynamic_settings, permissions, posts=[])

    prefetch.add(apple)
    assert apple in prefetch
    assert orange not in prefetch


def test_prefetch_posts_feed_related_objects_preloads_categories(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    default_category,
    sibling_category,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(0):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings,
            permissions,
            [],
            categories=[default_category, sibling_category],
        )

        assert data["categories"] == {
            default_category.id: default_category,
            sibling_category.id: sibling_category,
        }


def test_prefetch_posts_feed_related_objects_prefetches_threads_categories(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    default_category,
    thread,
    user_thread,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(1):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [], threads=[thread, user_thread]
        )
        assert data["categories"] == {default_category.id: default_category}


def test_prefetch_posts_feed_related_objects_prefetches_posts_categories(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    default_category,
    post,
    user_reply,
    other_user_reply,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(5):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply, other_user_reply]
        )
        assert data["categories"] == {default_category.id: default_category}


def test_prefetch_posts_feed_related_objects_prefetches_thread_update_context_categories(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    guests_group,
    sibling_category,
    thread_update_category_context,
):
    CategoryGroupPermission.objects.create(
        category=sibling_category,
        group=guests_group,
        permission=CategoryPermission.SEE,
    )

    thread_update_category_context.context_id = sibling_category.id
    thread_update_category_context.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(3):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings,
            permissions,
            [],
            thread_updates=[thread_update_category_context],
        )
        assert data["categories"] == {sibling_category.id: sibling_category}


def test_prefetch_posts_feed_related_objects_prefetches_attachments_categories(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    default_category,
    text_attachment,
    user_text_attachment,
    post,
):
    text_attachment.associate_with_post(post)
    text_attachment.save()

    user_text_attachment.associate_with_post(post)
    user_text_attachment.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(3):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings,
            permissions,
            [],
            attachments=[text_attachment, user_text_attachment],
        )
        assert data["categories"] == {default_category.id: default_category}


def test_prefetch_posts_feed_related_objects_excludes_inaccessible_categories(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    default_category,
    sibling_category,
    thread,
    user_thread,
):
    user_thread.category = sibling_category
    user_thread.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(1):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [], threads=[thread, user_thread]
        )
        assert data["categories"] == {default_category.id: default_category}


def test_prefetch_posts_feed_related_objects_preloads_threads(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    thread,
    user_thread,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(1):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [], threads=[thread, user_thread]
        )
        assert data["threads"] == {
            thread.id: thread,
            user_thread.id: user_thread,
        }


def test_prefetch_posts_feed_related_objects_prefetches_posts_threads(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    thread,
    post,
    user_reply,
    other_user_reply,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(5):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply, other_user_reply]
        )
        assert data["threads"] == {thread.id: thread}


def test_prefetch_posts_feed_related_objects_prefetches_thread_update_context_threads(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    other_thread,
    thread_update_thread_context,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(4):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings,
            permissions,
            [],
            thread_updates=[thread_update_thread_context],
        )
        assert data["threads"] == {other_thread.id: other_thread}


def test_prefetch_posts_feed_related_objects_prefetches_attachments_threads(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    thread,
    post,
    text_attachment,
):
    text_attachment.associate_with_post(post)
    text_attachment.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(3):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [], attachments=[text_attachment]
        )
        assert data["threads"] == {thread.id: thread}


def test_prefetch_posts_feed_related_objects_excludes_inaccessible_threads(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    default_category,
    thread,
    post,
    hidden_thread,
):
    post.metadata["posts"] = [hidden_thread.first_post_id]

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(1):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [], threads=[thread]
        )
        assert data["categories"] == {default_category.id: default_category}


def test_prefetch_posts_feed_related_objects_preloads_posts(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    post,
    user_reply,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(5):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply]
        )
        assert data["posts"] == {
            post.id: post,
            user_reply.id: user_reply,
        }


def test_prefetch_posts_feed_related_objects_prefetches_quoted_posts(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    post,
    user_reply,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    post.metadata["posts"] = [user_reply.id]

    with django_assert_num_queries(6):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post]
        )
        assert data["posts"] == {
            post.id: post,
            user_reply.id: user_reply,
        }


def test_prefetch_posts_feed_related_objects_prefetches_quoted_posts_for_unsaved_post(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    post,
    user_reply,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    post.id = None
    post.metadata["posts"] = [user_reply.id]

    with django_assert_num_queries(5):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post]
        )
        assert data["posts"] == {
            user_reply.id: user_reply,
        }


def test_prefetch_posts_feed_related_objects_excludes_inaccessible_quoted_posts(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    user,
    post,
    user_reply,
    private_thread,
    user_private_thread,
):
    permissions = UserPermissionsProxy(user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    post.metadata["posts"] = [
        private_thread.first_post_id,
        user_private_thread.first_post_id,
    ]
    post.save()

    with django_assert_num_queries(6):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply]
        )
        assert data["posts"] == {
            post.id: post,
            user_reply.id: user_reply,
            user_private_thread.first_post_id: user_private_thread.first_post,
        }


def test_prefetch_posts_feed_related_objects_fetches_posts_metadata_posts(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    post,
    user_reply,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    post.metadata["posts"] = [user_reply.id]
    post.save()

    with django_assert_num_queries(6):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post]
        )
        assert data["posts"] == {
            post.id: post,
            user_reply.id: user_reply,
        }


def test_prefetch_posts_feed_related_objects_prefetches_attachments_posts(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    post,
    user_reply,
    text_attachment,
    user_text_attachment,
):
    text_attachment.associate_with_post(post)
    text_attachment.save()

    user_text_attachment.associate_with_post(user_reply)
    user_text_attachment.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(5):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings,
            permissions,
            [],
            attachments=[text_attachment, user_text_attachment],
        )
        assert data["posts"] == {
            post.id: post,
            user_reply.id: user_reply,
        }


def test_prefetch_posts_feed_related_objects_preloads_attachments(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    text_attachment,
    user_text_attachment,
    post,
):
    text_attachment.associate_with_post(post)
    text_attachment.save()

    user_text_attachment.associate_with_post(post)
    user_text_attachment.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(3):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings,
            permissions,
            [],
            attachments=[text_attachment, user_text_attachment],
        )
        assert data["attachments"] == {
            text_attachment.id: text_attachment,
            user_text_attachment.id: user_text_attachment,
        }


def test_prefetch_posts_feed_related_objects_removes_preloaded_attachments_without_permission(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    user,
    text_attachment,
    user_text_attachment,
):
    permissions = UserPermissionsProxy(user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(1):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings,
            permissions,
            [],
            attachments=[text_attachment, user_text_attachment],
        )
        assert data["attachments"] == {
            user_text_attachment.id: user_text_attachment,
        }


def test_prefetch_posts_feed_related_objects_fetches_posts_attachments(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    text_attachment,
    user_text_attachment,
    post,
    user_reply,
):
    text_attachment.associate_with_post(post)
    text_attachment.save()

    user_text_attachment.associate_with_post(user_reply)
    user_text_attachment.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(5):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply]
        )
        assert data["attachments"] == {
            text_attachment.id: text_attachment,
            user_text_attachment.id: user_text_attachment,
        }


def test_prefetch_posts_feed_related_objects_fetches_posts_metadata_attachments(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    text_attachment,
    user_text_attachment,
    post,
    user_reply,
    other_user_reply,
):
    text_attachment.associate_with_post(other_user_reply)
    text_attachment.save()

    user_text_attachment.associate_with_post(other_user_reply)
    user_text_attachment.save()

    post.metadata["attachments"] = [text_attachment.id]
    post.save()

    user_reply.metadata["attachments"] = [user_text_attachment.id]
    user_reply.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(6):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply]
        )
        assert data["attachments"] == {
            text_attachment.id: text_attachment,
            user_text_attachment.id: user_text_attachment,
        }


@override_dynamic_settings(additional_embedded_attachments_limit=0)
def test_prefetch_posts_feed_related_objects_doesnt_fetch_posts_metadata_attachments_if_its_disabled(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    text_attachment,
    user_text_attachment,
    post,
    user_reply,
    other_user_reply,
):
    text_attachment.associate_with_post(other_user_reply)
    text_attachment.save()

    user_text_attachment.associate_with_post(other_user_reply)
    user_text_attachment.save()

    post.metadata["attachments"] = [text_attachment.id]
    post.save()

    user_reply.metadata["attachments"] = [user_text_attachment.id]
    user_reply.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(5):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply]
        )
        assert data["attachments"] == {}


@override_dynamic_settings(additional_embedded_attachments_limit=1)
def test_prefetch_posts_feed_related_objects_fetch_posts_metadata_attachments_is_limited(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    text_attachment,
    user_text_attachment,
    post,
    user_reply,
    other_user_reply,
):
    text_attachment.associate_with_post(other_user_reply)
    text_attachment.save()

    user_text_attachment.associate_with_post(other_user_reply)
    user_text_attachment.save()

    post.metadata["attachments"] = [text_attachment.id]
    post.save()

    user_reply.metadata["attachments"] = [user_text_attachment.id]
    user_reply.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(6):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply]
        )
        assert data["attachments"] == {user_text_attachment.id: user_text_attachment}


def test_prefetch_posts_feed_related_objects_fetch_posts_metadata_attachments_excludes_unused_attachments(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    user,
    text_attachment,
    user_text_attachment,
    post,
    user_reply,
    other_user_reply,
):
    text_attachment.associate_with_post(other_user_reply)
    text_attachment.save()

    post.metadata["attachments"] = [text_attachment.id]
    post.save()

    user_reply.metadata["attachments"] = [user_text_attachment.id]
    user_reply.save()

    permissions = UserPermissionsProxy(user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(6):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply]
        )
        assert data["attachments"] == {
            text_attachment.id: text_attachment,
        }


def test_prefetch_posts_feed_related_objects_fetch_posts_metadata_attachments_excludes_inaccessible_attachments(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    text_attachment,
    user_text_attachment,
    post,
    user_reply,
    other_user_reply,
    private_thread,
):
    text_attachment.associate_with_post(other_user_reply)
    text_attachment.save()

    user_text_attachment.associate_with_post(private_thread.first_post)
    user_text_attachment.save()

    post.metadata["attachments"] = [text_attachment.id]
    post.save()

    user_reply.metadata["attachments"] = [user_text_attachment.id]
    user_reply.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(7):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply]
        )
        assert data["attachments"] == {
            text_attachment.id: text_attachment,
        }


def test_prefetch_posts_feed_related_objects_fetch_posts_metadata_attachments_excludes_attachments_with_permission_denied(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    guests_group,
    text_attachment,
    user_text_attachment,
    sibling_category,
    post,
    user_reply,
    other_thread,
):
    CategoryGroupPermission.objects.create(
        category=sibling_category,
        group=guests_group,
        permission=CategoryPermission.SEE,
    )
    CategoryGroupPermission.objects.create(
        category=sibling_category,
        group=guests_group,
        permission=CategoryPermission.BROWSE,
    )

    other_thread.category = sibling_category
    other_thread.save()

    other_thread.first_post.category = sibling_category
    other_thread.first_post.save()

    text_attachment.associate_with_post(post)
    text_attachment.save()

    user_text_attachment.associate_with_post(other_thread.first_post)
    user_text_attachment.save()

    post.metadata["attachments"] = [text_attachment.id]
    post.save()

    user_reply.metadata["attachments"] = [user_text_attachment.id]
    user_reply.save()

    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(6):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post, user_reply]
        )
        assert data["attachments"] == {
            text_attachment.id: text_attachment,
        }
        assert tuple(data["attachment_errors"]) == (user_text_attachment.id,)
        assert data["attachment_errors"][user_text_attachment.id].permission_denied


def test_prefetch_posts_feed_related_objects_prefetches_private_threads_members(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    user_private_thread,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(2):
        prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [], threads=[user_private_thread]
        )

    with django_assert_num_queries(0):
        user_private_thread.private_thread_member_ids


def test_prefetch_posts_feed_related_objects_preloads_users(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    user,
    other_user,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(1):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [], users=[user, other_user]
        )
        assert data["users"] == {
            user.id: user,
            other_user.id: other_user,
        }


def test_prefetch_posts_feed_related_objects_preloads_user_from_permissions(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    user,
):
    permissions = UserPermissionsProxy(user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(1):
        data = prefetch_posts_feed_related_objects(dynamic_settings, permissions, [])
        assert data["users"] == {user.id: user}

    with django_assert_num_queries(0):
        assert data["users"][user.id].group


def test_prefetch_posts_feed_related_objects_doesnt_preload_anonymous_user_from_permissions(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    post,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(3):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post]
        )
        assert data["users"] == {}


def test_prefetch_posts_feed_related_objects_prefetches_posts_users(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    user,
    other_user,
    user_reply,
    other_user_reply,
):
    permissions = UserPermissionsProxy(user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(5):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [user_reply, other_user_reply]
        )
        assert data["users"] == {user.id: user, other_user.id: other_user}

    with django_assert_num_queries(0):
        assert data["users"][user.id].group
        assert data["users"][other_user.id].group


def test_prefetch_posts_feed_related_objects_prefetches_thread_update_users(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    anonymous_user,
    user,
    thread_update,
):
    permissions = UserPermissionsProxy(anonymous_user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(2):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [], thread_updates=[thread_update]
        )
        assert data["users"] == {user.id: user}

    with django_assert_num_queries(0):
        assert data["users"][user.id].group


def test_prefetch_posts_feed_related_objects_prefetches_thread_update_context_users(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    user,
    other_user,
    thread_update_user_context,
):
    permissions = UserPermissionsProxy(user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(2):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings,
            permissions,
            [],
            thread_updates=[thread_update_user_context],
        )
        assert data["users"] == {user.id: user, other_user.id: other_user}

    with django_assert_num_queries(0):
        assert data["users"][user.id].group
        assert data["users"][other_user.id].group


def test_prefetch_posts_feed_related_objects_doesnt_prefetch_anonymous_posts_users(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    user,
    post,
):
    permissions = UserPermissionsProxy(user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    with django_assert_num_queries(4):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings, permissions, [post]
        )
        assert data["users"] == {user.id: user}

    with django_assert_num_queries(0):
        assert data["users"][user.id].group


def test_prefetch_posts_feed_related_objects_prefetches_objects_in_cascade(
    django_assert_num_queries,
    dynamic_settings,
    cache_versions,
    user,
    other_user,
    default_category,
    private_threads_category,
    thread,
    post,
    user_reply,
    other_user_reply,
    user_private_thread,
    text_attachment,
    user_text_attachment,
    other_user_text_attachment,
):
    permissions = UserPermissionsProxy(user, cache_versions)
    permissions.permissions
    permissions.is_global_moderator

    text_attachment.associate_with_post(post)
    text_attachment.save()

    other_user_text_attachment.associate_with_post(user_private_thread.first_post)
    other_user_text_attachment.save()

    user_reply.metadata["attachments"] = [
        user_text_attachment.id,
        other_user_text_attachment.id,
    ]
    user_reply.save()

    with django_assert_num_queries(7):
        data = prefetch_posts_feed_related_objects(
            dynamic_settings,
            permissions,
            [post, user_reply, other_user_reply],
        )

        assert default_category.id in data["categories"]
        assert private_threads_category.id in data["categories"]

        assert thread.id in data["threads"]
        assert user_private_thread.id in data["threads"]

        assert data["threads"][user_private_thread.id].private_thread_member_ids

        assert post.id in data["posts"]
        assert user_reply.id in data["posts"]
        assert other_user_reply.id in data["posts"]
        assert user_private_thread.first_post_id in data["posts"]

        assert user.id in data["users"]
        assert other_user.id in data["users"]

        assert data["users"][user.id].group
        assert data["users"][other_user.id].group

        assert text_attachment.id in data["attachments"]
        assert other_user_text_attachment.id in data["attachments"]

rafalp / Misago / 15760070102

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous