diff --git a/src/test/test_two_chainz.py b/src/test/test_two_chainz.py index 3b7dc77..772d788 100644 --- a/src/test/test_two_chainz.py +++ b/src/test/test_two_chainz.py @@ -6,7 +6,7 @@ from unittest.mock import patch from two_chainz import app import two_chainz -client = TestClient(app) +client = TestClient(app, base_url="http://chainz.ansonbiggs.com") class TestApi: diff --git a/src/two_chainz/__init__.py b/src/two_chainz/__init__.py index eae3aba..3e1ae66 100644 --- a/src/two_chainz/__init__.py +++ b/src/two_chainz/__init__.py @@ -5,8 +5,14 @@ from fastapi.staticfiles import StaticFiles import tomllib from pathlib import Path import random +from fastapi.middleware.trustedhost import TrustedHostMiddleware + app = FastAPI() +app.add_middleware( + TrustedHostMiddleware, allowed_hosts=["ansonbiggs.com", "*.ansonbiggs.com"] +) + start_time = time.time()