mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 05:53:43 +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.
68 lines
3.2 KiB
Python
68 lines
3.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
from django.conf import settings
|
|
|
|
from zerver.lib.actions import internal_send_private_message
|
|
from zerver.lib.test_classes import ZulipTestCase
|
|
from zerver.lib.test_helpers import message_stream_count, most_recent_message
|
|
from zerver.models import get_realm, get_user, get_system_bot, UserProfile
|
|
|
|
import ujson
|
|
|
|
class TutorialTests(ZulipTestCase):
|
|
def setUp(self) -> None:
|
|
super().setUp()
|
|
# This emulates the welcome message sent by the welcome bot to hamlet@zulip.com
|
|
# This is only a quick fix - ideally, we would have this message sent by the initialization
|
|
# code in populate_db.py
|
|
user = self.example_user('hamlet')
|
|
welcome_bot = get_system_bot(settings.WELCOME_BOT)
|
|
content = 'Shortened welcome message.'
|
|
internal_send_private_message(welcome_bot.realm, welcome_bot, user, content)
|
|
|
|
def test_tutorial_status(self) -> None:
|
|
email = self.example_email('hamlet')
|
|
self.login(email)
|
|
|
|
cases = [
|
|
('started', UserProfile.TUTORIAL_STARTED),
|
|
('finished', UserProfile.TUTORIAL_FINISHED),
|
|
]
|
|
for incoming_status, expected_db_status in cases:
|
|
params = dict(status=ujson.dumps(incoming_status))
|
|
result = self.client_post('/json/users/me/tutorial_status', params)
|
|
self.assert_json_success(result)
|
|
user = self.example_user('hamlet')
|
|
self.assertEqual(user.tutorial_status, expected_db_status)
|
|
|
|
def test_single_response_to_pm(self) -> None:
|
|
realm = get_realm('zulip')
|
|
user_email = 'hamlet@zulip.com'
|
|
bot_email = 'welcome-bot@zulip.com'
|
|
content = 'whatever'
|
|
self.login(user_email)
|
|
self.send_personal_message(user_email, bot_email, content)
|
|
user = get_user(user_email, realm)
|
|
user_messages = message_stream_count(user)
|
|
expected_response = ("Congratulations on your first reply! :tada:\n\n"
|
|
"Feel free to continue using this space to practice your new messaging "
|
|
"skills. Or, try clicking on some of the stream names to your left!")
|
|
self.assertEqual(most_recent_message(user).content, expected_response)
|
|
# Welcome bot shouldn't respond to further PMs.
|
|
self.send_personal_message(user_email, bot_email, content)
|
|
self.assertEqual(message_stream_count(user), user_messages+1)
|
|
|
|
def test_no_response_to_group_pm(self) -> None:
|
|
realm = get_realm('zulip') # Assume realm is always 'zulip'
|
|
user1_email = self.example_email('hamlet')
|
|
user2_email = self.example_email('cordelia')
|
|
bot_email = self.example_email('welcome_bot')
|
|
content = "whatever"
|
|
self.login(user1_email)
|
|
self.send_huddle_message(user1_email, [bot_email, user2_email], content)
|
|
user1 = get_user(user1_email, realm)
|
|
user1_messages = message_stream_count(user1)
|
|
self.assertEqual(most_recent_message(user1).content, content)
|
|
# Welcome bot should still respond to initial PM after group PM.
|
|
self.send_personal_message(user1_email, bot_email, content)
|
|
self.assertEqual(message_stream_count(user1), user1_messages+2)
|