Add database connection test

Author: elibroftw

.github/workflows/build.yml

@@ -52,7 +52,7 @@ jobs:
           .venv\Scripts\Activate.ps1
           python build.py --ci --no-deps --no-build --no-install
 
-      - name: Save
+      - name: Save Failed Build
         if: failure() && steps.test.outcome == 'failure'
         uses: actions/upload-artifact@v4
         with:

conftest.py

@@ -18,3 +18,8 @@ def pytest_collection_modifyitems(config, items):
         for item in items:
             if 'no_ci' in item.keywords:
                 item.add_marker(skip_non_ci)
+    else:
+        skip_ci = pytest.mark.skip(reason='test only for CI')
+        for item in items:
+            if 'ci_only' in item.keywords:
+                item.add_marker(skip_ci)

src/modules/db.py

@@ -19,8 +19,9 @@ def create_connection():
         conn.row_factory = sqlite3.Row
         return conn
 
-    def __init__(self):
-        pass
+    def __init__(self, db_override=None):
+        if db_override is not None:
+            self.DATABASE_FILE = db_override
 
     def __enter__(self):
         self.conn = self.create_connection()

src/test_harness.py

@@ -11,6 +11,7 @@
 from PIL import Image
 import pytest
 
+from modules.db import DatabaseConnection
 from b64_images import DEFAULT_ART
 from meta import COVER_MINI, COVER_NORMAL, VERSION
 from shared import get_running_processes, is_already_running
@@ -607,3 +608,11 @@ def test_get_latest_release(uploading_after, test_auto_update):
         assert compare_ver < version
     else:
         assert compare_ver <= version
+
+@pytest.mark.ci_only
+def test_database():
+    DatabaseConnection.DEFAULT_DATABASE_FILE.parent.mkdir(parents=True, exist_ok=True)
+    with DatabaseConnection(DatabaseConnection.DEFAULT_DATABASE_FILE) as _:
+        pass
+    if DatabaseConnection.DEFAULT_DATABASE_FILE.exists():
+        os.remove(DatabaseConnection.DEFAULT_DATABASE_FILE)