init for jwt identity (#147)

tests/test_jwt_identity.py

@@ -0,0 +1,56 @@
+import pytest
+from aiohttp import web
+from aiohttp_security import AbstractAuthorizationPolicy
+from aiohttp_security import setup as _setup
+from aiohttp_security.jwt_identity import JWTIdentityPolicy
+from aiohttp_security.api import IDENTITY_KEY
+import jwt
+
+
+class Autz(AbstractAuthorizationPolicy):
+
+    async def permits(self, identity, permission, context=None):
+        pass
+
+    async def authorized_userid(self, identity):
+        pass
+
+
+async def test_no_pyjwt_installed(mocker):
+    mocker.patch('aiohttp_security.jwt_identity.jwt', None)
+    with pytest.raises(RuntimeError):
+        JWTIdentityPolicy('secret')
+
+
+async def test_identify(loop, test_client):
+    kwt_secret_key = 'Key'
+
+    async def create(request):
+        response = web.Response()
+        data = await request.post()
+
+        encoded_identity = jwt.encode({'identity': data['login']},
+                                      kwt_secret_key,
+                                      algorithm='HS256')
+
+        response.text = encoded_identity.decode('utf-8')
+        return response
+
+    async def check(request):
+        policy = request.app[IDENTITY_KEY]
+        user_id = await policy.identify(request)
+        assert 'Andrew' == user_id
+        return web.Response()
+
+    app = web.Application(loop=loop)
+    _setup(app, JWTIdentityPolicy(kwt_secret_key), Autz())
+    app.router.add_route('GET', '/', check)
+    app.router.add_route('POST', '/', create)
+    client = await test_client(app)
+    resp = await client.post('/', data={'login': 'Andrew'})
+    jwt_token = await resp.content.read()
+    assert 200 == resp.status
+    await resp.release()
+    headers = {'Authorization': str(jwt_token.decode('utf-8'))}
+    resp = await client.get('/', headers=headers)
+    assert 200 == resp.status