mirror of https://github.com/zulip/zulip.git
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import absolute_import
|
|
from __future__ import print_function
|
|
|
|
from typing import Any, Dict
|
|
|
|
from zerver.lib.test_helpers import (
|
|
get_user_profile_by_email,
|
|
most_recent_message,
|
|
)
|
|
|
|
from zerver.lib.test_classes import (
|
|
ZulipTestCase,
|
|
)
|
|
|
|
from zerver.models import (
|
|
UserProfile,
|
|
)
|
|
|
|
import ujson
|
|
|
|
def fix_params(raw_params):
|
|
# type: (Dict[str, Any]) -> Dict[str, str]
|
|
# A few of our few legacy endpoints need their
|
|
# individual parameters serialized as JSON.
|
|
return {k: ujson.dumps(v) for k, v in raw_params.items()}
|
|
|
|
class TutorialTests(ZulipTestCase):
|
|
def test_send_message(self):
|
|
# type: () -> None
|
|
email = 'hamlet@zulip.com'
|
|
user = get_user_profile_by_email(email)
|
|
self.login(email)
|
|
|
|
welcome_bot = get_user_profile_by_email("welcome-bot@zulip.com")
|
|
|
|
raw_params = dict(
|
|
type='stream',
|
|
recipient='Denmark',
|
|
topic='welcome',
|
|
content='hello'
|
|
)
|
|
params = fix_params(raw_params)
|
|
|
|
result = self.client_post("/json/tutorial_send_message", params)
|
|
|
|
self.assert_json_success(result)
|
|
message = most_recent_message(user)
|
|
self.assertEqual(message.content, 'hello')
|
|
self.assertEqual(message.sender, welcome_bot)
|
|
|
|
# now test some error cases
|
|
|
|
result = self.client_post("/json/tutorial_send_message", {})
|
|
self.assert_json_error(result, "Missing 'type' argument")
|
|
|
|
result = self.client_post("/json/tutorial_send_message", raw_params)
|
|
self.assert_json_error(result, 'argument "type" is not valid json.')
|
|
|
|
raw_params = dict(
|
|
type='INVALID',
|
|
recipient='Denmark',
|
|
topic='welcome',
|
|
content='hello'
|
|
)
|
|
params = fix_params(raw_params)
|
|
result = self.client_post("/json/tutorial_send_message", params)
|
|
self.assert_json_error(result, 'Bad data passed in to tutorial_send_message')
|
|
|
|
def test_tutorial_status(self):
|
|
# type: () -> None
|
|
email = 'hamlet@zulip.com'
|
|
self.login(email)
|
|
|
|
cases = [
|
|
('started', UserProfile.TUTORIAL_STARTED),
|
|
('finished', UserProfile.TUTORIAL_FINISHED),
|
|
]
|
|
for incoming_status, expected_db_status in cases:
|
|
raw_params = dict(status=incoming_status)
|
|
params = fix_params(raw_params)
|
|
result = self.client_post('/json/tutorial_status', params)
|
|
self.assert_json_success(result)
|
|
user = get_user_profile_by_email(email)
|
|
self.assertEqual(user.tutorial_status, expected_db_status)
|