DST-943 - Remove test cases from the viewer portals (#312)

django_app/core/management/commands/delete_breach_reports.py

@@ -0,0 +1,23 @@
+from django.core.management.base import BaseCommand
+from report_a_suspected_breach.models import Breach
+
+
+class Command(BaseCommand):
+    help = (
+        "Deletes suspected breach reports when giving a list of report references. "
+        "Usage: pipenv run django_app/python manage.py delete_breach_reports <reference> <reference> ..."
+    )
+
+    def add_arguments(self, parser):
+        parser.add_argument("breach_references", nargs="+", type=str)
+
+    def handle(self, *args, **options):
+        for reference in options["breach_references"]:
+            try:
+                breach_object = Breach.objects.get(reference=reference)
+                breach_object.delete()
+            except Breach.DoesNotExist:
+                self.stdout.write(self.style.ERROR(f"Breach {reference} does not exist"))
+                continue
+
+            self.stdout.write(self.style.SUCCESS(f"Successfully deleted breach report {reference}"))

tests/test_unit/test_core/test_management_commands/test_delete_breach_reports.py

@@ -0,0 +1,23 @@
+from django.core.management import call_command
+from report_a_suspected_breach.models import Breach
+
+from tests.factories import BreachFactory
+
+
+def test_successful_delete(db):
+    BreachFactory(reference="123")
+    BreachFactory(reference="456")
+
+    assert Breach.objects.count() == 2
+
+    call_command("delete_breach_reports", ["123", "456"])
+    assert Breach.objects.count() == 0
+
+
+def test_doesnt_exist_delete(db):
+    BreachFactory(reference="123")
+
+    assert Breach.objects.count() == 1
+
+    call_command("delete_breach_reports", ["456"])
+    assert Breach.objects.count() == 1