mirror of
https://github.com/zulip/zulip.git
synced 2025-11-20 22:48:16 +00:00
bbf2474bd0
MigrationsTestCase is intentionally omitted from this, since migrations tests are different in their nature and so whatever setUp() ZulipTestCase may do in the future, MigrationsTestCase may not necessarily want to replicate.
62 lines
2.8 KiB
Python
62 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from zerver.lib.actions import do_mark_hotspot_as_read, do_create_user
|
|
from zerver.lib.hotspots import ALL_HOTSPOTS, get_next_hotspots
|
|
from zerver.lib.test_classes import ZulipTestCase
|
|
from zerver.models import UserProfile, UserHotspot, get_realm
|
|
|
|
import ujson
|
|
|
|
# Splitting this out, since I imagine this will eventually have most of the
|
|
# complicated hotspots logic.
|
|
class TestGetNextHotspots(ZulipTestCase):
|
|
def setUp(self) -> None:
|
|
super().setUp()
|
|
self.user = do_create_user(
|
|
'user@zulip.com', 'password', get_realm('zulip'), 'user', 'user')
|
|
|
|
def test_first_hotspot(self) -> None:
|
|
hotspots = get_next_hotspots(self.user)
|
|
self.assertEqual(len(hotspots), 1)
|
|
self.assertEqual(hotspots[0]['name'], 'intro_reply')
|
|
|
|
def test_some_done_some_not(self) -> None:
|
|
do_mark_hotspot_as_read(self.user, 'intro_reply')
|
|
do_mark_hotspot_as_read(self.user, 'intro_compose')
|
|
hotspots = get_next_hotspots(self.user)
|
|
self.assertEqual(len(hotspots), 1)
|
|
self.assertEqual(hotspots[0]['name'], 'intro_streams')
|
|
|
|
def test_all_done(self) -> None:
|
|
self.assertNotEqual(self.user.tutorial_status, UserProfile.TUTORIAL_FINISHED)
|
|
for hotspot in ALL_HOTSPOTS:
|
|
do_mark_hotspot_as_read(self.user, hotspot)
|
|
self.assertEqual(self.user.tutorial_status, UserProfile.TUTORIAL_FINISHED)
|
|
self.assertEqual(get_next_hotspots(self.user), [])
|
|
|
|
def test_send_all(self) -> None:
|
|
with self.settings(DEVELOPMENT=True, ALWAYS_SEND_ALL_HOTSPOTS = True):
|
|
self.assertEqual(len(ALL_HOTSPOTS), len(get_next_hotspots(self.user)))
|
|
|
|
class TestHotspots(ZulipTestCase):
|
|
def test_do_mark_hotspot_as_read(self) -> None:
|
|
user = self.example_user('hamlet')
|
|
do_mark_hotspot_as_read(user, 'intro_compose')
|
|
self.assertEqual(list(UserHotspot.objects.filter(user=user)
|
|
.values_list('hotspot', flat=True)), ['intro_compose'])
|
|
|
|
def test_hotspots_url_endpoint(self) -> None:
|
|
user = self.example_user('hamlet')
|
|
self.login(user.email)
|
|
result = self.client_post('/json/users/me/hotspots',
|
|
{'hotspot': ujson.dumps('intro_reply')})
|
|
self.assert_json_success(result)
|
|
self.assertEqual(list(UserHotspot.objects.filter(user=user)
|
|
.values_list('hotspot', flat=True)), ['intro_reply'])
|
|
|
|
result = self.client_post('/json/users/me/hotspots',
|
|
{'hotspot': ujson.dumps('invalid')})
|
|
self.assert_json_error(result, "Unknown hotspot: invalid")
|
|
self.assertEqual(list(UserHotspot.objects.filter(user=user)
|
|
.values_list('hotspot', flat=True)), ['intro_reply'])
|