chg: [tp] Added model for storing test result

Signed-off-by: Romain Kieffer <[email protected]>

testing/admin.py

@@ -7,6 +7,7 @@
     MailDomain,
     TlsScanHistory,
     UserDomain,
+    TestReport
 )
 
 
@@ -26,3 +27,4 @@ def get_username(self, obj):
 admin.site.register(DMARCRecord)
 admin.site.register(DMARCReport)
 admin.site.register(TlsScanHistory)
+admin.site.register(TestReport)

testing/models.py

@@ -94,3 +94,12 @@ class DMARCReport(models.Model):
 
     def __str__(self):
         return f"{self.dmarc_record}-{self.mail_from}-{self.timestamp}"
+
+
+class TestReport(models.Model):
+    tested_site = models.CharField(max_length=200, blank=True, null=True)
+    test_ran = models.CharField(max_length=200, blank=True, null=True)
+    report = models.JSONField()
+
+    def __str__(self):
+        return f"{self.test_ran}_{self.tested_site.replace('.', '-')}"

testing/views.py

@@ -47,7 +47,7 @@
 
 from .zap import zap_scan
 
-from .models import DMARCRecord, DMARCReport, MailDomain
+from .models import DMARCRecord, DMARCReport, MailDomain, TestReport
 
 
 @login_required
@@ -140,8 +140,13 @@ def zap_test(request):
         nb_tests += 1
         context = json_report['site'][0]
         response = render(request, "check_zap.html", context)
+        test_report = TestReport(
+            tested_site=target,
+            test_ran="zap",
+            report=context
+        )
+        test_report.save()
         response.set_cookie("nb_tests", nb_tests)
-        print("wat")
         return response
         # return HttpResponse(html_report)
     else: