aviso-server 0.4.1

Notification service for data-driven workflows with live and replay APIs.
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135

// (C) Copyright 2024- ECMWF and individual contributors.
//
// This software is licensed under the terms of the Apache Licence Version 2.0
// which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
// In applying this licence, ECMWF does not waive the privileges and immunities
// granted to it by virtue of its status as an intergovernmental organisation nor
// does it submit to any jurisdiction.

use crate::helpers::spawn_streaming_test_app;
use crate::test_utils::{post_mars_notification, unique_suffix};
use serde_json::json;

fn extract_event_sequence(replay_body: &str, event_type: &str) -> Option<u64> {
    let marker = format!("\"id\":\"{}@", event_type);
    let id_start = replay_body.find(&marker)?;
    let id_tail = &replay_body[id_start + marker.len()..];
    let id_end = id_tail.find('"')?;
    id_tail[..id_end].parse::<u64>().ok()
}

fn replay_event_count(replay_body: &str) -> usize {
    replay_body.matches("\nevent: replay\ndata: ").count()
}

#[tokio::test]
async fn delete_notification_rejects_invalid_id_format() {
    let app = spawn_streaming_test_app().await;
    let response = reqwest::Client::new()
        .delete(format!(
            "{}/api/v1/admin/notification/not-a-valid-id",
            app.address
        ))
        .send()
        .await
        .expect("failed to send delete request");

    assert_eq!(response.status().as_u16(), 400);
}

#[tokio::test]
async fn delete_notification_returns_not_found_for_missing_sequence() {
    let app = spawn_streaming_test_app().await;
    let response = reqwest::Client::new()
        .delete(format!(
            "{}/api/v1/admin/notification/test_polygon@999999999",
            app.address
        ))
        .send()
        .await
        .expect("failed to send delete request");

    assert_eq!(response.status().as_u16(), 404);
}

#[tokio::test]
async fn delete_notification_accepts_event_type_alias_and_removes_message() {
    let app = spawn_streaming_test_app().await;
    let client = reqwest::Client::new();
    let unique_stream = format!("ens.member.{}", unique_suffix());
    let expected_note = format!("admin-delete-note-{}", unique_suffix());

    let publish_response =
        post_mars_notification(&client, &app.address, &expected_note, &unique_stream).await;
    assert_eq!(publish_response.status().as_u16(), 200);

    let replay_before = client
        .post(format!("{}/api/v1/replay", app.address))
        .header("Content-Type", "application/json")
        .json(&json!({
            "event_type": "mars",
            "identifier": {
                "class": "od",
                "expver": "0001",
                "domain": "g",
                "date": "20250706",
                "time": "1200",
                "stream": unique_stream,
                "step": "1"
            },
            "from_id": "1"
        }))
        .send()
        .await
        .expect("failed to send replay request");
    assert_eq!(replay_before.status().as_u16(), 200);
    let before_body = replay_before
        .text()
        .await
        .expect("failed to read replay body before delete");
    assert!(
        replay_event_count(&before_body) == 1,
        "expected replay to include exactly one historical message before delete; body: {before_body}"
    );

    let sequence = extract_event_sequence(&before_body, "mars")
        .expect("expected replay output to include mars sequence in CloudEvent id");
    let delete_response = client
        .delete(format!(
            "{}/api/v1/admin/notification/mars@{}",
            app.address, sequence
        ))
        .send()
        .await
        .expect("failed to send delete request");
    assert_eq!(delete_response.status().as_u16(), 200);

    let replay_after = client
        .post(format!("{}/api/v1/replay", app.address))
        .header("Content-Type", "application/json")
        .json(&json!({
            "event_type": "mars",
            "identifier": {
                "class": "od",
                "expver": "0001",
                "domain": "g",
                "date": "20250706",
                "time": "1200",
                "stream": unique_stream,
                "step": "1"
            },
            "from_id": "1"
        }))
        .send()
        .await
        .expect("failed to send replay request after delete");
    assert_eq!(replay_after.status().as_u16(), 200);
    let after_body = replay_after
        .text()
        .await
        .expect("failed to read replay body after delete");
    assert!(
        replay_event_count(&after_body) == 0,
        "expected replay to exclude deleted message; body: {after_body}"
    );
}