durable-actors 0.12.2

Standalone regional durable-actors control plane, host, and durability runtime
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use super::*;
use crate::host::HostId;

#[tokio::test]
async fn assignment_is_authenticated_single_use_and_waits_for_readiness() -> anyhow::Result<()> {
    for with_operation in [false, true] {
        let (send, receive) = tokio::sync::oneshot::channel();
        let listener = tokio::net::TcpListener::bind("127.0.0.1:0").await?;
        let url = format!("http://{}/assign", listener.local_addr()?);
        let server = tokio::spawn(async move {
            axum::serve(listener, router("secret".into(), send))
                .await
                .unwrap();
        });
        let client = reqwest::Client::new();
        let health = client
            .get(url.replace("/assign", "/healthz"))
            .send()
            .await?;
        assert_eq!(health.status(), 200);
        let response = client
            .post(&url)
            .json(&serde_json::json!({}))
            .send()
            .await?;
        assert_eq!(response.status(), 401);
        let actor =
            serde_json::json!({"project_id":"project", "actor_name":"Counter", "actor_id":"one"});
        let environment =
            serde_json::json!({"actor":"one", "DURABLE_ACTORS_ACTOR":actor.to_string()});
        let body = if with_operation {
            let mut invalid = serde_json::json!({"environment":environment, "operation":{
                "routingMs":1.0, "request":{"type":"invoke", "invocation":{
                    "actor":actor, "request_id":"first", "method":"read", "args":[]
                }, "grant":null}
            }});
            invalid["operation"]["request"]["invocation"]["actor"]["actor_id"] = "other".into();
            assert_eq!(
                client
                    .post(&url)
                    .bearer_auth("secret")
                    .json(&invalid)
                    .send()
                    .await?
                    .status(),
                400
            );
            invalid["operation"]["request"]["invocation"]["actor"]["actor_id"] = "one".into();
            invalid
        } else {
            environment
        };
        let request = client.post(&url).bearer_auth("secret").json(&body);
        let call = tokio::spawn(async move { request.send().await });
        let assigned = tokio::time::timeout(std::time::Duration::from_secs(1), receive).await??;
        assert_eq!(assigned.environment["actor"], "one");
        assert_eq!(assigned.operation.is_some(), with_operation);
        assert!(
            !call.is_finished(),
            "assignment must wait for hydration and ownership"
        );
        let duplicate = client
            .post(&url)
            .bearer_auth("secret")
            .json(&serde_json::json!({}))
            .send()
            .await?;
        assert_eq!(duplicate.status(), 409);
        assert!(
            assigned
                .ready
                .send(Ok(super::super::process::HostReadiness {
                    initial_reply: with_operation
                        .then(|| serde_json::json!({"type":"completed", "result":7})),
                    host_id: HostId::new("host.v3.test.one"),
                    session_id: "session".into(),
                    route: "https://host.test".into(),
                    canonical_region: "north-america-east".into(),
                    owner_epoch: 42,
                    lease: crate::host_leases::HostLease {
                        id: HostId::new("host.v3.test.one"),
                        session_id: "session".into(),
                        route: "https://host.test".into(),
                        expires_at_ms: 60_000,
                    },
                }))
                .is_ok()
        );
        let reply = call.await??;
        assert_eq!(reply.status(), 200);
        let reply = reply.json::<serde_json::Value>().await?;
        assert_eq!(reply["ownerEpoch"], 42);
        if with_operation {
            assert_eq!(reply["initialReply"]["result"], 7);
        }
        server.abort();
    }
    Ok(())
}

#[tokio::test]
async fn failed_initialization_never_reports_ready() -> anyhow::Result<()> {
    let (send, receive) = tokio::sync::oneshot::channel();
    let listener = tokio::net::TcpListener::bind("127.0.0.1:0").await?;
    let url = format!("http://{}/assign", listener.local_addr()?);
    let server = tokio::spawn(async move {
        axum::serve(listener, router("secret".into(), send))
            .await
            .unwrap();
    });
    let request = reqwest::Client::new()
        .post(url)
        .bearer_auth("secret")
        .json(&serde_json::json!({}));
    let call = tokio::spawn(async move { request.send().await });
    drop(receive.await?);
    assert_eq!(call.await??.status(), 503);
    server.abort();
    Ok(())
}