rho-coding-agent 2.30.0

A fast Rust agent harness with a small footprint and opinionated defaults
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

use std::{
    sync::{
        atomic::{AtomicBool, Ordering},
        Arc,
    },
    time::Duration,
};

use pretty_assertions::assert_eq;

use super::super::test_server::{Reply, TestServer};
use super::*;

// Covers: a downloaded spec is cached, revalidated with its ETag next session,
// and served stale with a warning when the refresh fails. Without a cache the
// failure is an error.
// Owner: OpenAPI spec cache
#[tokio::test]
async fn revalidates_with_etag_and_falls_back_to_the_cache() {
    let failing = Arc::new(AtomicBool::new(false));
    let fail = failing.clone();
    let server = TestServer::start(move |request| {
        if fail.load(Ordering::SeqCst) {
            Reply::new(500, "down")
        } else if request.headers.get("if-none-match").map(String::as_str) == Some("\"v1\"") {
            Reply::new(304, "")
        } else {
            Reply::new(200, "spec v1").header("etag", "\"v1\"")
        }
    })
    .await;
    let url = Url::parse(&server.url("/spec.json")).unwrap();
    let dir = tempfile::tempdir().unwrap();
    let cache = SpecCache::new(dir.path().join("openapi"));
    let client = crate::reqwest_client();
    let fetch = |cache| SpecFetch {
        client: &client,
        cache,
        timeout: Duration::from_secs(30),
        max_spec_bytes: 1024,
    };
    let location = SpecLocation::Url(url.clone());
    let result = |fetched: Result<FetchedSpec, String>| {
        fetched.map(|fetched| (String::from_utf8(fetched.bytes).unwrap(), fetched.warning))
    };

    // The loader caches a download only once it parses.
    let downloaded = fetch(Some(&cache)).fetch(&location).await;
    if let Ok(fetched) = &downloaded {
        cache.remember(fetched);
    }
    let fresh = result(downloaded);
    let revalidated = result(fetch(Some(&cache)).fetch(&location).await);
    failing.store(true, Ordering::SeqCst);
    let stale = result(fetch(Some(&cache)).fetch(&location).await);
    let uncached = result(fetch(None).fetch(&location).await);

    assert_eq!(
        (fresh, revalidated, stale, uncached),
        (
            Ok(("spec v1".to_owned(), None)),
            Ok(("spec v1".to_owned(), None)),
            Ok((
                "spec v1".to_owned(),
                Some(format!(
                    "could not refresh {url}: HTTP 500 Internal Server Error; using the cached copy"
                ))
            )),
            Err(format!(
                "could not download {url}: HTTP 500 Internal Server Error"
            )),
        )
    );
    let requests = server.requests();
    let validators: Vec<Option<&str>> = requests
        .iter()
        .map(|request| request.headers.get("if-none-match").map(String::as_str))
        .collect();
    assert_eq!(validators, vec![None, Some("\"v1\""), Some("\"v1\""), None]);
}