add test for management command

castellum_core/tests/castellum_auth/management_commands/test_deactivate_expired_user_accounts.py

+from datetime import date, timedelta
+from itertools import cycle
+
+import pytest
+from model_mommy import mommy
+
+from castellum_auth.models import User
+from castellum_auth.management.commands.deactivate_expired_user_accounts import Command
+
+
+TODAY = date.today()
+FUTURE_DATE = TODAY - timedelta(days=-1)
+PAST_DATE = TODAY - timedelta(days=1)
+
+
+@pytest.mark.django_db
+@pytest.mark.parametrize("inactive_user_count, expiration_dates", [
+    (0, [TODAY]),
+    (0, [TODAY, FUTURE_DATE]),
+    (2, [TODAY, FUTURE_DATE, PAST_DATE]),
+])
+def test_command(inactive_user_count, expiration_dates):
+    mommy.make(User, expiration_date=cycle(expiration_dates), _quantity=6)
+
+    Command().handle()
+
+    assert inactive_user_count == User.objects.filter(is_active=False).count()

castellum_core/tox.ini

@@ -28,6 +28,7 @@ max-line-length=100
 deps =
     pytest-django
     pytest-cov
+    model_mommy
 setenv =
     PYTHONPATH = {toxinidir}/castellum_core:{env:PYTHONPATH:}
 commands =