mirror of
https://github.com/zulip/zulip.git
synced 2025-11-04 14:03:30 +00:00
2e8dec233e
Note that while the test mocks the actual message send, we now have a `get_stream` call in the queue worker, so we have to set up a real stream for testing (or we could have mocked that as well, but it didn't seem necessary). The setup queries add to the amount of queries reported by the test, plus the `get_stream` call. I just made the query count a digits regex, which is a little bit lame, but I don't think it's worth risking test flakes for this.
210 lines
9.5 KiB
Python
210 lines
9.5 KiB
Python
import time
|
|
from typing import List
|
|
|
|
from bs4 import BeautifulSoup
|
|
from django.conf import settings
|
|
from django.test import override_settings
|
|
from unittest.mock import Mock, patch
|
|
from zerver.lib.actions import create_stream_if_needed
|
|
from zerver.lib.realm_icon import get_realm_icon_url
|
|
from zerver.lib.test_classes import ZulipTestCase
|
|
from zerver.middleware import is_slow_query, write_log_line
|
|
from zerver.models import get_realm, get_system_bot
|
|
|
|
class SlowQueryTest(ZulipTestCase):
|
|
SLOW_QUERY_TIME = 10
|
|
log_data = {'extra': '[transport=websocket]',
|
|
'time_started': 0,
|
|
'bugdown_requests_start': 0,
|
|
'bugdown_time_start': 0,
|
|
'remote_cache_time_start': 0,
|
|
'remote_cache_requests_start': 0}
|
|
|
|
def test_is_slow_query(self) -> None:
|
|
self.assertFalse(is_slow_query(1.1, '/some/random/url'))
|
|
self.assertTrue(is_slow_query(2, '/some/random/url'))
|
|
self.assertTrue(is_slow_query(5.1, '/activity'))
|
|
self.assertFalse(is_slow_query(2, '/activity'))
|
|
self.assertFalse(is_slow_query(2, '/json/report/error'))
|
|
self.assertFalse(is_slow_query(2, '/api/v1/deployments/report_error'))
|
|
self.assertFalse(is_slow_query(2, '/realm_activity/whatever'))
|
|
self.assertFalse(is_slow_query(2, '/user_activity/whatever'))
|
|
self.assertFalse(is_slow_query(9, '/accounts/webathena_kerberos_login/'))
|
|
self.assertTrue(is_slow_query(11, '/accounts/webathena_kerberos_login/'))
|
|
|
|
@override_settings(SLOW_QUERY_LOGS_STREAM="logs")
|
|
@patch('logging.info')
|
|
def test_slow_query_log(self, mock_logging_info: Mock) -> None:
|
|
error_bot = get_system_bot(settings.ERROR_BOT)
|
|
create_stream_if_needed(error_bot.realm, settings.SLOW_QUERY_LOGS_STREAM)
|
|
|
|
self.log_data['time_started'] = time.time() - self.SLOW_QUERY_TIME
|
|
write_log_line(self.log_data, path='/socket/open', method='SOCKET',
|
|
remote_ip='123.456.789.012', email='unknown', client_name='?')
|
|
last_message = self.get_last_message()
|
|
self.assertEqual(last_message.sender.email, "error-bot@zulip.com")
|
|
self.assertIn("logs", str(last_message.recipient))
|
|
self.assertEqual(last_message.topic_name(), "testserver: slow queries")
|
|
self.assertRegexpMatches(last_message.content,
|
|
r"123\.456\.789\.012 SOCKET 200 10\.\ds .*")
|
|
|
|
@override_settings(ERROR_BOT=None)
|
|
@patch('logging.info')
|
|
@patch('zerver.lib.actions.internal_send_stream_message')
|
|
def test_slow_query_log_without_error_bot(self,
|
|
mock_internal_send_stream_message: Mock,
|
|
mock_logging_info: Mock) -> None:
|
|
self.log_data['time_started'] = time.time() - self.SLOW_QUERY_TIME
|
|
write_log_line(self.log_data, path='/socket/open', method='SOCKET',
|
|
remote_ip='123.456.789.012', email='unknown', client_name='?')
|
|
mock_internal_send_stream_message.assert_not_called()
|
|
|
|
class OpenGraphTest(ZulipTestCase):
|
|
def check_title_and_description(self, path: str, title: str,
|
|
in_description: List[str],
|
|
not_in_description: List[str],
|
|
status_code: int=200) -> None:
|
|
response = self.client_get(path)
|
|
self.assertEqual(response.status_code, status_code)
|
|
decoded = response.content.decode('utf-8')
|
|
bs = BeautifulSoup(decoded, features='lxml')
|
|
open_graph_title = bs.select_one('meta[property="og:title"]').get('content')
|
|
self.assertEqual(open_graph_title, title)
|
|
|
|
open_graph_description = bs.select_one('meta[property="og:description"]').get('content')
|
|
for substring in in_description:
|
|
self.assertIn(substring, open_graph_description)
|
|
for substring in not_in_description:
|
|
self.assertNotIn(substring, open_graph_description)
|
|
|
|
def test_admonition_and_link(self) -> None:
|
|
# disable-message-edit-history starts with an {!admin-only.md!}, and has a link
|
|
# in the first paragraph.
|
|
self.check_title_and_description(
|
|
'/help/disable-message-edit-history',
|
|
"Disable message edit history (Zulip Help Center)",
|
|
["By default, Zulip displays messages",
|
|
"users can view the edit history of a message. | To remove the",
|
|
"best to delete the message entirely. "],
|
|
["Disable message edit history", "feature is only available", "Related articles",
|
|
"Restrict message editing"]
|
|
)
|
|
|
|
def test_double_quotes(self) -> None:
|
|
# night-mode has a quoted string "night mode"
|
|
self.check_title_and_description(
|
|
'/help/night-mode',
|
|
"Night mode (Zulip Help Center)",
|
|
['By default, Zulip has a white background. ',
|
|
'Zulip also provides a "night mode", which is great for working in a dark space.'],
|
|
[]
|
|
)
|
|
|
|
def test_settings_tab(self) -> None:
|
|
# deactivate-your-account starts with {settings_tab|your-account}
|
|
self.check_title_and_description(
|
|
'/help/deactivate-your-account',
|
|
"Deactivate your account (Zulip Help Center)",
|
|
["Any bots that you maintain will be disabled. | Deactivating "],
|
|
["Confirm by clicking", " ", "\n"])
|
|
|
|
def test_tabs(self) -> None:
|
|
# logging-out starts with {start_tabs}
|
|
self.check_title_and_description(
|
|
'/help/logging-out',
|
|
"Logging out (Zulip Help Center)",
|
|
# Ideally we'd do something better here
|
|
["We're here to help! Email us at zulip-admin@example.com with questions, feedback, or " +
|
|
"feature requests."],
|
|
["Click on the gear"])
|
|
|
|
def test_index_pages(self) -> None:
|
|
self.check_title_and_description(
|
|
'/help/',
|
|
"Zulip Help Center",
|
|
[("Zulip is a group chat app. Its most distinctive characteristic is that "
|
|
"conversation within an organization is divided into “streams” and further ")], [])
|
|
|
|
self.check_title_and_description(
|
|
'/api/',
|
|
"Zulip API Documentation",
|
|
[("Zulip's APIs allow you to integrate other services with Zulip. This "
|
|
"guide should help you find the API you need:")], [])
|
|
|
|
def test_nonexistent_page(self) -> None:
|
|
self.check_title_and_description(
|
|
'/help/not-a-real-page',
|
|
# Probably we should make this "Zulip Help Center"
|
|
"No such article. (Zulip Help Center)",
|
|
["No such article. | We're here to help!",
|
|
"Email us at zulip-admin@example.com with questions, feedback, or feature requests."],
|
|
[],
|
|
# Test that our open graph logic doesn't throw a 500
|
|
404)
|
|
|
|
def test_login_page_simple_description(self) -> None:
|
|
name = 'Zulip Dev'
|
|
description = "The Zulip development environment default organization. It's great for testing!"
|
|
|
|
self.check_title_and_description(
|
|
'/login/',
|
|
name,
|
|
[description],
|
|
[])
|
|
|
|
def test_login_page_markdown_description(self) -> None:
|
|
realm = get_realm('zulip')
|
|
description = ("Welcome to **Clojurians Zulip** - the place where the Clojure community meets.\n\n"
|
|
"Before you signup/login:\n\n"
|
|
"* note-1\n"
|
|
"* note-2\n"
|
|
"* note-3\n\n"
|
|
"Enjoy!")
|
|
realm.description = description
|
|
realm.save(update_fields=['description'])
|
|
|
|
self.check_title_and_description(
|
|
'/login/',
|
|
'Zulip Dev',
|
|
['Welcome to Clojurians Zulip - the place where the Clojure community meets',
|
|
'* note-1 * note-2 * note-3 | Enjoy!'],
|
|
[])
|
|
|
|
def test_login_page_realm_icon(self) -> None:
|
|
realm = get_realm('zulip')
|
|
realm.icon_source = 'U'
|
|
realm.save(update_fields=['icon_source'])
|
|
realm_icon = get_realm_icon_url(realm)
|
|
|
|
response = self.client_get('/login/')
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
decoded = response.content.decode('utf-8')
|
|
bs = BeautifulSoup(decoded, features='lxml')
|
|
open_graph_image = bs.select_one('meta[property="og:image"]').get('content')
|
|
self.assertEqual(open_graph_image, '%s%s' % (realm.uri, realm_icon))
|
|
|
|
def test_login_page_realm_icon_absolute_url(self) -> None:
|
|
realm = get_realm('zulip')
|
|
realm.icon_source = 'U'
|
|
realm.save(update_fields=['icon_source'])
|
|
icon_url = "https://foo.s3.amazonaws.com/%s/realm/icon.png?version=%s" % (realm.id, 1)
|
|
with patch('zerver.lib.realm_icon.upload_backend.get_realm_icon_url', return_value=icon_url):
|
|
response = self.client_get('/login/')
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
decoded = response.content.decode('utf-8')
|
|
bs = BeautifulSoup(decoded, features='lxml')
|
|
open_graph_image = bs.select_one('meta[property="og:image"]').get('content')
|
|
self.assertEqual(open_graph_image, icon_url)
|
|
|
|
def test_no_realm_api_page_og_url(self) -> None:
|
|
response = self.client_get('/api/', subdomain='')
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
decoded = response.content.decode('utf-8')
|
|
bs = BeautifulSoup(decoded, features='lxml')
|
|
open_graph_url = bs.select_one('meta[property="og:url"]').get('content')
|
|
|
|
self.assertTrue(open_graph_url.endswith('/api/'))
|