Skip to content

Commit a30d6c6

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 962c683 commit a30d6c6

File tree

1 file changed

+33
-51
lines changed

1 file changed

+33
-51
lines changed

integration_tests/base_routes.py

+33-51
Original file line numberDiff line numberDiff line change
@@ -1084,132 +1084,114 @@ def create_item(request, body: CreateItemBody, query: CreateItemQueryParamsParam
10841084

10851085
# --- Streaming responses ---
10861086

1087+
10871088
@app.get("/stream/sync")
10881089
def sync_stream():
10891090
def number_generator():
10901091
for i in range(5):
10911092
yield f"Chunk {i}\n".encode()
1092-
1093-
return Response(
1094-
status_code=200,
1095-
headers={"Content-Type": "text/plain"},
1096-
description=number_generator()
1097-
)
1093+
1094+
return Response(status_code=200, headers={"Content-Type": "text/plain"}, description=number_generator())
1095+
10981096

10991097
@app.get("/stream/async")
11001098
async def async_stream():
11011099
async def async_generator():
11021100
import asyncio
1101+
11031102
for i in range(5):
11041103
await asyncio.sleep(1) # Simulate async work
11051104
yield f"Async Chunk {i}\n".encode()
1106-
1107-
return Response(
1108-
status_code=200,
1109-
headers={"Content-Type": "text/plain"},
1110-
description=async_generator()
1111-
)
1105+
1106+
return Response(status_code=200, headers={"Content-Type": "text/plain"}, description=async_generator())
1107+
11121108

11131109
@app.get("/stream/mixed")
11141110
async def mixed_stream():
11151111
async def mixed_generator():
11161112
import asyncio
1113+
11171114
# Binary data
11181115
yield b"Binary chunk\n"
11191116
await asyncio.sleep(0.5)
1120-
1117+
11211118
# String data
11221119
yield "String chunk\n".encode()
11231120
await asyncio.sleep(0.5)
1124-
1121+
11251122
# Integer data
11261123
yield str(42).encode() + b"\n"
11271124
await asyncio.sleep(0.5)
1128-
1125+
11291126
# JSON data
11301127
import json
1128+
11311129
data = {"message": "JSON chunk", "number": 123}
11321130
yield json.dumps(data).encode() + b"\n"
1133-
1134-
return Response(
1135-
status_code=200,
1136-
headers={"Content-Type": "text/plain"},
1137-
description=mixed_generator()
1138-
)
1131+
1132+
return Response(status_code=200, headers={"Content-Type": "text/plain"}, description=mixed_generator())
1133+
11391134

11401135
@app.get("/stream/events")
11411136
async def server_sent_events():
11421137
async def event_generator():
11431138
import asyncio
11441139
import json
11451140
import time
1146-
1141+
11471142
# Regular event
11481143
yield f"event: message\ndata: {json.dumps({'time': time.time(), 'type': 'start'})}\n\n".encode()
11491144
await asyncio.sleep(1)
1150-
1145+
11511146
# Event with ID
11521147
yield f"id: 1\nevent: update\ndata: {json.dumps({'progress': 50})}\n\n".encode()
11531148
await asyncio.sleep(1)
1154-
1149+
11551150
# Multiple data lines
1156-
data = json.dumps({'status': 'complete', 'results': [1, 2, 3]}, indent=2)
1151+
data = json.dumps({"status": "complete", "results": [1, 2, 3]}, indent=2)
11571152
yield f"event: complete\ndata: {data}\n\n".encode()
1158-
1153+
11591154
return Response(
1160-
status_code=200,
1161-
headers={
1162-
"Content-Type": "text/event-stream",
1163-
"Cache-Control": "no-cache",
1164-
"Connection": "keep-alive"
1165-
},
1166-
description=event_generator()
1155+
status_code=200, headers={"Content-Type": "text/event-stream", "Cache-Control": "no-cache", "Connection": "keep-alive"}, description=event_generator()
11671156
)
11681157

1158+
11691159
@app.get("/stream/large-file")
11701160
async def stream_large_file():
11711161
async def file_generator():
11721162
# Simulate streaming a large file in chunks
11731163
chunk_size = 1024 # 1KB chunks
11741164
total_size = 10 * chunk_size # 10KB total
1175-
1165+
11761166
for offset in range(0, total_size, chunk_size):
11771167
# Simulate reading file chunk
11781168
chunk = b"X" * min(chunk_size, total_size - offset)
11791169
yield chunk
1180-
1170+
11811171
return Response(
11821172
status_code=200,
1183-
headers={
1184-
"Content-Type": "application/octet-stream",
1185-
"Content-Disposition": "attachment; filename=large-file.bin"
1186-
},
1187-
description=file_generator()
1173+
headers={"Content-Type": "application/octet-stream", "Content-Disposition": "attachment; filename=large-file.bin"},
1174+
description=file_generator(),
11881175
)
11891176

1177+
11901178
@app.get("/stream/csv")
11911179
async def stream_csv():
11921180
async def csv_generator():
11931181
# CSV header
11941182
yield "id,name,value\n".encode()
1195-
1183+
11961184
import asyncio
11971185
import random
1198-
1186+
11991187
# Generate rows
12001188
for i in range(5):
12011189
await asyncio.sleep(0.5) # Simulate data processing
12021190
row = f"{i},item-{i},{random.randint(1, 100)}\n"
12031191
yield row.encode()
1204-
1205-
return Response(
1206-
status_code=200,
1207-
headers={
1208-
"Content-Type": "text/csv",
1209-
"Content-Disposition": "attachment; filename=data.csv"
1210-
},
1211-
description=csv_generator()
1212-
)
1192+
1193+
return Response(status_code=200, headers={"Content-Type": "text/csv", "Content-Disposition": "attachment; filename=data.csv"}, description=csv_generator())
1194+
12131195

12141196
def main():
12151197
app.set_response_header("server", "robyn")

0 commit comments

Comments
 (0)