app-MAIL-temp/tests/utils.py

92 lines
2.2 KiB
Python
Raw Normal View History

2022-03-17 19:03:36 +01:00
import email
2020-11-15 19:34:13 +01:00
import json
2022-03-17 19:03:36 +01:00
import os
2022-03-18 15:44:07 +01:00
import random
import string
2022-03-17 19:03:36 +01:00
from email.message import EmailMessage
2022-03-18 15:55:59 +01:00
from typing import Optional, Dict
2020-11-15 19:34:13 +01:00
2022-03-18 15:44:07 +01:00
import jinja2
from flask import url_for
from app.models import User, PartnerUser
from app.proton.utils import get_proton_partner
from app.utils import random_string
def create_new_user(email: Optional[str] = None, name: Optional[str] = None) -> User:
if not email:
email = f"user_{random_token(10)}@mailbox.test"
if not name:
name = "Test User"
# new user has a different email address
user = User.create(
email=email,
2020-12-31 13:59:03 +01:00
password="password",
name=name,
2020-12-31 13:59:03 +01:00
activated=True,
flush=True,
)
return user
def create_partner_linked_user() -> tuple[User, PartnerUser]:
user = create_new_user()
partner_user = PartnerUser.create(
partner_id=get_proton_partner().id,
user_id=user.id,
external_user_id=random_token(10),
flush=True,
)
return user, partner_user
def login(flask_client, user: Optional[User] = None) -> User:
if not user:
user = create_new_user()
r = flask_client.post(
url_for("auth.login"),
data={"email": user.email, "password": "password"},
follow_redirects=True,
)
assert r.status_code == 200
assert b"/auth/logout" in r.data
return user
2020-11-15 19:34:13 +01:00
2022-04-18 11:55:14 +02:00
def random_domain() -> str:
return random_token() + ".test"
2022-03-18 15:44:07 +01:00
def random_token(length: int = 10) -> str:
return "".join(random.choices(string.ascii_lowercase + string.digits, k=length))
2020-11-15 19:34:13 +01:00
def pretty(d):
"""pretty print as json"""
print(json.dumps(d, indent=2))
2022-03-17 19:03:36 +01:00
2022-03-18 15:55:59 +01:00
def load_eml_file(
filename: str, template_values: Optional[Dict[str, str]] = None
) -> EmailMessage:
2022-03-17 21:36:25 +01:00
emails_dir = os.path.join(
os.path.dirname(os.path.realpath(__file__)), "example_emls"
)
2022-03-17 19:03:36 +01:00
fullpath = os.path.join(emails_dir, filename)
2022-03-21 12:03:11 +01:00
with open(fullpath) as fd:
template = jinja2.Template(fd.read())
if not template_values:
template_values = {}
rendered = template.render(**template_values)
return email.message_from_bytes(rendered.encode("utf-8"))
def random_email() -> str:
return "{rand}@{rand}.com".format(rand=random_string(20))