Function poll_request_completion

tests/integration/src/dfx_orbit/util.rs:18–54

Wait on a request to succeed (reach status completed)

(
    dfx_orbit: &DfxOrbit,
    request_id: UuidDTO,
    timeout: Duration,
)

/// Wait on a request to succeed (reach status completed)
pub(super) async fn poll_request_completion(
    dfx_orbit: &DfxOrbit,
    request_id: UuidDTO,
    timeout: Duration,
) {
    let timeout = Instant::now() + timeout;

    loop {
        let request = dfx_orbit
            .station
            .review_id(GetRequestInput {
                request_id: request_id.clone(),
                with_full_info: None,
            })
            .await
            .unwrap();

        match request.request.status {
            RequestStatusDTO::Completed { .. } => return,
            RequestStatusDTO::Rejected
            | RequestStatusDTO::Cancelled { .. }
            | RequestStatusDTO::Failed { .. } => {
                panic!("Expected request {request_id} to succeed")
            }
            RequestStatusDTO::Approved
            | RequestStatusDTO::Created
            | RequestStatusDTO::Scheduled { .. }
            | RequestStatusDTO::Processing { .. } => (),
        }

        if Instant::now() > timeout {
            panic!("Waiting for request {request_id} to succeed timed out");
        }

        tokio::time::sleep(Duration::from_secs(1)).await
    }
}

/// Fetches an asset from the local host and port
///