test: model unit tests updated

Author: clstaudt

tuttle_tests/test_model.py

@@ -80,51 +80,6 @@ def test_user():
     assert icloud_account.user.name == "Archibald Tuttle"
 
 
-def test_project():
-    project = model.Project(
-        title="Heating Repair",
-        tag="#heating-repair",
-        start_date=datetime.date.today(),
-        end_date=datetime.date.today() + datetime.timedelta(days=80),
-    )
-    assert store_and_retrieve(project)
-
-
-def test_contract():
-
-    the_client = model.Client(
-        name="Central Services",
-        invoicing_contact=model.Contact(
-            first_name="Central",
-            last_name="Services",
-            company="Central Services",
-            address=model.Address(
-                street="Down the Road",
-                number="55",
-                city="Somewhere",
-                postal_code="99999",
-                country="Brazil",
-            ),
-            email="[email protected]",
-        ),
-    )
-
-    the_contract = model.Contract(
-        title="CS Q1 2022",
-        client=the_client,
-        start_date=datetime.date(2022, 1, 1),
-        end_date=datetime.date(2022, 3, 31),
-        signature_date=datetime.date(2021, 10, 31),
-        rate=100,
-        unit=time.TimeUnit.hour,
-        currency="EUR",
-        billing_cycle=time.Cycle.monthly,
-        volume=3 * 8 * 8,
-        units_per_workday=8,
-    )
-    assert store_and_retrieve(the_contract)
-
-
 class TestContact:
     def test_valid_contact_instantiation(self):
         contact = Contact(
@@ -148,24 +103,61 @@ def test_invalid_email_instantiation(self):
 
 
 class TestClient:
+    """Tests for the Client model."""
+
     def test_valid_instantiation(self):
         invoicing_contact = Contact(
             first_name="Sam",
             last_name="Lowry",
             email="[email protected]",
             company="Ministry of Information",
         )
-        client = Client(
-            name="Ministry of Information", invoicing_contact=invoicing_contact
+        client = Client.validate(
+            dict(
+                name="Ministry of Information",
+                invoicing_contact=invoicing_contact,
+            )
         )
         assert store_and_retrieve(client)
 
     def test_missing_fields_instantiation(self):
         with pytest.raises(ValidationError):
-            Client()  # type: ignore
+            Client.validate(dict())
 
 
 class TestContract:
+    """Tests for the Contract model."""
+
+    def test_valid_instantiation(self):
+        client = Client(name="Ministry of Information")
+        contract = Contract.validate(
+            dict(
+                title="Project X Contract",
+                client=client,
+                signature_date=datetime.date(2022, 10, 1),
+                start_date=datetime.date(2022, 10, 2),
+                end_date=datetime.date(2022, 12, 31),
+                rate=100,
+                is_completed=False,
+                currency="USD",
+                VAT_rate=0.19,
+                unit=TimeUnit.hour,
+                units_per_workday=8,
+                volume=100,
+                term_of_payment=31,
+                billing_cycle=Cycle.monthly,
+            )
+        )
+        assert store_and_retrieve(contract)
+
+    def test_missing_fields_instantiation(self):
+        with pytest.raises(ValidationError):
+            Contract.validate(dict())
+
+
+class TestProject:
+    """Tests for the Project model."""
+
     def test_valid_instantiation(self):
         client = Client(name="Ministry of Information")
         contract = Contract(
@@ -184,8 +176,30 @@ def test_valid_instantiation(self):
             term_of_payment=31,
             billing_cycle=Cycle.monthly,
         )
-        assert store_and_retrieve(contract)
+        project = Project.validate(
+            dict(
+                title="Project X",
+                description="The description of Project X",
+                tag="#project_x",
+                start_date=datetime.date(2022, 10, 2),
+                end_date=datetime.date(2022, 12, 31),
+                contract=contract,
+            )
+        )
+        assert store_and_retrieve(project)
 
     def test_missing_fields_instantiation(self):
         with pytest.raises(ValidationError):
-            Contract()  # type: ignore
+            Project.validate(dict())
+
+    def test_invalid_tag_instantiation(self):
+        with pytest.raises(ValidationError):
+            Project.validate(
+                dict(
+                    title="Project X",
+                    description="The description of Project X",
+                    tag="project_x",
+                    start_date=datetime.date(2022, 10, 2),
+                    end_date=datetime.date(2022, 12, 31),
+                )
+            )