mirror of
https://github.com/zulip/zulip.git
synced 2026-07-18 21:04:19 +08:00
2bd9c8cb42
This commit introduces a simple field where the user can now specify custom HTTP headers. This commit does not introduce an improved system for storing HTTP headers as fixtures - such a change would modify both the existing unit tests as well as this devtool.
106 lines
5.2 KiB
Python
106 lines
5.2 KiB
Python
import ujson
|
|
from mock import MagicMock, patch
|
|
from zerver.lib.test_classes import ZulipTestCase
|
|
from zerver.models import get_user, get_realm, Message, Stream
|
|
|
|
class TestIntegrationsDevPanel(ZulipTestCase):
|
|
|
|
zulip_realm = get_realm("zulip")
|
|
|
|
def test_check_send_webhook_fixture_message_for_error(self) -> None:
|
|
bot = get_user('webhook-bot@zulip.com', self.zulip_realm)
|
|
url = "/api/v1/external/airbrake?api_key={key}".format(key=bot.api_key)
|
|
target_url = "/devtools/integrations/check_send_webhook_fixture_message"
|
|
body = "{}" # This empty body should generate a KeyError on the webhook code side.
|
|
|
|
data = {
|
|
"url": url,
|
|
"body": body,
|
|
"custom_headers": "",
|
|
}
|
|
|
|
response = self.client_post(target_url, data)
|
|
|
|
self.assertEqual(response.status_code, 500) # Since the response would be forwarded.
|
|
expected_response = {"result": "error", "msg": "Internal server error"}
|
|
self.assertEqual(ujson.loads(response.content), expected_response)
|
|
|
|
def test_check_send_webhook_fixture_message_for_success_without_headers(self) -> None:
|
|
bot = get_user('webhook-bot@zulip.com', self.zulip_realm)
|
|
url = "/api/v1/external/airbrake?api_key={key}&stream=Denmark&topic=Airbrake Notifications".format(key=bot.api_key)
|
|
target_url = "/devtools/integrations/check_send_webhook_fixture_message"
|
|
with open("zerver/webhooks/airbrake/fixtures/error_message.json", "r") as f:
|
|
body = f.read()
|
|
|
|
data = {
|
|
"url": url,
|
|
"body": body,
|
|
"custom_headers": "",
|
|
}
|
|
|
|
response = self.client_post(target_url, data)
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
latest_msg = Message.objects.latest('id')
|
|
expected_message = "[ZeroDivisionError](https://zulip.airbrake.io/projects/125209/groups/1705190192091077626): \"Error message from logger\" occurred."
|
|
self.assertEqual(latest_msg.content, expected_message)
|
|
self.assertEqual(Stream.objects.get(id=latest_msg.recipient.type_id).name, "Denmark")
|
|
self.assertEqual(latest_msg.topic_name(), "Airbrake Notifications")
|
|
|
|
def test_check_send_webhook_fixture_message_for_success_with_headers(self) -> None:
|
|
bot = get_user('webhook-bot@zulip.com', self.zulip_realm)
|
|
url = "/api/v1/external/github?api_key={key}&stream=Denmark&topic=GitHub Notifications".format(key=bot.api_key)
|
|
target_url = "/devtools/integrations/check_send_webhook_fixture_message"
|
|
with open("zerver/webhooks/github/fixtures/ping_organization.json", "r") as f:
|
|
body = f.read()
|
|
|
|
data = {
|
|
"url": url,
|
|
"body": body,
|
|
"custom_headers": ujson.dumps({"X_GITHUB_EVENT": "ping"}),
|
|
}
|
|
|
|
response = self.client_post(target_url, data)
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
latest_msg = Message.objects.latest('id')
|
|
expected_message = "GitHub webhook has been successfully configured by eeshangarg."
|
|
self.assertEqual(latest_msg.content, expected_message)
|
|
self.assertEqual(Stream.objects.get(id=latest_msg.recipient.type_id).name, "Denmark")
|
|
self.assertEqual(latest_msg.topic_name(), "GitHub Notifications")
|
|
|
|
def test_get_fixtures_for_nonexistant_integration(self) -> None:
|
|
target_url = "/devtools/integrations/somerandomnonexistantintegration/fixtures"
|
|
response = self.client_get(target_url)
|
|
expected_response = {'msg': '"somerandomnonexistantintegration" is not a valid webhook integration.', 'result': 'error'}
|
|
self.assertEqual(response.status_code, 404)
|
|
self.assertEqual(ujson.loads(response.content), expected_response)
|
|
|
|
@patch("zerver.views.development.integrations.os.path.exists")
|
|
def test_get_fixtures_for_integration_without_fixtures(self, os_path_exists_mock: MagicMock) -> None:
|
|
os_path_exists_mock.return_value = False
|
|
target_url = "/devtools/integrations/airbrake/fixtures"
|
|
response = self.client_get(target_url)
|
|
expected_response = {'msg': 'The integration "airbrake" does not have fixtures.', 'result': 'error'}
|
|
self.assertEqual(response.status_code, 404)
|
|
self.assertEqual(ujson.loads(response.content), expected_response)
|
|
|
|
def test_get_fixtures_for_integration_without_json_fixtures(self) -> None:
|
|
target_url = "/devtools/integrations/deskdotcom/fixtures"
|
|
response = self.client_get(target_url)
|
|
expected_response = {'msg': 'The integration "deskdotcom" has non-JSON fixtures.', 'result': 'error'}
|
|
self.assertEqual(response.status_code, 400)
|
|
self.assertEqual(ujson.loads(response.content), expected_response)
|
|
|
|
def test_get_fixtures_for_success(self) -> None:
|
|
target_url = "/devtools/integrations/airbrake/fixtures"
|
|
response = self.client_get(target_url)
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertIsNotNone(ujson.loads(response.content)["fixtures"])
|
|
|
|
def test_get_dev_panel_page(self) -> None:
|
|
# Just to satisfy the test suite.
|
|
target_url = "/devtools/integrations/"
|
|
response = self.client_get(target_url)
|
|
self.assertEqual(response.status_code, 200)
|