1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
import bottle
import os
import pytest
from cork import Cork
@pytest.fixture
def chdir_to_tmpdir(tmpdir):
# Chdir into the current tmpdir (used by Cork to find email templates)
tmpdir.chdir()
assert tmpdir.strpath == os.getcwd()
@pytest.fixture
def templates_dir(tmpdir, chdir_to_tmpdir):
# Setup email templates
tmpdir.mkdir('views')
tmpdir.join('views/registration_email.tpl').write("""Username:{{username}} Email:{{email_addr}} Code:{{registration_code}}""")
tmpdir.join('views/password_reset_email.tpl').write("""Username:{{username}} Email:{{email_addr}} Code:{{reset_code}}""")
assert tmpdir.join('views/password_reset_email.tpl').exists()
tmpdir.mkdir('examples')
tmpdir.mkdir('examples/views')
tmpdir.join('examples/views/password_reset_email.tpl').write(
"""Hello {{username}},<br/>
You are receiving this email because you requested a password reset on Cork Demo Webapp.<br/>
<br/>
If you wish to complete the password reset please click on:<br/>
<a href="http://localhost:8080/change_password/{{reset_code}}">Confirm</a>
<br/><br/>
""")
return tmpdir
@pytest.fixture
def mytmpdir(tmpdir):
"""Setup tmp directory with test files
"""
tmpdir.mkdir('views')
tmpdir.join('users.json').write("""{"admin": {"email_addr": null, "desc": null, "role": "admin", "hash": "69f75f38ac3bfd6ac813794f3d8c47acc867adb10b806e8979316ddbf6113999b6052efe4ba95c0fa9f6a568bddf60e8e5572d9254dbf3d533085e9153265623", "creation_date": "2012-04-09 14:22:27.075596"}}""")
tmpdir.join('roles.json').write("""{"special": 200, "admin": 100, "user": 50}""")
tmpdir.join('register.json').write("""{}""")
tmpdir.join('registration_email.tpl').write("""Username:{{username}} Email:{{email_addr}} Code:{{registration_code}}""")
tmpdir.join('password_reset_email.tpl').write("""Username:{{username}} Email:{{email_addr}} Code:{{reset_code}}""")
return tmpdir
# used by test_scrypt.py
@pytest.fixture
def aaa(mytmpdir):
aaa = Cork(mytmpdir, smtp_server='localhost', email_sender='test@localhost')
return aaa
# mb = self.setup_test_db()
# self.aaa = MockedUnauthenticatedCork(backend=mb,
# smtp_server='localhost', email_sender='test@localhost')
# cookie_name = None
# if hasattr(self, 'purge_test_db'):
# self.purge_test_db()
#
# del(self.aaa)
# cookie_name = None
def assert_is_redirect(e, path):
"""Check if an HTTPResponse is a redirect.
:param path: relative path without leading slash.
:type path: str
"""
assert isinstance(e, bottle.HTTPResponse), "Incorrect exception type passed to assert_is_redirect"
assert e.status_code == 302, "HTTPResponse status should be 302 but is '%s'" % e.status
redir_location = e.headers['Location'].rsplit('/', 1)[1]
assert redir_location == path, "Redirected to %s instead of %s" % (redir_location, path)
|
