mirror of
https://github.com/zulip/zulip.git
synced 2025-11-06 15:03:34 +00:00
9941b408a0
Importing the Django test client is somewhat expensive, and we only use it within one view function that's not used in production. So there's a significant startup-time performance optimization in doing an import inside the view code.
99 lines
3.6 KiB
Python
Executable File
99 lines
3.6 KiB
Python
Executable File
from django.conf import settings
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.shortcuts import render, redirect
|
|
from django.views.decorators.http import require_GET
|
|
from django.views.decorators.csrf import csrf_exempt
|
|
|
|
from zerver.models import get_realm, get_user
|
|
from zerver.lib.notifications import enqueue_welcome_emails
|
|
from zerver.lib.response import json_success
|
|
from zproject.email_backends import (
|
|
get_forward_address,
|
|
set_forward_address,
|
|
)
|
|
import urllib
|
|
from confirmation.models import Confirmation, confirmation_url
|
|
|
|
import os
|
|
from typing import List, Dict, Any, Optional
|
|
import datetime
|
|
ZULIP_PATH = os.path.join(os.path.dirname(os.path.abspath(__file__)), '../../')
|
|
|
|
def email_page(request: HttpRequest) -> HttpResponse:
|
|
if request.method == 'POST':
|
|
set_forward_address(request.POST["forward_address"])
|
|
return json_success()
|
|
try:
|
|
with open(settings.EMAIL_CONTENT_LOG_PATH, "r+") as f:
|
|
content = f.read()
|
|
except FileNotFoundError:
|
|
content = ""
|
|
return render(request, 'zerver/email_log.html',
|
|
{'log': content,
|
|
'forward_address': get_forward_address()})
|
|
|
|
def clear_emails(request: HttpRequest) -> HttpResponse:
|
|
try:
|
|
os.remove(settings.EMAIL_CONTENT_LOG_PATH)
|
|
except FileNotFoundError: # nocoverage
|
|
pass
|
|
return redirect(email_page)
|
|
|
|
@require_GET
|
|
def generate_all_emails(request: HttpRequest) -> HttpResponse:
|
|
# We import the Django test client inside the view function,
|
|
# because it isn't needed in production elsewhere, and not
|
|
# importing it saves ~50ms of unnecessary manage.py startup time.
|
|
from django.test import Client
|
|
client = Client()
|
|
|
|
# write fake data for all variables
|
|
registered_email = "hamlet@zulip.com"
|
|
unregistered_email_1 = "new-person@zulip.com"
|
|
unregistered_email_2 = "new-person-2@zulip.com"
|
|
realm = get_realm("zulip")
|
|
host_kwargs = {'HTTP_HOST': realm.host}
|
|
|
|
# Password reset email
|
|
result = client.post('/accounts/password/reset/', {'email': registered_email}, **host_kwargs)
|
|
assert result.status_code == 302
|
|
|
|
# Confirm account email
|
|
result = client.post('/accounts/home/', {'email': unregistered_email_1}, **host_kwargs)
|
|
assert result.status_code == 302
|
|
|
|
# Find account email
|
|
result = client.post('/accounts/find/', {'emails': registered_email}, **host_kwargs)
|
|
assert result.status_code == 302
|
|
|
|
# New login email
|
|
logged_in = client.login(dev_auth_username=registered_email, realm=realm)
|
|
assert logged_in
|
|
|
|
# New user invite and reminder emails
|
|
result = client.post("/json/invites",
|
|
{"invitee_emails": unregistered_email_2, "stream": ["Denmark"]},
|
|
**host_kwargs)
|
|
assert result.status_code == 200
|
|
|
|
# Verification for new email
|
|
result = client.patch('/json/settings',
|
|
urllib.parse.urlencode({'email': 'hamlets-new@zulip.com'}),
|
|
**host_kwargs)
|
|
assert result.status_code == 200
|
|
|
|
# Email change successful
|
|
key = Confirmation.objects.filter(type=Confirmation.EMAIL_CHANGE).latest('id').confirmation_key
|
|
url = confirmation_url(key, realm.host, Confirmation.EMAIL_CHANGE)
|
|
user_profile = get_user(registered_email, realm)
|
|
result = client.get(url)
|
|
assert result.status_code == 200
|
|
|
|
# Reset the email value so we can run this again
|
|
user_profile.email = registered_email
|
|
user_profile.save(update_fields=['email'])
|
|
|
|
# Follow up day1 day2 emails
|
|
enqueue_welcome_emails(user_profile)
|
|
return redirect(email_page)
|