Skip to content

Commit 8b75aff

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

File tree

10 files changed

+364
-375
lines changed

10 files changed

+364
-375
lines changed

integration_tests/base_routes.py

Lines changed: 21 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1060,91 +1060,105 @@ def create_item(request, body: CreateItemBody, query: CreateItemQueryParamsParam
10601060

10611061
# ===== Server-Sent Events (SSE) Routes =====
10621062

1063+
10631064
@app.get("/sse/basic")
10641065
def sse_basic(request):
10651066
"""Basic SSE endpoint that sends 3 messages"""
1067+
10661068
def event_generator():
10671069
for i in range(3):
10681070
yield f"data: Test message {i}\n\n"
1071+
10691072
return sse_response(event_generator())
10701073

10711074

10721075
@app.get("/sse/formatted")
10731076
def sse_formatted(request):
10741077
"""SSE endpoint using sse_message formatter"""
1078+
10751079
def event_generator():
10761080
for i in range(3):
10771081
yield sse_message(f"Formatted message {i}", event="test", id=str(i))
1082+
10781083
return sse_response(event_generator())
10791084

10801085

10811086
@app.get("/sse/json")
10821087
def sse_json(request):
10831088
"""SSE endpoint that sends JSON data"""
10841089
import json
1090+
10851091
def event_generator():
10861092
for i in range(3):
10871093
data = {"id": i, "message": f"JSON message {i}", "type": "test"}
10881094
yield f"data: {json.dumps(data)}\n\n"
1095+
10891096
return sse_response(event_generator())
10901097

10911098

10921099
@app.get("/sse/named_events")
10931100
def sse_named_events(request):
10941101
"""SSE endpoint with different event types"""
1102+
10951103
def event_generator():
1096-
events = [
1097-
("start", "Test started"),
1098-
("progress", "Test in progress"),
1099-
("end", "Test completed")
1100-
]
1104+
events = [("start", "Test started"), ("progress", "Test in progress"), ("end", "Test completed")]
11011105
for event_type, message in events:
11021106
yield sse_message(message, event=event_type)
1107+
11031108
return sse_response(event_generator())
11041109

11051110

11061111
@app.get("/sse/async")
11071112
def sse_async(request):
11081113
"""Async SSE endpoint (sync implementation)"""
1114+
11091115
def event_generator():
11101116
# Simple generator for testing
11111117
for i in range(3):
11121118
yield f"data: Async message {i}\n\n"
1113-
1119+
11141120
return sse_response(event_generator())
11151121

11161122

11171123
@app.get("/sse/single")
11181124
def sse_single(request):
11191125
"""SSE endpoint that sends a single message and closes"""
1126+
11201127
def event_generator():
11211128
yield "data: Single message\n\n"
1129+
11221130
return sse_response(event_generator())
11231131

11241132

11251133
@app.get("/sse/empty")
11261134
def sse_empty(request):
11271135
"""SSE endpoint that sends no messages"""
1136+
11281137
def event_generator():
11291138
return
11301139
yield # This will never be reached
1140+
11311141
return sse_response(event_generator())
11321142

11331143

11341144
@app.get("/sse/with_headers")
11351145
def sse_with_headers(request):
11361146
"""SSE endpoint with custom headers"""
11371147
headers = Headers({"X-Custom-Header": "custom-value"})
1148+
11381149
def event_generator():
11391150
yield "data: Message with custom headers\n\n"
1151+
11401152
return sse_response(event_generator(), headers=headers)
11411153

11421154

11431155
@app.get("/sse/status_code")
11441156
def sse_status_code(request):
11451157
"""SSE endpoint with custom status code"""
1158+
11461159
def event_generator():
11471160
yield "data: Message with custom status\n\n"
1161+
11481162
return sse_response(event_generator(), status_code=201)
11491163

11501164

@@ -1170,7 +1184,7 @@ def authenticate(self, request: Request) -> Optional[Identity]:
11701184
return None
11711185

11721186
app.configure_authentication(BasicAuthHandler(token_getter=BearerGetter()))
1173-
1187+
11741188
# Read port from environment variable if set, otherwise default to 8080
11751189
port = int(os.getenv("ROBYN_PORT", "8080"))
11761190
app.start(port=port, _check_port=False)

integration_tests/conftest.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -43,12 +43,12 @@ def start_server(domain: str, port: int, is_dev: bool = False) -> subprocess.Pop
4343
command = ["python3", base_routes]
4444
if is_dev:
4545
command.append("--dev")
46-
46+
4747
# Ensure environment variables are properly set for the subprocess
4848
env = os.environ.copy()
4949
env["ROBYN_HOST"] = domain
5050
env["ROBYN_PORT"] = str(port)
51-
51+
5252
if platform.system() == "Windows":
5353
command[0] = "python"
5454
process = subprocess.Popen(command, shell=True, creationflags=subprocess.CREATE_NEW_PROCESS_GROUP, env=env)
@@ -70,7 +70,7 @@ def start_server(domain: str, port: int, is_dev: bool = False) -> subprocess.Pop
7070
break # We were able to reach the server, exit the loop
7171
except Exception:
7272
time.sleep(0.5) # Longer delay before retrying
73-
73+
7474
# Give the server a moment to fully initialize after accepting connections
7575
time.sleep(1)
7676
return process

integration_tests/test_base_url.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,4 +28,3 @@ def test_global_index_request(global_session):
2828
res = requests.get(f"{BASE_URL}")
2929
assert os.getenv("ROBYN_HOST") == f"{host}"
3030
assert res.status_code == 200
31-

0 commit comments

Comments
 (0)