(server: LightningStoreServer)
| 105 | |
| 106 | |
| 107 | async def _run_server_side_operations(server: LightningStoreServer) -> None: |
| 108 | await server.update_resources("metrics-server", {}) |
| 109 | await server.get_latest_resources() |
| 110 | started = await server.start_rollout(input={"origin": "server"}, config=RolloutConfig(timeout_seconds=1.0)) |
| 111 | queued = await server.enqueue_rollout(input={"origin": "server-queue"}) |
| 112 | dequeued = await server.dequeue_rollout(worker_id="metrics-server-worker") |
| 113 | assert dequeued is not None |
| 114 | |
| 115 | await server.add_span(_make_span(dequeued.rollout_id, dequeued.attempt.attempt_id, 0, "server-span")) |
| 116 | await server.update_attempt(queued.rollout_id, dequeued.attempt.attempt_id, status="running") |
| 117 | await server.update_attempt(queued.rollout_id, dequeued.attempt.attempt_id, status="succeeded") |
| 118 | await server.update_rollout(queued.rollout_id, status="succeeded") |
| 119 | await server.wait_for_rollouts(rollout_ids=[queued.rollout_id], timeout=0.1) |
| 120 | assert started is not None |
| 121 | |
| 122 | |
| 123 | async def _run_client_side_operations(client: LightningStoreClient) -> None: |
no test coverage detected