Got content added to new pages.

src/document.rs

@@ -2,6 +2,7 @@ use crate::{
     queue::{Message, MsgType, Queue},
     ActionType, ErrorType,
 };
+use serde_json::Value;
 use std::{
     collections::HashMap,
     sync::mpsc::{channel, Receiver},
@@ -51,8 +52,9 @@ impl Document {
 
     fn add(&mut self, msg: Message) {
         let name = msg.get_data("name").unwrap().to_string();
-        let doc = msg.get_data("doc").unwrap().to_string();
-        self.data.insert(name, doc);
+        let doc: Value = serde_json::from_str(&msg.get_data("doc").unwrap().to_string()).unwrap();
+        self.data
+            .insert(name, doc["template"].as_str().unwrap().to_string());
         self.queue.send(msg.reply(MsgType::ActionOk)).unwrap();
     }
 
@@ -66,7 +68,7 @@ impl Document {
                 let mut holder = msg.reply(MsgType::Document);
                 holder.add_data("doc", data.clone());
                 holder
-            },
+            }
             None => {
                 let mut holder = msg.reply(MsgType::Error);
                 holder.add_data("error_type", ErrorType::DocumentNotFound);
@@ -80,6 +82,7 @@ impl Document {
 #[cfg(test)]
 pub mod documents {
     use super::*;
+    use serde_json::json;
     use std::time::Duration;
     use uuid::Uuid;
 
@@ -88,7 +91,10 @@ pub mod documents {
     fn setup_document() -> (Queue, Receiver<Message>) {
         let queue = Queue::new();
         let (tx, rx) = channel();
-        queue.add(tx, [MsgType::ActionOk, MsgType::Document, MsgType::Error].to_vec());
+        queue.add(
+            tx,
+            [MsgType::ActionOk, MsgType::Document, MsgType::Error].to_vec(),
+        );
         Document::start(queue.clone());
         (queue, rx)
     }
@@ -153,10 +159,13 @@ pub mod documents {
         let (queue, rx) = setup_document();
         let name = format!("name-{}", Uuid::new_v4());
         let content = format!("content-{}", Uuid::new_v4());
+        let input = json!({
+            "template": content.clone()
+        });
         let mut msg1 = Message::new(MsgType::DocumentRequest);
         msg1.add_data("name", name.clone());
         msg1.add_data("action", ActionType::Add);
-        msg1.add_data("doc", content.clone());
+        msg1.add_data("doc", input.to_string());
         queue.send(msg1.clone()).unwrap();
         let reply1 = rx.recv_timeout(TIMEOUT).unwrap();
         assert_eq!(reply1.get_id(), msg1.get_id());