add simulation test

1 files changed, 21 insertions, 7 deletions

diff --git a/src/transport/mycelium.rs b/src/transport/mycelium.rs
index f519928..efab475 100644
--- a/src/transport/mycelium.rs
+++ b/src/transport/mycelium.rs
@@ -561,13 +561,7 @@ impl Transport for MyceliumTransport {
     }
 
     fn encode(&self, peer: &str, message: &ParsedMessage) -> Result<EncodedMessage> {
-        let (outbound, payload_len) = format_message(peer, message)?;
-        Ok(EncodedMessage {
-            peer: peer.to_owned(),
-            body: serde_json::to_vec(&outbound)?.into(),
-            kind: message.kind(),
-            payload_len,
-        })
+        encode_message(peer, message)
     }
 
     async fn send_encoded(&self, message: &EncodedMessage) -> Result<()> {
@@ -712,6 +706,26 @@ impl Transport for MyceliumTransport {
     }
 }
 
+/// A message as the Mycelium REST API takes it (the JSON body of an outbound message).
+/// Also the simulator's wire format (`transport::sim`), so tests go through the same
+/// encoding.
+pub(crate) fn encode_message(peer: &str, message: &ParsedMessage) -> Result<EncodedMessage> {
+    let (outbound, payload_len) = format_message(peer, message)?;
+    Ok(EncodedMessage {
+        peer: peer.to_owned(),
+        body: serde_json::to_vec(&outbound)?.into(),
+        kind: message.kind(),
+        payload_len,
+    })
+}
+
+/// The message in a body made by [`encode_message`].
+#[cfg(test)]
+pub(crate) fn decode_message(body: &[u8]) -> Result<ParsedMessage> {
+    let outbound: OutboundMessage = serde_json::from_slice(body)?;
+    Ok(decode_payload(&outbound.topic, &outbound.payload)?)
+}
+
 fn encode_topic(topic: &str) -> String {
     BASE64.encode(topic.as_bytes())
 }