From 8ef8744ba1a002bf2783d245b667003cc70fa276 Mon Sep 17 00:00:00 2001
From: DJ2LS <dj2ls@proton.me>
Date: Wed, 7 Feb 2024 19:49:00 +0100
Subject: [PATCH] fixed database test

---
 tests/test_message_database.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_message_database.py b/tests/test_message_database.py
index 0e0ebf25..6a20cae1 100644
--- a/tests/test_message_database.py
+++ b/tests/test_message_database.py
@@ -38,9 +38,9 @@ class TestDataFrameFactory(unittest.TestCase):
         received_message = MessageP2P.from_payload(payload)
         received_message_dict = MessageP2P.to_dict(received_message)
         self.database_manager.add_message(received_message_dict)
+        result = self.database_manager.get_message_by_id(message.id)
 
-        result = self.database_manager.get_all_messages()
-        self.assertEqual(result[0]["destination"], message.destination)
+        self.assertEqual(result["destination"], message.destination)
 
     def testDeleteFromDatabase(self):
         attachment = {