paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-10-23 04:52:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

upload: Start storing content-type for new uploads.

Browse Source
This commit is contained in:
Alex Vandiver
2024-06-20 16:54:49 +00:00
committed by Tim Abbott
parent 45afdc2d35
commit 0a296b2a6e
4 changed files with 27 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
analytics/tests/test_counts.py
View File

@@ -204,7 +204,12 @@ class AnalyticsTestCase(ZulipTestCase):
return Message.objects.create(**kwargs)
def create_attachment(
self, user_profile: UserProfile, filename: str, size: int, create_time: datetime
self,
user_profile: UserProfile,
filename: str,
size: int,
create_time: datetime,
content_type: str,
) -> Attachment:
return Attachment.objects.create(
file_name=filename,
@@ -213,6 +218,7 @@ class AnalyticsTestCase(ZulipTestCase):
realm=user_profile.realm,
size=size,
create_time=create_time,
content_type=content_type,
)
# kwargs should only ever be a UserProfile or Stream.
@@ -551,9 +557,9 @@ class TestCountStats(AnalyticsTestCase):
user2 = self.create_user()
user_second_realm = self.create_user(realm=self.second_realm)
self.create_attachment(user1, "file1", 100, self.TIME_LAST_HOUR)
attachment2 = self.create_attachment(user2, "file2", 200, self.TIME_LAST_HOUR)
self.create_attachment(user_second_realm, "file3", 10, self.TIME_LAST_HOUR)
self.create_attachment(user1, "file1", 100, self.TIME_LAST_HOUR, "text/plain")
attachment2 = self.create_attachment(user2, "file2", 200, self.TIME_LAST_HOUR, "text/plain")
self.create_attachment(user_second_realm, "file3", 10, self.TIME_LAST_HOUR, "text/plain")
do_fill_count_stat_at_hour(stat, self.TIME_ZERO)