from starlette.applications import Starlette
+from starlette.middleware import Middleware
from starlette.middleware.gzip import GZipMiddleware
from starlette.responses import PlainTextResponse, StreamingResponse
+from starlette.routing import Route
def test_gzip_responses(test_client_factory):
- app = Starlette()
-
- app.add_middleware(GZipMiddleware)
-
- @app.route("/")
def homepage(request):
return PlainTextResponse("x" * 4000, status_code=200)
+ app = Starlette(
+ routes=[Route("/", endpoint=homepage)],
+ middleware=[Middleware(GZipMiddleware)],
+ )
+
client = test_client_factory(app)
response = client.get("/", headers={"accept-encoding": "gzip"})
assert response.status_code == 200
def test_gzip_not_in_accept_encoding(test_client_factory):
- app = Starlette()
-
- app.add_middleware(GZipMiddleware)
-
- @app.route("/")
def homepage(request):
return PlainTextResponse("x" * 4000, status_code=200)
+ app = Starlette(
+ routes=[Route("/", endpoint=homepage)],
+ middleware=[Middleware(GZipMiddleware)],
+ )
+
client = test_client_factory(app)
response = client.get("/", headers={"accept-encoding": "identity"})
assert response.status_code == 200
def test_gzip_ignored_for_small_responses(test_client_factory):
- app = Starlette()
-
- app.add_middleware(GZipMiddleware)
-
- @app.route("/")
def homepage(request):
return PlainTextResponse("OK", status_code=200)
+ app = Starlette(
+ routes=[Route("/", endpoint=homepage)],
+ middleware=[Middleware(GZipMiddleware)],
+ )
+
client = test_client_factory(app)
response = client.get("/", headers={"accept-encoding": "gzip"})
assert response.status_code == 200
def test_gzip_streaming_response(test_client_factory):
- app = Starlette()
-
- app.add_middleware(GZipMiddleware)
-
- @app.route("/")
def homepage(request):
async def generator(bytes, count):
for index in range(count):
streaming = generator(bytes=b"x" * 400, count=10)
return StreamingResponse(streaming, status_code=200)
+ app = Starlette(
+ routes=[Route("/", endpoint=homepage)],
+ middleware=[Middleware(GZipMiddleware)],
+ )
+
client = test_client_factory(app)
response = client.get("/", headers={"accept-encoding": "gzip"})
assert response.status_code == 200