Move site directory init check to app creation

QuLogic · QuLogic · commit a52631a1e6cd · 2022-03-09T20:12:40.000-05:00
diff --git a/test_webhook.py b/test_webhook.py
@@ -65,8 +65,9 @@ async def test_update_repo(tmp_path_factory):
     assert dest_commit == src_commit
 
 
-async def test_github_webhook_errors(aiohttp_client, monkeypatch):
+async def test_github_webhook_errors(aiohttp_client, monkeypatch, tmp_path):
     """Test invalid inputs to webhook."""
+    monkeypatch.setenv('SITE_DIR', str(tmp_path))
     client = await aiohttp_client(create_app())
 
     # Only /gh/<repo-name> exists.
@@ -142,12 +143,12 @@ async def test_github_webhook_errors(aiohttp_client, monkeypatch):
 
 async def test_github_webhook_valid(aiohttp_client, monkeypatch, tmp_path):
     """Test valid input to webhook."""
+    monkeypatch.setenv('SITE_DIR', str(tmp_path))
     client = await aiohttp_client(create_app())
 
     # Do no actual work, since that's tested above.
     monkeypatch.setattr(webhook, 'verify_signature',
                         mock.Mock(verify_signature, return_value=True))
-    monkeypatch.setenv('SITE_DIR', str(tmp_path))
     ur_mock = mock.Mock(update_repo, return_value=None)
     monkeypatch.setattr(webhook, 'update_repo', ur_mock)
 
diff --git a/webhook.py b/webhook.py
@@ -156,9 +156,8 @@ async def github_webhook(request: web.Request):
                  delivery, ref, expected_branch)
         return web.Response(status=200)
 
-    site_dir = Path(os.environ.get('SITE_DIR', 'sites')).resolve()
-    checkout = (site_dir / repository).resolve()
-    if not checkout.is_relative_to(site_dir):
+    checkout = (request.app['site_dir'] / repository).resolve()
+    if not checkout.is_relative_to(request.app['site_dir']):
         raise web.HTTPBadRequest(
             reason=(f'{delivery}: Checkout for {organization}/{repository} '
                     'does not exist'))
@@ -176,16 +175,18 @@ async def github_webhook(request: web.Request):
 
 def create_app():
     """Create the aiohttp app and setup routes."""
+    site_dir = Path(os.environ.get('SITE_DIR', 'sites')).resolve()
+    assert site_dir.is_dir()
+
     app = web.Application()
+    app['site_dir'] = site_dir
     app.add_routes([
         web.post('/gh/{repo}', github_webhook),
     ])
     return app
 
 
 if __name__ == '__main__':
-    assert Path(os.environ.get('SITE_DIR', 'sites')).is_dir()
-
     if len(sys.argv) > 1:
         from urllib.parse import urlparse
         url = sys.argv[1]
