feat(tests): add comprehensive benchmarks for database performance, memory usage, and application stability, including new test files for various frontend and backend functionalities

tests/backend/test_database_robustness.py

@@ -0,0 +1,87 @@
+import unittest
+import os
+import sqlite3
+import tempfile
+import shutil
+from meshchatx.src.backend.database.provider import DatabaseProvider
+from meshchatx.src.backend.database.schema import DatabaseSchema
+
+
+class TestDatabaseRobustness(unittest.TestCase):
+    def setUp(self):
+        self.test_dir = tempfile.mkdtemp()
+        self.db_path = os.path.join(self.test_dir, "test_meshchat.db")
+        # Ensure we start with a fresh provider instance
+        if hasattr(DatabaseProvider, "_instance"):
+            DatabaseProvider._instance = None
+        self.provider = DatabaseProvider.get_instance(self.db_path)
+        self.schema = DatabaseSchema(self.provider)
+
+    def tearDown(self):
+        self.provider.close_all()
+        if hasattr(DatabaseProvider, "_instance"):
+            DatabaseProvider._instance = None
+        shutil.rmtree(self.test_dir)
+
+    def test_missing_column_healing(self):
+        # 1. Create a "legacy" table without the peer_hash column
+        self.provider.execute("""
+            CREATE TABLE lxmf_messages (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                hash TEXT UNIQUE,
+                source_hash TEXT,
+                destination_hash TEXT
+            )
+        """)
+
+        # 2. Also need the config table so initialize doesn't fail on version check
+        self.provider.execute("""
+            CREATE TABLE config (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                key TEXT UNIQUE,
+                value TEXT
+            )
+        """)
+        self.provider.execute(
+            "INSERT INTO config (key, value) VALUES (?, ?)", ("database_version", "1")
+        )
+
+        # 3. Attempt initialization.
+        # Previously this would crash with OperationalError: no such column: peer_hash
+        try:
+            self.schema.initialize()
+        except Exception as e:
+            self.fail(f"Initialization failed with missing column: {e}")
+
+        # 4. Verify the column was added
+        cursor = self.provider.execute("PRAGMA table_info(lxmf_messages)")
+        columns = [row[1] for row in cursor.fetchall()]
+        self.assertIn("peer_hash", columns)
+        self.assertIn("is_spam", columns)
+
+    def test_corrupt_config_initialization(self):
+        # 1. Create a database where the version is missing or garbled
+        self.provider.execute("""
+            CREATE TABLE config (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                key TEXT UNIQUE,
+                value TEXT
+            )
+        """)
+        # No version inserted
+
+        # 2. Initialization should still work
+        try:
+            self.schema.initialize()
+        except Exception as e:
+            self.fail(f"Initialization failed with missing version: {e}")
+
+        # 3. Version should now be set to LATEST
+        row = self.provider.fetchone(
+            "SELECT value FROM config WHERE key = 'database_version'"
+        )
+        self.assertEqual(int(row["value"]), self.schema.LATEST_VERSION)
+
+
+if __name__ == "__main__":
+    unittest.main()