CodeCow/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2026-07-12 21:04:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

user_topics: Update tests to use the 'time_machine' module.

Browse Source 
This commit updates the tests to use the 'time_machine'
module for mocking datetime.
This commit is contained in:
Prakhar Pratyush 2023-04-03 04:46:10 +05:30 committed by Tim Abbott
parent 5c86d3ca9e
commit e33eb45507
1 changed files with 4 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
zerver/tests/test_user_topics.py
View File

@ -1,7 +1,7 @@
from datetime import datetime, timezone
from typing import Any, Dict, List
from unittest import mock
import time_machine
from django.utils.timezone import now as timezone_now
from zerver.actions.user_topics import do_set_user_topic_visibility_policy
@ -25,10 +25,7 @@ class MutedTopicsTests(ZulipTestCase):
url = "/api/v1/users/me/subscriptions/muted_topics"
data = {"stream_id": stream.id, "topic": "Verona3", "op": "add"}
with mock.patch(
"zerver.views.user_topics.timezone_now",
return_value=datetime(2020, 1, 1, tzinfo=timezone.utc),
):
with time_machine.travel(datetime(2020, 1, 1, tzinfo=timezone.utc), tick=False):
result = self.api_patch(user, url, data)
self.assert_json_success(result)
@ -60,10 +57,7 @@ class MutedTopicsTests(ZulipTestCase):
data = {"stream_id": stream.id, "topic": "test TOPIC", "op": "add"}
def mute_topic_for_user(user: UserProfile) -> None:
with mock.patch(
"zerver.views.user_topics.timezone_now",
return_value=datetime(2020, 1, 1, tzinfo=timezone.utc),
):
with time_machine.travel(datetime(2020, 1, 1, tzinfo=timezone.utc), tick=False):
result = self.api_patch(user, url, data)
self.assert_json_success(result)
@ -102,10 +96,7 @@ class MutedTopicsTests(ZulipTestCase):
mock_date_muted = datetime(2020, 1, 1, tzinfo=timezone.utc).timestamp()
for data in payloads:
with mock.patch(
"zerver.views.user_topics.timezone_now",
return_value=datetime(2020, 1, 1, tzinfo=timezone.utc),
):
with time_machine.travel(datetime(2020, 1, 1, tzinfo=timezone.utc), tick=False):
result = self.api_patch(user, url, data)
self.assert_json_success(result)