Update test_runserver_main.py

Author: Dreamsorcerer

tests/test_runserver_main.py

@@ -1,5 +1,6 @@
 import asyncio
 import json
+import ssl
 from unittest import mock
 
 import aiohttp
@@ -14,7 +15,6 @@
 from aiohttp_devtools.runserver.serve import (
     WS, create_auxiliary_app, create_main_app, modify_main_app, src_reload, start_main_app)
 from aiohttp_devtools.runserver.watch import AppTask
-import ssl
 
 from .conftest import SIMPLE_APP, forked, linux_forked
 
@@ -314,12 +314,12 @@ async def test_websocket_reload(aux_cli):
 async def check_ssl_server_running(check_callback):
     port_open = False
     ssl_context = ssl.create_default_context()
-    ssl_context.load_verify_locations('test_certs/rootCA.pem')
+    ssl_context.load_verify_locations("test_certs/rootCA.pem")
 
     async with aiohttp.ClientSession(timeout=ClientTimeout(total=1)) as session:
         for i in range(50):  # pragma: no branch
             try:
-                async with session.get('https://localhost:8000/', ssl=ssl_context):
+                async with session.get("https://localhost:8000/", ssl=ssl_context):
                     pass
             except OSError:
                 await asyncio.sleep(0.1)
@@ -333,35 +333,35 @@ async def check_ssl_server_running(check_callback):
 
 @pytest.mark.filterwarnings(r"ignore:unclosed:ResourceWarning")
 @linux_forked
-@pytest.mark.datafiles('tests/test_certs', keep_top_dir=True)
+@pytest.mark.datafiles("tests/test_certs", keep_top_dir=True)
 def test_start_runserver_ssl(datafiles, tmpworkdir, smart_caplog):
     mktree(tmpworkdir, {
-        'app.py': """\
+        "app.py": """\
 from aiohttp import web
 import ssl
 async def hello(request):
-    return web.Response(text='<h1>hello world</h1>', content_type='text/html')
+    return web.Response(text="<h1>hello world</h1>", content_type="text/html")
 
 async def has_error(request):
     raise ValueError()
 
 def create_app():
     app = web.Application()
-    app.router.add_get('/', hello)
-    app.router.add_get('/error', has_error)
+    app.router.add_get("/", hello)
+    app.router.add_get("/error", has_error)
     return app
 
 def get_ssl_context():
     ssl_context = ssl.create_default_context(ssl.Purpose.CLIENT_AUTH)
-    ssl_context.load_cert_chain('test_certs/server.crt', 'test_certs/server.key')
+    ssl_context.load_cert_chain("test_certs/server.crt", "test_certs/server.key")
     return ssl_context
     """,
-        'static_dir/foo.js': 'var bar=1;',
+        "static_dir/foo.js": "var bar=1;",
     })
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
     args = runserver(app_path="app.py", static_path="static_dir",
-                     bind_address="0.0.0.0", ssl_context_factory_name='get_ssl_context')
+                     bind_address="0.0.0.0", ssl_context_factory_name="get_ssl_context")
     aux_app = args["app"]
     aux_port = args["port"]
     runapp_host = args["host"]
@@ -373,17 +373,17 @@ def get_ssl_context():
 
     async def check_callback(session, ssl_context):
         print(session, ssl_context)
-        async with session.get('https://localhost:8000/', ssl=ssl_context) as r:
+        async with session.get("https://localhost:8000/", ssl=ssl_context) as r:
             assert r.status == 200
-            assert r.headers['content-type'].startswith('text/html')
+            assert r.headers["content-type"].startswith("text/html")
             text = await r.text()
             print(text)
-            assert '<h1>hello world</h1>' in text
+            assert "<h1>hello world</h1>" in text
             assert '<script src="http://localhost:8001/livereload.js"></script>' in text
 
-        async with session.get('https://localhost:8000/error', ssl=ssl_context) as r:
+        async with session.get("https://localhost:8000/error", ssl=ssl_context) as r:
             assert r.status == 500
-            assert 'raise ValueError()' in (await r.text())
+            assert "raise ValueError()" in (await r.text())
 
     try:
         loop.run_until_complete(check_ssl_server_running(check_callback))
@@ -392,8 +392,8 @@ async def check_callback(session, ssl_context):
             loop.run_until_complete(shutdown(aux_app))
         loop.run_until_complete(aux_app.cleanup())
     assert (
-        'adev.server.dft INFO: Starting aux server at http://localhost:8001 ◆\n'
-        'adev.server.dft INFO: serving static files from ./static_dir/ at http://localhost:8001/static/\n'
-        'adev.server.dft INFO: Starting dev server at https://localhost:8000 ●\n'
+        "adev.server.dft INFO: Starting aux server at http://localhost:8001 ◆\n"
+        "adev.server.dft INFO: serving static files from ./static_dir/ at http://localhost:8001/static/\n"
+        "adev.server.dft INFO: Starting dev server at https://localhost:8000 ●\n"
     ) in smart_caplog
     loop.run_until_complete(asyncio.sleep(.25))  # TODO(aiohttp 4): Remove this hack