mirror of https://github.com/zulip/zulip.git
399 lines
14 KiB
Python
399 lines
14 KiB
Python
import re
|
|
from typing import Any, Dict
|
|
|
|
import orjson
|
|
from django.core.exceptions import ValidationError
|
|
from django.http import HttpResponse
|
|
|
|
from zerver.lib.test_classes import ZulipTestCase
|
|
from zerver.lib.validator import check_widget_content
|
|
from zerver.lib.widget import get_widget_data, get_widget_type
|
|
from zerver.models import SubMessage, UserProfile
|
|
|
|
|
|
class WidgetContentTestCase(ZulipTestCase):
|
|
def test_validation(self) -> None:
|
|
def assert_error(obj: object, msg: str) -> None:
|
|
with self.assertRaisesRegex(ValidationError, re.escape(msg)):
|
|
check_widget_content(obj)
|
|
|
|
assert_error(5, "widget_content is not a dict")
|
|
|
|
assert_error({}, "widget_type is not in widget_content")
|
|
|
|
assert_error(dict(widget_type="whatever"), "extra_data is not in widget_content")
|
|
|
|
assert_error(dict(widget_type="zform", extra_data=4), "extra_data is not a dict")
|
|
|
|
assert_error(dict(widget_type="bogus", extra_data={}), "unknown widget type: bogus")
|
|
|
|
extra_data: Dict[str, Any] = {}
|
|
obj = dict(widget_type="zform", extra_data=extra_data)
|
|
|
|
assert_error(obj, "zform is missing type field")
|
|
|
|
extra_data["type"] = "bogus"
|
|
assert_error(obj, "unknown zform type: bogus")
|
|
|
|
extra_data["type"] = "choices"
|
|
assert_error(obj, "heading key is missing from extra_data")
|
|
|
|
extra_data["heading"] = "whatever"
|
|
assert_error(obj, "choices key is missing from extra_data")
|
|
|
|
extra_data["choices"] = 99
|
|
assert_error(obj, 'extra_data["choices"] is not a list')
|
|
|
|
extra_data["choices"] = [99]
|
|
assert_error(obj, 'extra_data["choices"][0] is not a dict')
|
|
|
|
extra_data["choices"] = [
|
|
dict(long_name="foo", reply="bar"),
|
|
]
|
|
assert_error(obj, 'short_name key is missing from extra_data["choices"][0]')
|
|
|
|
extra_data["choices"] = [
|
|
dict(short_name="a", long_name="foo", reply="bar"),
|
|
]
|
|
|
|
check_widget_content(obj)
|
|
|
|
def test_message_error_handling(self) -> None:
|
|
sender = self.example_user("cordelia")
|
|
stream_name = "Verona"
|
|
|
|
payload = dict(
|
|
type="stream",
|
|
to=stream_name,
|
|
topic="whatever",
|
|
content="whatever",
|
|
)
|
|
|
|
payload["widget_content"] = "{{{{{{" # unparsable
|
|
result = self.api_post(sender, "/api/v1/messages", payload)
|
|
self.assert_json_error_contains(result, "Widgets: API programmer sent invalid JSON")
|
|
|
|
bogus_data = dict(color="red", foo="bar", x=2)
|
|
payload["widget_content"] = orjson.dumps(bogus_data).decode()
|
|
result = self.api_post(sender, "/api/v1/messages", payload)
|
|
self.assert_json_error_contains(result, "Widgets: widget_type is not in widget_content")
|
|
|
|
def test_get_widget_data_for_non_widget_messages(self) -> None:
|
|
# This is a pretty important test, despite testing the
|
|
# "negative" case. We never want widgets to interfere
|
|
# with normal messages.
|
|
|
|
test_messages = [
|
|
"",
|
|
" ",
|
|
"this is an ordinary message",
|
|
"/bogus_command",
|
|
"/me shrugs",
|
|
"use /poll",
|
|
]
|
|
|
|
for message in test_messages:
|
|
self.assertEqual(get_widget_data(content=message), (None, None))
|
|
|
|
# Add a positive check for context
|
|
self.assertEqual(get_widget_data(content="/todo"), ("todo", None))
|
|
|
|
def test_explicit_widget_content(self) -> None:
|
|
# Users can send widget_content directly on messages
|
|
# using the `widget_content` field.
|
|
|
|
sender = self.example_user("cordelia")
|
|
stream_name = "Verona"
|
|
content = "does-not-matter"
|
|
zform_data = dict(
|
|
type="choices",
|
|
heading="Options:",
|
|
choices=[],
|
|
)
|
|
|
|
widget_content = dict(
|
|
widget_type="zform",
|
|
extra_data=zform_data,
|
|
)
|
|
|
|
check_widget_content(widget_content)
|
|
|
|
payload = dict(
|
|
type="stream",
|
|
to=stream_name,
|
|
topic="whatever",
|
|
content=content,
|
|
widget_content=orjson.dumps(widget_content).decode(),
|
|
)
|
|
result = self.api_post(sender, "/api/v1/messages", payload)
|
|
self.assert_json_success(result)
|
|
|
|
message = self.get_last_message()
|
|
self.assertEqual(message.content, content)
|
|
|
|
expected_submessage_content = dict(
|
|
widget_type="zform",
|
|
extra_data=zform_data,
|
|
)
|
|
|
|
submessage = SubMessage.objects.get(message_id=message.id)
|
|
self.assertEqual(submessage.msg_type, "widget")
|
|
self.assertEqual(orjson.loads(submessage.content), expected_submessage_content)
|
|
|
|
def test_todo(self) -> None:
|
|
# This also helps us get test coverage that could apply
|
|
# to future widgets.
|
|
|
|
sender = self.example_user("cordelia")
|
|
stream_name = "Verona"
|
|
content = "/todo"
|
|
|
|
payload = dict(
|
|
type="stream",
|
|
to=stream_name,
|
|
topic="whatever",
|
|
content=content,
|
|
)
|
|
result = self.api_post(sender, "/api/v1/messages", payload)
|
|
self.assert_json_success(result)
|
|
|
|
message = self.get_last_message()
|
|
self.assertEqual(message.content, content)
|
|
|
|
expected_submessage_content = dict(
|
|
widget_type="todo",
|
|
extra_data=None,
|
|
)
|
|
|
|
submessage = SubMessage.objects.get(message_id=message.id)
|
|
self.assertEqual(submessage.msg_type, "widget")
|
|
self.assertEqual(orjson.loads(submessage.content), expected_submessage_content)
|
|
|
|
def test_poll_command_extra_data(self) -> None:
|
|
sender = self.example_user("cordelia")
|
|
stream_name = "Verona"
|
|
# We test for both trailing and leading spaces, along with blank lines
|
|
# for the poll options.
|
|
content = "/poll What is your favorite color?\n\nRed\nGreen \n\n Blue\n - Yellow"
|
|
|
|
payload = dict(
|
|
type="stream",
|
|
to=stream_name,
|
|
topic="whatever",
|
|
content=content,
|
|
)
|
|
result = self.api_post(sender, "/api/v1/messages", payload)
|
|
self.assert_json_success(result)
|
|
|
|
message = self.get_last_message()
|
|
self.assertEqual(message.content, content)
|
|
|
|
expected_submessage_content = dict(
|
|
widget_type="poll",
|
|
extra_data=dict(
|
|
options=["Red", "Green", "Blue", "Yellow"],
|
|
question="What is your favorite color?",
|
|
),
|
|
)
|
|
|
|
submessage = SubMessage.objects.get(message_id=message.id)
|
|
self.assertEqual(submessage.msg_type, "widget")
|
|
self.assertEqual(orjson.loads(submessage.content), expected_submessage_content)
|
|
|
|
# Now don't supply a question.
|
|
|
|
content = "/poll"
|
|
payload["content"] = content
|
|
result = self.api_post(sender, "/api/v1/messages", payload)
|
|
self.assert_json_success(result)
|
|
|
|
expected_submessage_content = dict(
|
|
widget_type="poll",
|
|
extra_data=dict(
|
|
options=[],
|
|
question="",
|
|
),
|
|
)
|
|
|
|
message = self.get_last_message()
|
|
self.assertEqual(message.content, content)
|
|
submessage = SubMessage.objects.get(message_id=message.id)
|
|
self.assertEqual(submessage.msg_type, "widget")
|
|
self.assertEqual(orjson.loads(submessage.content), expected_submessage_content)
|
|
|
|
def test_poll_permissions(self) -> None:
|
|
cordelia = self.example_user("cordelia")
|
|
hamlet = self.example_user("hamlet")
|
|
stream_name = "Verona"
|
|
content = "/poll Preference?\n\nyes\nno"
|
|
|
|
payload = dict(
|
|
type="stream",
|
|
to=stream_name,
|
|
topic="whatever",
|
|
content=content,
|
|
)
|
|
result = self.api_post(cordelia, "/api/v1/messages", payload)
|
|
self.assert_json_success(result)
|
|
|
|
message = self.get_last_message()
|
|
|
|
def post(sender: UserProfile, data: Dict[str, object]) -> HttpResponse:
|
|
payload = dict(
|
|
message_id=message.id, msg_type="widget", content=orjson.dumps(data).decode()
|
|
)
|
|
return self.api_post(sender, "/api/v1/submessage", payload)
|
|
|
|
result = post(cordelia, dict(type="question", question="Tabs or spaces?"))
|
|
self.assert_json_success(result)
|
|
|
|
result = post(hamlet, dict(type="question", question="Tabs or spaces?"))
|
|
self.assert_json_error(result, "You can't edit a question unless you are the author.")
|
|
|
|
def test_poll_type_validation(self) -> None:
|
|
sender = self.example_user("cordelia")
|
|
stream_name = "Verona"
|
|
content = "/poll Preference?\n\nyes\nno"
|
|
|
|
payload = dict(
|
|
type="stream",
|
|
to=stream_name,
|
|
topic="whatever",
|
|
content=content,
|
|
)
|
|
result = self.api_post(sender, "/api/v1/messages", payload)
|
|
self.assert_json_success(result)
|
|
|
|
message = self.get_last_message()
|
|
|
|
def post_submessage(content: str) -> HttpResponse:
|
|
payload = dict(
|
|
message_id=message.id,
|
|
msg_type="widget",
|
|
content=content,
|
|
)
|
|
return self.api_post(sender, "/api/v1/submessage", payload)
|
|
|
|
def assert_error(content: str, error: str) -> None:
|
|
result = post_submessage(content)
|
|
self.assert_json_error_contains(result, error)
|
|
|
|
assert_error("bogus", "Invalid json for submessage")
|
|
assert_error('""', "not a dict")
|
|
assert_error("[]", "not a dict")
|
|
|
|
assert_error('{"type": "bogus"}', "Unknown type for poll data: bogus")
|
|
assert_error('{"type": "vote"}', "key is missing")
|
|
assert_error('{"type": "vote", "key": "1,1,", "vote": 99}', "Invalid poll data")
|
|
|
|
assert_error('{"type": "question"}', "key is missing")
|
|
assert_error('{"type": "question", "question": 7}', "not a string")
|
|
|
|
assert_error('{"type": "new_option"}', "key is missing")
|
|
assert_error('{"type": "new_option", "idx": 7, "option": 999}', "not a string")
|
|
assert_error('{"type": "new_option", "idx": -1, "option": "pizza"}', "too small")
|
|
assert_error('{"type": "new_option", "idx": 1001, "option": "pizza"}', "too large")
|
|
assert_error('{"type": "new_option", "idx": "bogus", "option": "maybe"}', "not an int")
|
|
|
|
def assert_success(data: Dict[str, object]) -> None:
|
|
content = orjson.dumps(data).decode()
|
|
result = post_submessage(content)
|
|
self.assert_json_success(result)
|
|
|
|
# Note that we only validate for types. The server code may, for,
|
|
# example, allow a vote for a non-existing option, and we rely
|
|
# on the clients to ignore those.
|
|
|
|
assert_success(dict(type="vote", key="1,1", vote=1))
|
|
assert_success(dict(type="new_option", idx=7, option="maybe"))
|
|
assert_success(dict(type="question", question="what's for dinner?"))
|
|
|
|
def test_todo_type_validation(self) -> None:
|
|
sender = self.example_user("cordelia")
|
|
stream_name = "Verona"
|
|
content = "/todo"
|
|
|
|
payload = dict(
|
|
type="stream",
|
|
to=stream_name,
|
|
topic="whatever",
|
|
content=content,
|
|
)
|
|
result = self.api_post(sender, "/api/v1/messages", payload)
|
|
self.assert_json_success(result)
|
|
|
|
message = self.get_last_message()
|
|
|
|
def post_submessage(content: str) -> HttpResponse:
|
|
payload = dict(
|
|
message_id=message.id,
|
|
msg_type="widget",
|
|
content=content,
|
|
)
|
|
return self.api_post(sender, "/api/v1/submessage", payload)
|
|
|
|
def assert_error(content: str, error: str) -> None:
|
|
result = post_submessage(content)
|
|
self.assert_json_error_contains(result, error)
|
|
|
|
assert_error('{"type": "bogus"}', "Unknown type for todo data: bogus")
|
|
|
|
assert_error('{"type": "new_task"}', "key is missing")
|
|
assert_error(
|
|
'{"type": "new_task", "key": 7, "task": 7, "desc": "", "completed": false}',
|
|
'data["task"] is not a string',
|
|
)
|
|
assert_error(
|
|
'{"type": "new_task", "key": -1, "task": "eat", "desc": "", "completed": false}',
|
|
'data["key"] is too small',
|
|
)
|
|
assert_error(
|
|
'{"type": "new_task", "key": 1001, "task": "eat", "desc": "", "completed": false}',
|
|
'data["key"] is too large',
|
|
)
|
|
|
|
assert_error('{"type": "strike"}', "key is missing")
|
|
assert_error('{"type": "strike", "key": 999}', 'data["key"] is not a string')
|
|
|
|
def assert_success(data: Dict[str, object]) -> None:
|
|
content = orjson.dumps(data).decode()
|
|
result = post_submessage(content)
|
|
self.assert_json_success(result)
|
|
|
|
assert_success(dict(type="new_task", key=7, task="eat", desc="", completed=False))
|
|
assert_success(dict(type="strike", key="5,9"))
|
|
|
|
def test_get_widget_type(self) -> None:
|
|
sender = self.example_user("cordelia")
|
|
stream_name = "Verona"
|
|
# We test for both trailing and leading spaces, along with blank lines
|
|
# for the poll options.
|
|
content = "/poll Preference?\n\nyes\nno"
|
|
|
|
payload = dict(
|
|
type="stream",
|
|
to=stream_name,
|
|
topic="whatever",
|
|
content=content,
|
|
)
|
|
result = self.api_post(sender, "/api/v1/messages", payload)
|
|
self.assert_json_success(result)
|
|
|
|
message = self.get_last_message()
|
|
|
|
[submessage] = SubMessage.objects.filter(message_id=message.id)
|
|
|
|
self.assertEqual(get_widget_type(message_id=message.id), "poll")
|
|
|
|
submessage.content = "bogus non-json"
|
|
submessage.save()
|
|
self.assertEqual(get_widget_type(message_id=message.id), None)
|
|
|
|
submessage.content = '{"bogus": 1}'
|
|
submessage.save()
|
|
self.assertEqual(get_widget_type(message_id=message.id), None)
|
|
|
|
submessage.content = '{"widget_type": "todo"}'
|
|
submessage.save()
|
|
self.assertEqual(get_widget_type(message_id=message.id), "todo")
|