hydrus-ptr-client/tests/endpoints.rs

use futures_util::StreamExt;

mod common;

#[tokio::test]
async fn test_options() {
    let client = common::get_client();
    client.get_options().await.unwrap();
}

#[tokio::test]
async fn test_metadata() {
    let client = common::get_client();
    let metadata = client.get_metadata(0).await.unwrap();
    assert!(metadata.update_hashes().len() > 0);
}

const DEFINITIONS_UPDATE_HASH: &str =
    "4a4d13c1fcdf0cf734927ec4c9637fdac6144512ad7dc919e0f222e7b0e71586";
const CONTENT_UPDATE_HASH: &str =
    "cd1418ffeba0b8fe46aefa51a7adf1210356523ead658b182762ff61b73ebae5";

#[tokio::test]
async fn test_update() {
    let client = common::get_client();

    client.get_update(DEFINITIONS_UPDATE_HASH).await.unwrap();
    client.get_update(CONTENT_UPDATE_HASH).await.unwrap();
}

#[tokio::test]
async fn test_update_stream() {
    let client = common::get_client(); // 3230
    let mut update_stream = client.stream_updates(0).await.unwrap();
    let mut retry_count = 3;
    let mut test_count = 25;

    while let Some(update) = update_stream.next().await {
        test_count -= 1;
        if let Err(e) = update {
            if retry_count > 0 {
                retry_count -= 1;
                update_stream.retry_latest();
            } else {
                assert!(
                    false,
                    "fetching next update failed within retry limit: {}",
                    e
                )
            }
        } else {
            retry_count = 3;
            if test_count <= 0 {
                break;
            }
        }
    }
}