trustchain-oc1-orchestral
/
authentik
Archived
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

stages/email: add tests for inaccessible email templates 

Signed-off-by: Jens Langhammer <jens.langhammer@beryju.org>
This commit is contained in:
Jens Langhammer 2021-06-07 16:09:39 +02:00
parent e51226432f
commit 0f693158b6
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
authentik/stages/email/tests/test_templates.py
View File

@ -1,5 +1,5 @@
"""email tests""" """email tests"""
from os import unlink from os import chmod, unlink
from pathlib import Path from pathlib import Path
from tempfile import gettempdir, mkstemp from tempfile import gettempdir, mkstemp
from typing import Any from typing import Any
@ -24,5 +24,10 @@ class TestEmailStageTemplates(TestCase):
"""Test with custom template""" """Test with custom template"""
with self.settings(TEMPLATES=get_templates_setting(gettempdir())): with self.settings(TEMPLATES=get_templates_setting(gettempdir())):
_, file = mkstemp(suffix=".html") _, file = mkstemp(suffix=".html")
self.assertEqual(get_template_choices()[-1][0], Path(file).name) _, file2 = mkstemp(suffix=".html")
chmod(file2, 0o000) # Remove all permissions so we can't read the file
choices = get_template_choices()
self.assertEqual(choices[-1][0], Path(file).name)
self.assertEqual(len(choices), 3)
unlink(file) unlink(file)
unlink(file2)