paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-10-23 04:52:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

generate-integration-docs-screenshot: Notify on webhook's stream.

Browse Source 
Use the stream specified by `stream_name` on the WebhookIntegration, instead
of using a common stream to send messages.
This commit is contained in:
Puneeth Chaganti
2020-04-24 15:00:24 +05:30
committed by Tim Abbott
parent 844bca55f8
commit ac7d8ce4b3
1 changed files with 13 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tools/generate-integration-docs-screenshot
View File

@@ -29,13 +29,14 @@ from typing import Any, Dict, Optional
import requests
import ujson
from tools.lib.test_script import prepare_puppeteer_run
from zerver.models import UserProfile, Message, get_user_by_delivery_email, get_realm
from zerver.lib.actions import do_create_user, notify_created_bot
from zerver.lib.actions import (
do_create_user, notify_created_bot, bulk_add_subscriptions, do_change_avatar_fields)
from zerver.lib.streams import create_stream_if_needed
from zerver.lib.upload import upload_avatar_image
from zerver.lib.actions import do_change_avatar_fields
from zerver.lib.integrations import WebhookIntegration, INTEGRATIONS, split_fixture_path
from zerver.lib.webhooks.common import get_fixture_http_headers
from tools.lib.test_script import prepare_puppeteer_run
from zerver.lib.storage import static_path
def create_integration_bot(integration: WebhookIntegration) -> UserProfile:
@@ -65,6 +66,11 @@ def create_integration_bot(integration: WebhookIntegration) -> UserProfile:
return bot
def create_integration_stream(integration: WebhookIntegration, bot: UserProfile) -> None:
assert isinstance(bot.bot_owner, UserProfile)
stream, created = create_stream_if_needed(bot.bot_owner.realm, integration.stream_name)
bulk_add_subscriptions([stream], [bot, bot.bot_owner], from_stream_creation=created)
def get_integration(integration_name: str) -> WebhookIntegration:
integration = INTEGRATIONS[integration_name]
assert isinstance(integration, WebhookIntegration), "Not a WebhookIntegration"
@@ -101,8 +107,9 @@ def send_bot_payload_message(bot: UserProfile, integration: WebhookIntegration,
Message.objects.filter(sender=bot).delete()
assert isinstance(bot.bot_owner, UserProfile)
url = "{}/{}?api_key={}&stream=devel".format(
bot.bot_owner.realm.uri, integration.url, bot.api_key
stream = integration.stream_name or 'devel'
url = "{}/{}?api_key={}&stream={}".format(
bot.bot_owner.realm.uri, integration.url, bot.api_key, stream
)
with open(fixture_path) as f:
data = ujson.load(f)
@@ -143,5 +150,6 @@ prepare_puppeteer_run()
integration_name, fixture_name = split_fixture_path(options.fixture)
integration = get_integration(integration_name)
bot = create_integration_bot(integration)
create_integration_stream(integration, bot)
send_bot_payload_message(bot, integration, options.fixture, options.custom_headers)
capture_last_message_screenshot(bot, integration, fixture_name)