From b96ade872f5e65d199b0087904130e94f3a5702b Mon Sep 17 00:00:00 2001
From: Joscha <joscha@plugh.de>
Date: Tue, 14 Jun 2022 09:34:22 +0200
Subject: [PATCH] Fix dummy store tree retrieval

---
 cove-tui/src/store/dummy.rs | 8 ++++----
 cove-tui/src/ui.rs          | 3 ++-
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/cove-tui/src/store/dummy.rs b/cove-tui/src/store/dummy.rs
index 6f52559..5116bc2 100644
--- a/cove-tui/src/store/dummy.rs
+++ b/cove-tui/src/store/dummy.rs
@@ -82,10 +82,10 @@ impl DummyStore {
     fn collect_tree(&self, id: usize, result: &mut Vec<DummyMsg>) {
         if let Some(msg) = self.msgs.get(&id) {
             result.push(msg.clone());
-            if let Some(children) = self.children.get(&id) {
-                for child in children {
-                    self.collect_tree(*child, result);
-                }
+        }
+        if let Some(children) = self.children.get(&id) {
+            for child in children {
+                self.collect_tree(*child, result);
             }
         }
     }
diff --git a/cove-tui/src/ui.rs b/cove-tui/src/ui.rs
index 2bbbce3..2749657 100644
--- a/cove-tui/src/ui.rs
+++ b/cove-tui/src/ui.rs
@@ -45,7 +45,8 @@ impl Ui {
         // Prepare dummy message store and chat for testing
         let store = DummyStore::new()
             .msg(DummyMsg::new(1, "nick", "content"))
-            .msg(DummyMsg::new(2, "Some1Else", "reply").parent(1));
+            .msg(DummyMsg::new(2, "Some1Else", "reply").parent(1))
+            .msg(DummyMsg::new(4, "nick", "reply to nothing").parent(3));
         let chat = Chat::new(store, "testroom".to_string());
 
         // Run main UI.