@@ -24,59 +24,58 @@ def delete_cookie(self, key):
2424def test_signatures ():
2525 ip = IdentityPolicy ()
2626
27- assert ip .unsign (ip .sign (' admin' , ' username' ), ' username' ) == ' admin'
28- assert ip .unsign (ip .sign (' admin ' , ' username' ), ' username' ) == ' admin '
29- assert ip .unsign (ip .sign (' admin' , ' username' ), ' role' ) is None
27+ assert ip .unsign (ip .sign (" admin" , " username" ), " username" ) == " admin"
28+ assert ip .unsign (ip .sign (" admin " , " username" ), " username" ) == " admin "
29+ assert ip .unsign (ip .sign (" admin" , " username" ), " role" ) is None
3030
3131
3232def test_expired_signature ():
3333
3434 ip = IdentityPolicy (max_age = 0.1 )
35- signed = ip .sign (' admin' , ' username' )
35+ signed = ip .sign (" admin" , " username" )
3636
37- assert ip .unsign (signed , ' username' ) == ' admin'
37+ assert ip .unsign (signed , " username" ) == " admin"
3838 time .sleep (1.0 )
39- assert ip .unsign (signed , ' username' ) is None
39+ assert ip .unsign (signed , " username" ) is None
4040
4141
4242def test_policy ():
4343 ip = IdentityPolicy ()
44- identity = morepath .Identity (userid = ' aaron' , role = ' admin' )
44+ identity = morepath .Identity (userid = " aaron" , role = " admin" )
4545 request = TestRequest ()
4646 response = TestResponse ()
4747
4848 ip .remember (response , request , identity )
4949
50- assert ' userid' in response .cookies
51- assert ' role' not in response .cookies
50+ assert " userid" in response .cookies
51+ assert " role" not in response .cookies
5252
5353
5454def test_custom_policy ():
55-
5655 class CustomIdentityPolicy (IdentityPolicy ):
57- required_keys = (' userid' , ' role' )
56+ required_keys = (" userid" , " role" )
5857
5958 ip = CustomIdentityPolicy ()
60- identity = morepath .Identity (userid = ' aaron' , role = ' admin' )
59+ identity = morepath .Identity (userid = " aaron" , role = " admin" )
6160 request = TestRequest ()
6261 response = TestResponse ()
6362
6463 ip .remember (response , request , identity )
6564
66- assert response .cookies [' userid' ].startswith (b' aaron.' )
67- assert response .cookies [' role' ].startswith (b' admin.' )
68- assert response . cookie_args [ 'userid' ] == response . cookie_args [ 'role' ] == {
69- 'max_age' : 3600 ,
70- 'secure' : True ,
71- ' httponly' : True
72- }
65+ assert response .cookies [" userid" ].startswith (b" aaron." )
66+ assert response .cookies [" role" ].startswith (b" admin." )
67+ assert (
68+ response . cookie_args [ "userid" ]
69+ == response . cookie_args [ "role" ]
70+ == { "max_age" : 3600 , "secure" : True , " httponly" : True }
71+ )
7372
7473 request .cookies = response .cookies
7574
76- assert ip .identify (request ).userid == ' aaron'
77- assert ip .identify (request ).role == ' admin'
75+ assert ip .identify (request ).userid == " aaron"
76+ assert ip .identify (request ).role == " admin"
7877
79- del request .cookies [' role' ]
78+ del request .cookies [" role" ]
8079
8180 assert ip .identify (request ) is None
8281
@@ -87,11 +86,11 @@ class CustomIdentityPolicy(IdentityPolicy):
8786
8887def test_cookie_settings ():
8988 ip = IdentityPolicy ()
90- assert ip .cookie_settings [' max_age' ] == 3600
91- assert ip .cookie_settings [' secure' ]
92- assert ip .cookie_settings [' httponly' ]
89+ assert ip .cookie_settings [" max_age" ] == 3600
90+ assert ip .cookie_settings [" secure" ]
91+ assert ip .cookie_settings [" httponly" ]
9392
9493 ip = IdentityPolicy (max_age = 1 , secure = False , httponly = False )
95- assert ip .cookie_settings [' max_age' ] == 1
96- assert not ip .cookie_settings [' secure' ]
97- assert not ip .cookie_settings [' httponly' ]
94+ assert ip .cookie_settings [" max_age" ] == 1
95+ assert not ip .cookie_settings [" secure" ]
96+ assert not ip .cookie_settings [" httponly" ]
0 commit comments