beenull/simple-login
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Fix signature

Browse source
This commit is contained in:
Adrià Casajús 2022-03-18 15:55:59 +01:00
parent 4a533bb03b
commit d53ea381a0
No known key found for this signature in database
GPG key ID: F0033226A5AFC9B9
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/utils.py
View file

@ -4,6 +4,7 @@ import os
import random import random
import string import string
from email.message import EmailMessage from email.message import EmailMessage
from typing import Optional, Dict
import jinja2 import jinja2
from flask import url_for from flask import url_for
@ -64,11 +65,15 @@ def pretty(d):
print(json.dumps(d, indent=2)) print(json.dumps(d, indent=2))
def load_eml_file(filename: str, template_values={}) -> EmailMessage: def load_eml_file(
filename: str, template_values: Optional[Dict[str, str]] = None
) -> EmailMessage:
emails_dir = os.path.join( emails_dir = os.path.join(
os.path.dirname(os.path.realpath(__file__)), "example_emls" os.path.dirname(os.path.realpath(__file__)), "example_emls"
) )
fullpath = os.path.join(emails_dir, filename) fullpath = os.path.join(emails_dir, filename)
template = jinja2.Template(open(fullpath).read()) template = jinja2.Template(open(fullpath).read())
if not template_values:
template_values = {}
rendered = template.render(**template_values) rendered = template.render(**template_values)
return email.message_from_string(rendered) return email.message_from_string(rendered)