Some database tests

cry/database.py

@@ -76,7 +76,6 @@ def local_origin(path: pathlib.Path | None = None) -> str:
 
 
 def database_path(origin: str) -> pathlib.Path:
-    # TODO: Determine the name/slug from local state if necessary
     return pathlib.Path.home() / "Dropbox" / "cry" / f"{origin}.db"
 
 

tests/test_database.py

@@ -0,0 +1,66 @@
+import pathlib
+import random
+import string
+import tempfile
+
+from cry import database
+
+
+def random_slug() -> str:
+    return "".join(
+        random.choices(
+            string.ascii_uppercase + string.ascii_lowercase + string.digits, k=8
+        )
+    )
+
+
+def test_database_origin_path():
+    op = database.origin_path()
+    assert op is not None
+
+
+def test_database_local_origin():
+    with tempfile.TemporaryDirectory() as op:
+        origin_file = pathlib.Path(op) / "origin"
+        assert not origin_file.exists()
+
+        origin = database.local_origin(origin_file)
+
+        assert origin_file.exists()
+        assert len(origin) > 0
+
+
+def test_database_local_origin_repeatable():
+    with tempfile.TemporaryDirectory() as op:
+        origin_file = pathlib.Path(op) / "origin"
+
+        a = database.local_origin(origin_file)
+        b = database.local_origin(origin_file)
+
+        assert len(a) > 0
+        assert a == b
+
+
+def test_database_origin_in_path():
+    slug = random_slug()
+    p = database.database_path(slug)
+    assert slug in str(p)
+
+
+def test_database_schema():
+    db = database.Database(":memory:", random_slug())
+    db.ensure_database_schema()
+
+    c = db.db.execute("SELECT value FROM properties WHERE name = 'version'")
+    row = c.fetchone()
+    assert int(row[0]) == len(database.SCHEMA_STATEMENTS)
+
+
+def test_database_prop_get_set():
+    db = database.Database(":memory:", random_slug())
+    db.ensure_database_schema()
+
+    assert db.get_property("foo") is None
+    val = random_slug()
+    db.set_property("foo", val)
+    assert db.get_property("foo") == val