paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-04 22:13:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_classes: Extract a thumbnailing output format helper.

Browse Source
This commit is contained in:
Alex Vandiver
2024-07-19 17:37:40 +00:00
committed by Tim Abbott
parent 94ff443c00
commit aacf28f7e3
2 changed files with 40 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
zerver/lib/test_classes.py
View File

@@ -74,6 +74,7 @@ from zerver.lib.test_helpers import (
instrument_url,
queries_captured,
)
from zerver.lib.thumbnail import ThumbnailFormat
from zerver.lib.topic import RESOLVED_TOPIC_PREFIX, filter_by_topic_name_via_message
from zerver.lib.user_groups import get_system_user_group_for_user
from zerver.lib.users import get_api_key
@@ -2018,6 +2019,14 @@ Output:
user_group.direct_subgroups.set(direct_subgroups)
return user_group
@contextmanager
def thumbnail_formats(self, *thumbnail_formats: ThumbnailFormat) -> Iterator[None]:
with (
mock.patch("zerver.lib.thumbnail.THUMBNAIL_OUTPUT_FORMATS", thumbnail_formats),
mock.patch("zerver.views.upload.THUMBNAIL_OUTPUT_FORMATS", thumbnail_formats),
):
yield
class ZulipTestCase(ZulipTestCaseMixin, TestCase):
@contextmanager