[RELEASE] RexiMemo: Open Source Edition

tests/test_database.py

@@ -0,0 +1,42 @@
+import tempfile
+import unittest
+from pathlib import Path
+
+from database import DatabaseBackend
+
+
+class DatabaseTests(unittest.TestCase):
+    def setUp(self):
+        self.temp = tempfile.TemporaryDirectory()
+        self.db = DatabaseBackend(Path(self.temp.name) / "test.sqlite3")
+
+    def tearDown(self):
+        self.db.close()
+        self.temp.cleanup()
+
+    def test_default_admin(self):
+        account = self.db.verify_account("reximemo", "alpine")
+        self.assertIsNotNone(account)
+        self.assertEqual(account["is_admin"], 1)
+
+    def test_ip_login_and_logout(self):
+        aid = self.db.create_account("tester", "pass")
+        self.db.login_ip("192.0.2.10", aid)
+        self.assertEqual(self.db.account_for_ip("192.0.2.10")["username"], "tester")
+        self.db.logout_ip("192.0.2.10")
+        self.assertIsNone(self.db.account_for_ip("192.0.2.10"))
+
+    def test_guest_room_is_stable_per_ip(self):
+        first = self.db.get_or_create_creator("192.0.2.20", fsid="A000000000000010", display_name="Guest")
+        second = self.db.get_or_create_creator("192.0.2.20", fsid="A000000000000011", display_name="Other")
+        self.assertEqual(first["id"], second["id"])
+
+    def test_ip_ban(self):
+        self.db.ban_ip("192.0.2.30", "test")
+        self.assertTrue(self.db.is_ip_banned("192.0.2.30"))
+        self.db.unban_ip("192.0.2.30")
+        self.assertFalse(self.db.is_ip_banned("192.0.2.30"))
+
+
+if __name__ == "__main__":
+    unittest.main()