app-MAIL-temp/tests/utils.py

55 lines
1.1 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
from email.message import EmailMessage
2020-11-15 19:34:13 +01:00
from flask import url_for
from app.models import User
def login(flask_client) -> User:
# create user, user is activated
user = User.create(
2020-12-31 13:59:03 +01:00
email="a@b.c",
password="password",
name="Test User",
activated=True,
commit=True,
)
r = flask_client.post(
url_for("auth.login"),
data={"email": "a@b.c", "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
def create_user(flask_client) -> User:
# create user, user is activated
return User.create(
email="a@b.c",
password="password",
name="Test User",
activated=True,
commit=True,
)
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
def load_eml_file(filename: str) -> 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)
return email.message_from_file(open(fullpath))