mostro_core/
lib.rs

1pub mod crypto;
2pub mod dispute;
3pub mod error;
4pub mod message;
5pub mod order;
6pub mod prelude;
7pub mod rating;
8pub mod user;
9#[cfg(test)]
10mod test {
11    use crate::error::CantDoReason;
12    use crate::message::{Action, Message, MessageKind, Payload, Peer};
13    use crate::order::{Kind, SmallOrder, Status};
14    use nostr_sdk::Keys;
15    use uuid::uuid;
16
17    #[test]
18    fn test_status_string() {
19        assert_eq!(Status::Active.to_string(), "active");
20        assert_eq!(Status::CompletedByAdmin.to_string(), "completed-by-admin");
21        assert_eq!(Status::FiatSent.to_string(), "fiat-sent");
22        assert_ne!(Status::Pending.to_string(), "Pending");
23    }
24
25    #[test]
26    fn test_kind_string() {
27        assert_ne!(Kind::Sell.to_string(), "active");
28        assert_eq!(Kind::Sell.to_string(), "sell");
29        assert_eq!(Kind::Buy.to_string(), "buy");
30        assert_ne!(Kind::Buy.to_string(), "active");
31    }
32
33    #[test]
34    fn test_order_message() {
35        let uuid = uuid!("308e1272-d5f4-47e6-bd97-3504baea9c23");
36        let payment_methods = "SEPA,Bank transfer".to_string();
37        let payload = Payload::Order(
38            SmallOrder::new(
39                Some(uuid),
40                Some(Kind::Sell),
41                Some(Status::Pending),
42                100,
43                "eur".to_string(),
44                None,
45                None,
46                100,
47                payment_methods,
48                1,
49                None,
50                None,
51                None,
52                Some(1627371434),
53                None,
54                None,
55                None,
56            ),
57            None,
58        );
59
60        let test_message = Message::Order(MessageKind::new(
61            Some(uuid),
62            Some(1),
63            Some(2),
64            Action::NewOrder,
65            Some(payload),
66        ));
67        let test_message_json = test_message.as_json().unwrap();
68        let sample_message = r#"{"order":{"version":1,"id":"308e1272-d5f4-47e6-bd97-3504baea9c23","request_id":1,"trade_index":2,"action":"new-order","payload":{"order":[{"id":"308e1272-d5f4-47e6-bd97-3504baea9c23","kind":"sell","status":"pending","amount":100,"fiat_code":"eur","fiat_amount":100,"payment_method":"SEPA,Bank transfer","premium":1,"created_at":1627371434},null]}}}"#;
69        let message = Message::from_json(sample_message).unwrap();
70        assert!(message.verify());
71        let message_json = message.as_json().unwrap();
72        assert_eq!(message_json, test_message_json);
73    }
74
75    #[test]
76    fn test_payment_request_payload_message() {
77        let uuid = uuid!("308e1272-d5f4-47e6-bd97-3504baea9c23");
78        let test_message = Message::Order(MessageKind::new(
79            Some(uuid),
80            Some(1),
81            Some(3),
82            Action::PayInvoice,
83            Some(Payload::PaymentRequest(
84                Some(SmallOrder::new(
85                    Some(uuid),
86                    Some(Kind::Sell),
87                    Some(Status::WaitingPayment),
88                    100,
89                    "eur".to_string(),
90                    None,
91                    None,
92                    100,
93                    "Face to face".to_string(),
94                    1,
95                    None,
96                    None,
97                    None,
98                    Some(1627371434),
99                    None,
100                    None,
101                    None,
102                )),
103                "lnbcrt78510n1pj59wmepp50677g8tffdqa2p8882y0x6newny5vtz0hjuyngdwv226nanv4uzsdqqcqzzsxqyz5vqsp5skn973360gp4yhlpmefwvul5hs58lkkl3u3ujvt57elmp4zugp4q9qyyssqw4nzlr72w28k4waycf27qvgzc9sp79sqlw83j56txltz4va44j7jda23ydcujj9y5k6k0rn5ms84w8wmcmcyk5g3mhpqepf7envhdccp72nz6e".to_string(),
104                None,
105                None,
106            )),
107        ));
108        let sample_message = r#"{"order":{"version":1,"id":"308e1272-d5f4-47e6-bd97-3504baea9c23","request_id":1,"trade_index":3,"action":"pay-invoice","payload":{"payment_request":[{"id":"308e1272-d5f4-47e6-bd97-3504baea9c23","kind":"sell","status":"waiting-payment","amount":100,"fiat_code":"eur","fiat_amount":100,"payment_method":"Face to face","premium":1,"created_at":1627371434},"lnbcrt78510n1pj59wmepp50677g8tffdqa2p8882y0x6newny5vtz0hjuyngdwv226nanv4uzsdqqcqzzsxqyz5vqsp5skn973360gp4yhlpmefwvul5hs58lkkl3u3ujvt57elmp4zugp4q9qyyssqw4nzlr72w28k4waycf27qvgzc9sp79sqlw83j56txltz4va44j7jda23ydcujj9y5k6k0rn5ms84w8wmcmcyk5g3mhpqepf7envhdccp72nz6e",null,null]}}}"#;
109        let message = Message::from_json(sample_message).unwrap();
110        assert!(message.verify());
111        let message_json = message.as_json().unwrap();
112        let test_message_json = test_message.as_json().unwrap();
113        assert_eq!(message_json, test_message_json);
114    }
115
116    #[test]
117    fn test_message_payload_signature() {
118        let uuid = uuid!("308e1272-d5f4-47e6-bd97-3504baea9c23");
119        let peer = Peer::new(
120            "npub1testjsf0runcqdht5apkfcalajxkf8txdxqqk5kgm0agc38ke4vsfsgzf8".to_string(),
121            None,
122        );
123        let payload = Payload::Peer(peer);
124        let test_message = Message::Order(MessageKind::new(
125            Some(uuid),
126            Some(1),
127            Some(2),
128            Action::FiatSentOk,
129            Some(payload),
130        ));
131        assert!(test_message.verify());
132        let test_message_json = test_message.as_json().unwrap();
133        // Message should be signed with the trade keys
134        let trade_keys =
135            Keys::parse("110e43647eae221ab1da33ddc17fd6ff423f2b2f49d809b9ffa40794a2ab996c")
136                .unwrap();
137        let sig = Message::sign(test_message_json.clone(), &trade_keys);
138
139        assert!(Message::verify_signature(
140            test_message_json,
141            trade_keys.public_key(),
142            sig
143        ));
144    }
145
146    #[test]
147    fn test_cant_do_message_serialization() {
148        // Test all CantDoReason variants
149        let reasons = vec![
150            CantDoReason::InvalidSignature,
151            CantDoReason::InvalidTradeIndex,
152            CantDoReason::InvalidAmount,
153            CantDoReason::InvalidInvoice,
154            CantDoReason::InvalidPaymentRequest,
155            CantDoReason::InvalidPeer,
156            CantDoReason::InvalidRating,
157            CantDoReason::InvalidTextMessage,
158            CantDoReason::InvalidOrderStatus,
159            CantDoReason::InvalidPubkey,
160            CantDoReason::InvalidParameters,
161            CantDoReason::OrderAlreadyCanceled,
162            CantDoReason::CantCreateUser,
163            CantDoReason::IsNotYourOrder,
164            CantDoReason::NotAllowedByStatus,
165            CantDoReason::OutOfRangeFiatAmount,
166            CantDoReason::OutOfRangeSatsAmount,
167            CantDoReason::IsNotYourDispute,
168            CantDoReason::NotFound,
169        ];
170
171        for reason in reasons {
172            let cant_do = Message::CantDo(MessageKind::new(
173                None,
174                None,
175                None,
176                Action::CantDo,
177                Some(Payload::CantDo(Some(reason.clone()))),
178            ));
179            let message = Message::from_json(&cant_do.as_json().unwrap()).unwrap();
180            assert!(message.verify());
181            assert_eq!(message.as_json().unwrap(), cant_do.as_json().unwrap());
182        }
183
184        // Test None case
185        let cant_do = Message::CantDo(MessageKind::new(
186            None,
187            None,
188            None,
189            Action::CantDo,
190            Some(Payload::CantDo(None)),
191        ));
192        let message = Message::from_json(&cant_do.as_json().unwrap()).unwrap();
193        assert!(message.verify());
194        assert_eq!(message.as_json().unwrap(), cant_do.as_json().unwrap());
195    }
196}