Add nix flake check for python unit tests

Author: jhh

flake.nix

@@ -62,6 +62,7 @@
             pillowHeifOverrides = import ./lib/overrides-pillow-heif.nix { inherit pkgs; };
             psycopgOverrides = import ./lib/overrides-psycopg.nix { inherit pkgs; };
             opencvOverrides = import ./lib/overrides-opencv.nix { inherit pkgs; };
+            strykeforceOverrides = import ./lib/overrides-strykeforce.nix { inherit pkgs workspace; };
             tbaApiOverrides = import ./lib/overrides-tba-api-v3client.nix { inherit pkgs; };
           in
           baseSet.overrideScope (
@@ -71,6 +72,7 @@
               pillowHeifOverrides
               psycopgOverrides
               opencvOverrides
+              strykeforceOverrides
               tbaApiOverrides
             ]
           );
@@ -79,6 +81,8 @@
 
       in
       {
+        checks = pythonSet.website.passthru.tests;
+
         packages = {
           venv = pythonSet.mkVirtualEnv "strykeforce-env" workspace.deps.default;
 

lib/overrides-strykeforce.nix

@@ -0,0 +1,39 @@
+{
+  # lib,
+  # nixosModule,
+  pkgs,
+  workspace,
+}:
+let
+  inherit (pkgs.stdenv) isLinux mkDerivation;
+in
+final: prev: {
+  #
+  website = prev.website.overrideAttrs (old: {
+    passthru = old.passthru // {
+      tests =
+        let
+          venv = final.mkVirtualEnv "strykeforce-check-env" workspace.deps.default;
+        in
+        (old.tests or { })
+        // {
+          unittest = mkDerivation {
+            name = "strykeforce-unittest";
+            inherit (final.website) src;
+            nativeBuildInputs = [ venv ];
+            dontConfigure = true;
+            dontInstall = true;
+
+            buildPhase = ''
+              runHook preBuild
+              export DJANGO_SETTINGS_MODULE="website.settings.test"
+              export TBA_READ_KEY=
+
+              ${venv}/bin/python ./website/manage.py test website.events.tests > $out 2>&1
+              runHook postBuild
+            '';
+          };
+        };
+    };
+  });
+}

website/events/management/commands/import_events.py

@@ -116,12 +116,14 @@ def copy_tba_event(self, tba_event, event):
                 case _:
                     pass
 
-    def handle(self, *args, **options):
-        self.download_events(options["year"])
-
+    def handle_events(self):
         for tba_event in self.events:
             with transaction.atomic():
                 event, created = Event.objects.get_or_create(key=tba_event.key)
                 if not event.edited_on:
                     self.copy_tba_event(tba_event, event)
                     event.save()
+
+    def handle(self, *args, **options):
+        self.download_events(options["year"])
+        self.handle_events()

website/events/tests/test_import_events.py

@@ -112,7 +112,7 @@ def test_dont_overwrite_edited_events(self):
         command.awards = data["awards"]
         command.results = data["results"]
 
-        command.handle(year=2024)
+        command.handle_events()
         event = Event.objects.get(key="2024misjo")
         self.assertIsNone(event.edited_on)
         self.assertEqual(event.status, "won")
@@ -121,6 +121,6 @@ def test_dont_overwrite_edited_events(self):
         event.edited_on = datetime.datetime.now()
         event.save()
 
-        command.handle(year=2024)
+        command.handle_events()
         event = Event.objects.get(key="2024misjo")
         self.assertEqual(event.status, "edited")

website/settings/test.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
-from .dev import *  # noqa
+from .base import *  # noqa
+
 
 DEBUG = False
 
@@ -10,5 +11,3 @@
         "NAME": ":memory:",
     },
 }
-
-MIGRATION_MODULES = {app: None for app in INSTALLED_APPS}  # noqa