fakecloud-sns 0.9.2

SNS implementation for FakeCloud
Documentation 
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
136
137
138
139
140
141
142

use chrono::{DateTime, Utc};
use parking_lot::RwLock;
use std::collections::{BTreeMap, HashMap};
use std::sync::Arc;

#[derive(Debug, Clone)]
pub struct SnsTopic {
    pub topic_arn: String,
    pub name: String,
    pub attributes: HashMap<String, String>,
    pub tags: Vec<(String, String)>,
    pub is_fifo: bool,
    pub created_at: DateTime<Utc>,
}

#[derive(Debug, Clone)]
pub struct SnsSubscription {
    pub subscription_arn: String,
    pub topic_arn: String,
    pub protocol: String,
    pub endpoint: String,
    pub owner: String,
    pub attributes: HashMap<String, String>,
    pub confirmed: bool,
    /// Token used for HTTP/HTTPS subscription confirmation.
    pub confirmation_token: Option<String>,
}

/// An SNS message attribute (key-value with a data type).
#[derive(Debug, Clone)]
pub struct MessageAttribute {
    pub data_type: String,
    pub string_value: Option<String>,
    pub binary_value: Option<Vec<u8>>,
}

#[derive(Debug, Clone)]
pub struct PublishedMessage {
    pub message_id: String,
    pub topic_arn: String,
    pub message: String,
    pub subject: Option<String>,
    pub message_attributes: HashMap<String, MessageAttribute>,
    pub message_group_id: Option<String>,
    pub message_dedup_id: Option<String>,
    pub timestamp: DateTime<Utc>,
}

#[derive(Debug, Clone)]
pub struct PlatformApplication {
    pub arn: String,
    pub name: String,
    pub platform: String,
    pub attributes: HashMap<String, String>,
    pub endpoints: HashMap<String, PlatformEndpoint>,
}

#[derive(Debug, Clone)]
pub struct PlatformEndpoint {
    pub arn: String,
    pub token: String,
    pub attributes: HashMap<String, String>,
    pub enabled: bool,
    pub messages: Vec<PublishedMessage>,
}

/// A recorded Lambda invocation from SNS delivery.
#[derive(Debug, Clone)]
pub struct LambdaInvocation {
    pub function_arn: String,
    pub message: String,
    pub subject: Option<String>,
    pub timestamp: DateTime<Utc>,
}

/// A recorded email delivery from SNS (stub).
#[derive(Debug, Clone)]
pub struct SentEmail {
    pub email_address: String,
    pub message: String,
    pub subject: Option<String>,
    pub topic_arn: String,
    pub timestamp: DateTime<Utc>,
}

pub struct SnsState {
    pub account_id: String,
    pub region: String,
    pub endpoint: String,
    pub topics: BTreeMap<String, SnsTopic>, // arn -> topic (ordered for predictable iteration)
    pub subscriptions: BTreeMap<String, SnsSubscription>, // sub_arn -> subscription
    pub published: Vec<PublishedMessage>,
    pub platform_applications: BTreeMap<String, PlatformApplication>,
    pub sms_attributes: HashMap<String, String>,
    pub opted_out_numbers: Vec<String>,
    pub sms_messages: Vec<(String, String)>, // (phone_number, message)
    /// Recorded Lambda invocations (stub deliveries).
    pub lambda_invocations: Vec<LambdaInvocation>,
    /// Recorded email deliveries (stub — not actually sent).
    pub sent_emails: Vec<SentEmail>,
}

impl SnsState {
    pub fn new(account_id: &str, region: &str, endpoint: &str) -> Self {
        Self {
            account_id: account_id.to_string(),
            region: region.to_string(),
            endpoint: endpoint.to_string(),
            topics: BTreeMap::new(),
            subscriptions: BTreeMap::new(),
            published: Vec::new(),
            platform_applications: BTreeMap::new(),
            sms_attributes: HashMap::new(),
            opted_out_numbers: Vec::new(),
            sms_messages: Vec::new(),
            lambda_invocations: Vec::new(),
            sent_emails: Vec::new(),
        }
    }

    pub fn reset(&mut self) {
        self.topics.clear();
        self.subscriptions.clear();
        self.published.clear();
        self.platform_applications.clear();
        self.sms_attributes.clear();
        self.opted_out_numbers.clear();
        self.sms_messages.clear();
        self.lambda_invocations.clear();
        self.sent_emails.clear();
    }

    /// Seed default opt-out phone numbers.
    pub fn seed_default_opted_out(&mut self) {
        if self.opted_out_numbers.is_empty() {
            self.opted_out_numbers.push("+15005550099".to_string());
            self.opted_out_numbers.push("+447428545399".to_string());
        }
    }
}

pub type SharedSnsState = Arc<RwLock<SnsState>>;