test: invoice tests updated

Author: clstaudt

tuttle/model.py

@@ -619,24 +619,14 @@ def sum(self) -> Decimal:
     def VAT_total(self) -> Decimal:
         """Sum of VAT over all invoice items."""
         s = sum(item.VAT for item in self.items)
-        return Decimal(s)
+        return Decimal(round(s, 2))
 
     @property
     def total(self) -> Decimal:
         """Total invoiced amount."""
         t = self.sum + self.VAT_total
         return Decimal(t)
 
-    def generate_number(self, pattern=None, counter=None) -> str:
-        """Generate an invoice number"""
-        date_prefix = self.date.strftime("%Y-%m-%d")
-        # suffix = hashlib.shake_256(str(uuid.uuid4()).encode("utf-8")).hexdigest(2)
-        # TODO: auto-increment suffix for invoices generated on the same day
-        if counter is None:
-            counter = 1
-        suffix = f"{counter:02}"
-        self.number = f"{date_prefix}-{suffix}"
-
     @property
     def due_date(self) -> Optional[datetime.date]:
         """Date until which payment is due."""

tuttle_tests/test_invoice.py

@@ -1,6 +1,7 @@
 """Tests for the invoice module."""
 
 import datetime
+from decimal import Decimal
 from pathlib import Path
 
 from tuttle import invoicing, timetracking, rendering
@@ -12,36 +13,39 @@ def test_invoice():
     the_invoice = Invoice(
         number="27B-6",
         date=datetime.date.today(),
-        due_date=datetime.date.today() + datetime.timedelta(days=14),
-        sent_date=datetime.date.today(),
         sent=True,
-        paid="foo",
+        paid=False,
         cancelled=False,
     )
 
     item_1 = InvoiceItem(
         invoice=the_invoice,
-        date=datetime.date.today(),
+        start_date=datetime.date.today(),
+        end_date=datetime.date.today(),
         quantity=10,
         unit="hours",
-        unit_price=50,
+        unit_price=Decimal(50),
         description="work work",
-        VAT_rate=0.20,
+        VAT_rate=Decimal(0.20),
     )
 
     item_2 = InvoiceItem(
         invoice=the_invoice,
-        date=datetime.date.today(),
+        start_date=datetime.date.today(),
+        end_date=datetime.date.today(),
         quantity=10,
         unit="hours",
-        unit_price=100,
+        unit_price=Decimal(100),
         description="work work",
-        VAT_rate=0.20,
+        VAT_rate=Decimal(0.20),
     )
 
-    assert the_invoice.sum == 1500
-    assert the_invoice.VAT_total == 300
-    assert the_invoice.total == 1800
+    assert item_1.invoice == the_invoice
+    assert item_2.invoice == the_invoice
+
+    assert the_invoice.sum == Decimal(1500)
+    assert the_invoice.VAT_total == Decimal(300)
+    assert the_invoice.total == Decimal(1800)
 
 
 def test_generate_invoice(

tuttle_tests/test_model.py

@@ -64,22 +64,6 @@ def test_model_creation():
             pass
 
 
-def test_user():
-    user = model.User(
-        name="Archibald Tuttle",
-        subtitle="Heating Engineer",
-        email="harry@tuttle.com",
-    )
-
-    icloud_account = model.ICloudAccount(
-        user_name=user.email,
-    )
-
-    user.icloud_account = icloud_account
-
-    assert icloud_account.user.name == "Archibald Tuttle"
-
-
 class TestUser:
     """Tests for the User model."""