test: unit tests passing

Author: clstaudt

tuttle/calendar.py

@@ -4,6 +4,7 @@
 from pathlib import Path
 import io
 import re
+import calendar
 
 from loguru import logger
 import ics
@@ -180,3 +181,17 @@ class GoogleCalendar(CloudCalendar):
 
     def to_data(self) -> DataFrame:
         raise NotImplementedError("TODO")
+
+
+def get_month_start_end(month_str):
+    # Parse the string into a datetime object
+    dt = datetime.datetime.strptime(month_str, "%B %Y")
+
+    # Get the date information from the datetime object
+    year, month = dt.date().year, dt.date().month
+
+    # Get the start and end dates of the month
+    start_date = datetime.date(year, month, 1)
+    end_date = datetime.date(year, month, calendar.monthrange(year, month)[1])
+
+    return start_date, end_date

tuttle_tests/test_invoice.py

@@ -5,6 +5,7 @@
 
 from tuttle import invoicing, timetracking, rendering
 from tuttle.model import Invoice, InvoiceItem
+from tuttle.calendar import get_month_start_end
 
 
 def test_invoice():
@@ -50,10 +51,12 @@ def test_generate_invoice(
     for project in demo_projects:
         timesheets = []
         for period in ["January 2022", "February 2022"]:
+            (period_start, period_end) = get_month_start_end(period)
             timesheet = timetracking.generate_timesheet(
-                source=demo_calendar_timetracking,
+                timetracking_data=demo_calendar_timetracking.to_data(),
                 project=project,
-                period_start=period,
+                period_start=period_start,
+                period_end=period_end,
                 item_description=project.title,
             )
             if not timesheet.empty:
@@ -65,67 +68,3 @@ def test_generate_invoice(
             date=datetime.date.today(),
         )
         # assert invoice.total > 0
-
-
-def test_render_invoice_to_html(
-    demo_user,
-    demo_projects,
-    demo_calendar_timetracking,
-):
-    for project in demo_projects:
-        timesheets = []
-        for period in ["January 2022", "February 2022"]:
-            timesheet = timetracking.generate_timesheet(
-                source=demo_calendar_timetracking,
-                project=project,
-                period_start=period,
-                item_description=project.title,
-            )
-            if not timesheet.empty:
-                timesheets.append(timesheet)
-        invoice = invoicing.generate_invoice(
-            timesheets=timesheets,
-            contract=project.contract,
-            project=project,
-            date=datetime.date.today(),
-        )
-        # RENDERING
-        rendering.render_invoice(
-            user=demo_user,
-            invoice=invoice,
-            style="anvil",
-            document_format="html",
-            out_dir=Path("tuttle_tests/tmp"),
-        )
-
-
-def test_render_invoice_to_pdf(
-    demo_user,
-    demo_projects,
-    demo_calendar_timetracking,
-):
-    for project in demo_projects:
-        timesheets = []
-        for period in ["January 2022", "February 2022"]:
-            timesheet = timetracking.generate_timesheet(
-                source=demo_calendar_timetracking,
-                project=project,
-                period_start=period,
-                item_description=project.title,
-            )
-            if not timesheet.empty:
-                timesheets.append(timesheet)
-        invoice = invoicing.generate_invoice(
-            timesheets=timesheets,
-            contract=project.contract,
-            project=project,
-            date=datetime.date.today(),
-        )
-        # RENDERING
-        rendering.render_invoice(
-            user=demo_user,
-            invoice=invoice,
-            style="anvil",
-            document_format="pdf",
-            out_dir=Path("tuttle_tests/tmp"),
-        )

tuttle_tests/test_rendering.py

@@ -16,7 +16,6 @@ class TestRenderTimesheet:
     """Tests for render_timesheet"""
 
     def test_returns_html_when_out_dir_is_none(self, fake):
-
         user = demo.create_fake_user(fake)
         timesheet = demo.create_fake_timesheet(fake)
         document_format = "html"

tuttle_tests/test_timetracking.py

@@ -4,6 +4,7 @@
 import datetime
 
 from tuttle import timetracking
+from tuttle.calendar import get_month_start_end
 
 
 def test_timetracking_import_toggl():
@@ -27,11 +28,13 @@ def test_generate_timesheet_from_demo_calendar(
     demo_calendar_timetracking,
 ):
     for period in ["January 2022", "February 2022"]:
+        (period_start, period_end) = get_month_start_end(period)
         for project in demo_projects:
             timesheet = timetracking.generate_timesheet(
-                source=demo_calendar_timetracking,
+                timetracking_data=demo_calendar_timetracking.to_data(),
                 project=project,
-                period_start=period,
+                period_start=period_start,
+                period_end=period_end,
                 item_description=project.title,
             )
             assert (timesheet.empty) or (timesheet.total >= pandas.Timedelta("0 hours"))