Skip to content

Keep GPU bridge test independent of daily cleanup time #37

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
gaochangw merged 1 commit into from
Sep 14, 2026
Merged

Keep GPU bridge test independent of daily cleanup time #37

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 14 additions & 3 deletions tests/integration/test_gpu_bridge.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,14 +5,17 @@
import json
import time
import zipfile
from types import SimpleNamespace

import pytest
from fastapi.testclient import TestClient

from opendpd.services.recipes import instantiate
from opendpd.web import gpu_broker
from opendpd.web.app import create_web_app
from opendpd.web.gpu_archive import pack, unpack, read_regular
from opendpd.web.policy import WebConfig
from opendpd.web.runtime import DAY

pytestmark = pytest.mark.integration
TOKEN = "private-test-gpu-token-" * 4
Expand All @@ -29,9 +32,15 @@ def eventually(predicate):
raise AssertionError("condition did not become true")


def test_private_gpu_lease_stream_replay_and_cross_tenant_cancel(tmp_path):
def test_private_gpu_lease_stream_replay_and_cross_tenant_cancel(tmp_path, monkeypatch):
# Keep session and job expiry on the same fixed daytime clock. Real
# monotonic time still drives polling and the lost-lease assertion below.
def now():
return DAY * 20000 + 3600

monkeypatch.setattr(gpu_broker, "time", SimpleNamespace(time=now, monotonic=time.monotonic))
config = WebConfig(tmp_path / "web", "https://opendpd.com", "api.opendpd.com", TUNNEL, gpu_token=TOKEN)
app = create_web_app(config)
app = create_web_app(config, now=now)
with TestClient(app, base_url="http://127.0.0.1", client=("127.0.0.1", 10)) as client:
private = {"X-OpenDPD-GPU": TOKEN}
public = {"Host": TUNNEL, "Origin": "https://opendpd.com", "X-Forwarded-Proto": "https", "CF-Connecting-IP": "203.0.113.10"}
Expand All @@ -40,7 +49,9 @@ def test_private_gpu_lease_stream_replay_and_cross_tenant_cancel(tmp_path):
assert client.post("/_gpu/poll", json={"name": "CUDA test device"}, headers=private).json() == {"job": None}
auth = []
for _ in range(2):
token = client.post("/api/v1/web/sessions", json={}, headers=public).json()["access_token"]
session = client.post("/api/v1/web/sessions", json={}, headers=public)
assert session.status_code == 201, session.text
token = session.json()["access_token"]
auth.append({**public, "Authorization": "Bearer " + token})
capability = client.get("/api/v1/system/capabilities", headers=auth[0]).json()
assert next(d for d in capability["devices"] if d["device"] == "cuda")["detected"]
Expand Down
Loading