The_Great_Archive/ArchiveBox
1
0
Fork 0
mirror of https://github.com/ArchiveBox/ArchiveBox.git synced 2026-09-12 19:50:57 +05:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Keep SSE disconnect test from starting a background server

Browse Source
This commit is contained in:
Nick Sweeting 2026-09-02 22:10:42 -07:00
parent 75cf3f1261
commit ff2766c007
No known key found for this signature in database
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
archivebox/tests/test_opencode_agent.py
View File

@ -280,7 +280,8 @@ def test_opencode_proxy_sse_returns_headers_before_restart_finishes(admin_client
from archivebox.opencode import views
from django.conf import settings as django_settings
views._stop_owned_process()
owned_process = views._PROCESS
assert owned_process is not None
session_cookie_name = django_settings.SESSION_COOKIE_NAME
session_cookie = admin_client.cookies[session_cookie_name].value
path = "/admin/agent/opencode/global/event"
@ -307,17 +308,25 @@ def test_opencode_proxy_sse_returns_headers_before_restart_finishes(admin_client
},
)
views._PROCESS_LOCK.acquire()
views._PROCESS = None
try:
await communicator.send_input({"type": "http.request", "body": b"", "more_body": False})
response_start = await communicator.receive_output(timeout=2)
assert response_start["type"] == "http.response.start"
assert response_start["status"] == 200
finally:
views._PROCESS_LOCK.release()
await communicator.send_input({"type": "http.disconnect"})
await communicator.wait(timeout=5)
try:
await communicator.send_input({"type": "http.disconnect"})
await communicator.wait(timeout=5)
finally:
views._PROCESS = owned_process
views._PROCESS_LOCK.release()
asyncio.run(request_event_stream())
ok, error = views._ensure_opencode(live_opencode.settings)
assert ok, error
assert views._PROCESS is owned_process
assert owned_process.poll() is None
def test_opencode_starts_with_isolated_state(live_opencode):