Add begin and end date range to the renders (#12)

Author: tdpauw

DMARCReporting/constants.py

@@ -1,4 +1,6 @@
 headers = [
+    "Begin Date",
+    "End Date",
     "Source IP",
     "Source Host",
     "Payload From (From:)",

tests/test_console_renderer.py

@@ -31,6 +31,8 @@ def test_render_with_data():
 
         data = [
             [
+                "begin_date",
+                "end_date",
                 "source_ip",
                 "source_host",
                 "payload_from",
@@ -49,9 +51,9 @@ def test_render_with_data():
         sut.render("ruareport_file", data)
 
         expected = (
-            "Source IP    Source Host    Payload From (From:)    Envelop From (MAIL FROM)    DMARC              DKIM Align    DKIM Auth    SPF Align    SPF Auth      Count  File\n"  # noqa E501
-            "-----------  -------------  ----------------------  --------------------------  -----------------  ------------  -----------  -----------  ----------  -------  --------------\n"  # noqa E501
-            "source_ip    source_host    payload_from            envelop_from                dmarc_disposition  dkim_aligned  dkim_auth    spf_aligned  spf_auth         42  ruareport_file\n"  # noqa E501
+            "Begin Date    End Date    Source IP    Source Host    Payload From (From:)    Envelop From (MAIL FROM)    DMARC              DKIM Align    DKIM Auth    SPF Align    SPF Auth      Count  File\n"  # noqa E501
+            "------------  ----------  -----------  -------------  ----------------------  --------------------------  -----------------  ------------  -----------  -----------  ----------  -------  --------------\n"  # noqa E501
+            "begin_date    end_date    source_ip    source_host    payload_from            envelop_from                dmarc_disposition  dkim_aligned  dkim_auth    spf_aligned  spf_auth         42  ruareport_file\n"  # noqa E501
         )
         actual = output.getvalue()
 

tests/test_csv_renderer.py

@@ -15,7 +15,7 @@ def test_render_without_data():
         sut = CSVRenderer()
         sut.render(csv_file, data)
 
-        expected = "Source IP,Source Host,Payload From (From:),Envelop From (MAIL FROM),DMARC,DKIM Align,DKIM Auth,SPF Align,SPF Auth,Count,File\n"  # noqa E501
+        expected = "Begin Date,End Date,Source IP,Source Host,Payload From (From:),Envelop From (MAIL FROM),DMARC,DKIM Align,DKIM Auth,SPF Align,SPF Auth,Count,File\n"  # noqa E501
         actual = open(csv_file, "r").read()
 
         assert expected == actual
@@ -31,6 +31,8 @@ def test_render_with_data():
 
         data = [
             [
+                "begin_date",
+                "end_date",
                 "source_ip",
                 "source_host",
                 "payload_from",
@@ -49,8 +51,8 @@ def test_render_with_data():
         sut.render(csv_file, data)
 
         expected = (
-            "Source IP,Source Host,Payload From (From:),Envelop From (MAIL FROM),DMARC,DKIM Align,DKIM Auth,SPF Align,SPF Auth,Count,File\n"  # noqa E501
-            "source_ip,source_host,payload_from,envelop_from,dmarc_disposition,dkim_aligned,dkim_auth,spf_aligned,spf_auth,42,ruareport_file\n"  # noqa E501
+            "Begin Date,End Date,Source IP,Source Host,Payload From (From:),Envelop From (MAIL FROM),DMARC,DKIM Align,DKIM Auth,SPF Align,SPF Auth,Count,File\n"  # noqa E501
+            "begin_date,end_date,source_ip,source_host,payload_from,envelop_from,dmarc_disposition,dkim_aligned,dkim_auth,spf_aligned,spf_auth,42,ruareport_file\n"  # noqa E501
         )
         actual = actual = open(csv_file, "r").read()