Merge pull request #393 from ZlatyChlapec/aiohttp-ssl-context-support

ext.aiohttp: Add option for SSL

Author: wwwjfy

gino/ext/aiohttp.py

@@ -82,6 +82,7 @@ class Gino(_Gino):
     * ``pool_min_size`` - the initial number of connections of the db pool.
     * ``pool_max_size`` - the maximum number of connections in the db pool.
     * ``echo`` - enable SQLAlchemy echo mode.
+    * ``ssl`` - SSL context passed to ``asyncpg.connect``, default is ``None``.
 
     If the ``db`` is set as an aiohttp middleware, then a lazy connection is
     available at ``request['connection']``. By default, a database connection
@@ -129,6 +130,7 @@ async def before_server_start(app_):
                 echo=config.setdefault('echo', False),
                 min_size=config.setdefault('pool_min_size', 5),
                 max_size=config.setdefault('pool_max_size', 10),
+                ssl=config.setdefault("ssl"),
                 loop=app_.loop,
             )
 

tests/test_aiohttp.py

@@ -85,6 +85,18 @@ async def test_client_dsn():
     await _test_client(dict(dsn=PG_URL))
 
 
+@pytest.fixture
+@async_generator
+async def test_client_ssl():
+    import ssl
+
+    ctx = ssl.create_default_context()
+    ctx.check_hostname = False
+    ctx.verify_mode = ssl.CERT_NONE
+
+    await _test_client(dict(dsn=PG_URL, ssl=ctx))
+
+
 async def _test_index_returns_200(test_client):
     response = await test_client.get('/')
     assert response.status == 200
@@ -122,3 +134,7 @@ async def test(test_client):
 
 async def test_dsn(test_client_dsn):
     await _test(test_client_dsn)
+
+
+async def test_ssl(test_client_ssl):
+    await _test(test_client_ssl)