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(SmallOrder::new(
38            Some(uuid),
39            Some(Kind::Sell),
40            Some(Status::Pending),
41            100,
42            "eur".to_string(),
43            None,
44            None,
45            100,
46            payment_methods,
47            1,
48            None,
49            None,
50            None,
51            Some(1627371434),
52            None,
53        ));
54
55        let test_message = Message::Order(MessageKind::new(
56            Some(uuid),
57            Some(1),
58            Some(2),
59            Action::NewOrder,
60            Some(payload),
61        ));
62        let test_message_json = test_message.as_json().unwrap();
63        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}}}}"#;
64        let message = Message::from_json(sample_message).unwrap();
65        assert!(message.verify());
66        let message_json = message.as_json().unwrap();
67        assert_eq!(message_json, test_message_json);
68    }
69
70    #[test]
71    fn test_payment_request_payload_message() {
72        let uuid = uuid!("308e1272-d5f4-47e6-bd97-3504baea9c23");
73        let test_message = Message::Order(MessageKind::new(
74            Some(uuid),
75            Some(1),
76            Some(3),
77            Action::PayInvoice,
78            Some(Payload::PaymentRequest(
79                Some(SmallOrder::new(
80                    Some(uuid),
81                    Some(Kind::Sell),
82                    Some(Status::WaitingPayment),
83                    100,
84                    "eur".to_string(),
85                    None,
86                    None,
87                    100,
88                    "Face to face".to_string(),
89                    1,
90                    None,
91                    None,
92                    None,
93                    Some(1627371434),
94                    None,
95                )),
96                "lnbcrt78510n1pj59wmepp50677g8tffdqa2p8882y0x6newny5vtz0hjuyngdwv226nanv4uzsdqqcqzzsxqyz5vqsp5skn973360gp4yhlpmefwvul5hs58lkkl3u3ujvt57elmp4zugp4q9qyyssqw4nzlr72w28k4waycf27qvgzc9sp79sqlw83j56txltz4va44j7jda23ydcujj9y5k6k0rn5ms84w8wmcmcyk5g3mhpqepf7envhdccp72nz6e".to_string(),
97                None,
98            )),
99        ));
100        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]}}}"#;
101        let message = Message::from_json(sample_message).unwrap();
102        assert!(message.verify());
103        let message_json = message.as_json().unwrap();
104        let test_message_json = test_message.as_json().unwrap();
105        assert_eq!(message_json, test_message_json);
106    }
107
108    #[test]
109    fn test_message_payload_signature() {
110        let uuid = uuid!("308e1272-d5f4-47e6-bd97-3504baea9c23");
111        let peer = Peer::new(
112            "npub1testjsf0runcqdht5apkfcalajxkf8txdxqqk5kgm0agc38ke4vsfsgzf8".to_string(),
113            None,
114        );
115        let payload = Payload::Peer(peer);
116        let test_message = Message::Order(MessageKind::new(
117            Some(uuid),
118            Some(1),
119            Some(2),
120            Action::FiatSentOk,
121            Some(payload),
122        ));
123        assert!(test_message.verify());
124        let test_message_json = test_message.as_json().unwrap();
125        // Message should be signed with the trade keys
126        let trade_keys =
127            Keys::parse("110e43647eae221ab1da33ddc17fd6ff423f2b2f49d809b9ffa40794a2ab996c")
128                .unwrap();
129        let sig = Message::sign(test_message_json.clone(), &trade_keys);
130
131        assert!(Message::verify_signature(
132            test_message_json,
133            trade_keys.public_key(),
134            sig
135        ));
136    }
137
138    #[test]
139    fn test_cant_do_message_serialization() {
140        // Test all CantDoReason variants
141        let reasons = vec![
142            CantDoReason::InvalidSignature,
143            CantDoReason::InvalidTradeIndex,
144            CantDoReason::InvalidAmount,
145            CantDoReason::InvalidInvoice,
146            CantDoReason::InvalidPaymentRequest,
147            CantDoReason::InvalidPeer,
148            CantDoReason::InvalidRating,
149            CantDoReason::InvalidTextMessage,
150            CantDoReason::InvalidOrderStatus,
151            CantDoReason::InvalidPubkey,
152            CantDoReason::InvalidParameters,
153            CantDoReason::OrderAlreadyCanceled,
154            CantDoReason::CantCreateUser,
155            CantDoReason::IsNotYourOrder,
156            CantDoReason::NotAllowedByStatus,
157            CantDoReason::OutOfRangeFiatAmount,
158            CantDoReason::OutOfRangeSatsAmount,
159            CantDoReason::IsNotYourDispute,
160            CantDoReason::NotFound,
161        ];
162
163        for reason in reasons {
164            let cant_do = Message::CantDo(MessageKind::new(
165                None,
166                None,
167                None,
168                Action::CantDo,
169                Some(Payload::CantDo(Some(reason.clone()))),
170            ));
171            let message = Message::from_json(&cant_do.as_json().unwrap()).unwrap();
172            assert!(message.verify());
173            assert_eq!(message.as_json().unwrap(), cant_do.as_json().unwrap());
174        }
175
176        // Test None case
177        let cant_do = Message::CantDo(MessageKind::new(
178            None,
179            None,
180            None,
181            Action::CantDo,
182            Some(Payload::CantDo(None)),
183        ));
184        let message = Message::from_json(&cant_do.as_json().unwrap()).unwrap();
185        assert!(message.verify());
186        assert_eq!(message.as_json().unwrap(), cant_do.as_json().unwrap());
187    }
188}