mirror of https://github.com/zulip/zulip.git
96 lines
3.4 KiB
Python
Executable File
96 lines
3.4 KiB
Python
Executable File
from django.conf import settings
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.shortcuts import render, redirect
|
|
from django.test import Client
|
|
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__)), '../../')
|
|
client = Client()
|
|
|
|
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:
|
|
|
|
# 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)
|