From f31fb3bcdcea1c992db5607cd43822397d1e609a Mon Sep 17 00:00:00 2001
From: TuDatTr <tuan-dat.tran@tudattr.dev>
Date: Fri, 2 Feb 2024 03:53:18 +0100
Subject: [PATCH] Fixed challenge 03

Signed-off-by: TuDatTr <tuan-dat.tran@tudattr.dev>
---
 src/lib.rs         | 2 +-
 src/messages.rs    | 2 +-
 src/routes.rs      | 4 ++--
 tests/broadcast.rs | 2 +-
 4 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 71e3c24..b0368a0 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -11,7 +11,7 @@ type AppState = Arc<Mutex<State>>;
 #[derive(Default)]
 pub struct State {
     uid: IdCounter,
-    message: String,
+    message: Vec<String>,
 }
 
 #[derive(Debug, Default)]
diff --git a/src/messages.rs b/src/messages.rs
index e9e17cd..d34db21 100644
--- a/src/messages.rs
+++ b/src/messages.rs
@@ -42,7 +42,7 @@ pub enum MessageBody {
     ReadOk {
         msg_id: u64,
         in_reply_to: u64,
-        message: String,
+        message: Vec<String>,
     },
     Default,
 }
diff --git a/src/routes.rs b/src/routes.rs
index 76e304b..a8bc25d 100644
--- a/src/routes.rs
+++ b/src/routes.rs
@@ -1,5 +1,5 @@
 use axum::{extract::State, http::StatusCode, Json};
-use tracing::{info};
+use tracing::info;
 
 use crate::{
     messages::{Message, MessageBody},
@@ -51,7 +51,7 @@ pub async fn challenge(
                 let in_reply_to = msg_id;
                 {
                     let mut local_state = state.lock().unwrap();
-                    local_state.message = message;
+                    local_state.message.push(message);
                 }
                 MessageBody::BroadcastOk {
                     msg_id,
diff --git a/tests/broadcast.rs b/tests/broadcast.rs
index 43bd5e5..5110331 100644
--- a/tests/broadcast.rs
+++ b/tests/broadcast.rs
@@ -85,7 +85,7 @@ async fn test_rw() {
                     "type": "read_ok",
                     "msg_id": 2,
                     "in_reply_to": 2,
-                    "message": message
+                    "message": vec![message]
                 }
             }
         );