Added additional integration tests for measure and add

2024-03-28 09:29:39 -04:00
parent 19c414da4b
commit a6ae808389
4 changed files with 204 additions and 5 deletions
--- a/tests_integration/tests/spool/test_add.py
+++ b/tests_integration/tests/spool/test_add.py
@@ -195,3 +195,145 @@ def test_add_spool_both_used_and_remaining_weight(random_filament: dict[str, Any
        },
    )
    assert result.status_code == 400  # Cannot set both used and remaining weight
+
+
+def test_add_spool_initial_weight(random_filament: dict[str, Any]):
+    """Test adding a spool to the database."""
+    # Execute
+    remaining_weight = 750
+    initial_weight = 1245
+    location = "The Pantry"
+    lot_nr = "123456789"
+    comment = "abcdefghåäö"
+    archived = True
+    price = 25
+    result = httpx.post(
+        f"{URL}/api/v1/spool",
+        json={
+            "first_used": "2023-01-02T12:00:00+01:00",
+            "last_used": "2023-01-02T11:00:00Z",
+            "filament_id": random_filament["id"],
+            "remaining_weight": remaining_weight,
+            "initial_weight": initial_weight,
+            "location": location,
+            "lot_nr": lot_nr,
+            "comment": comment,
+            "archived": archived,
+            "price": price,
+        },
+    )
+    result.raise_for_status()
+
+    # Verify
+    used_weight = initial_weight - random_filament["spool_weight"] - remaining_weight
+    used_length = length_from_weight(
+        weight=used_weight,
+        density=random_filament["density"],
+        diameter=random_filament["diameter"],
+    )
+    remaining_length = length_from_weight(
+        weight=remaining_weight,
+        density=random_filament["density"],
+        diameter=random_filament["diameter"],
+    )
+
+    spool = result.json()
+    assert_dicts_compatible(
+        spool,
+        {
+            "id": spool["id"],
+            "registered": spool["registered"],
+            "first_used": "2023-01-02T11:00:00Z",
+            "last_used": "2023-01-02T11:00:00Z",
+            "filament": random_filament,
+            "initial_weight": pytest.approx(initial_weight),
+            "empty_weight": pytest.approx(random_filament["spool_weight"]),
+            "remaining_weight": pytest.approx(remaining_weight),
+            "used_weight": pytest.approx(used_weight),
+            "remaining_length": pytest.approx(remaining_length),
+            "used_length": pytest.approx(used_length),
+            "location": location,
+            "lot_nr": lot_nr,
+            "comment": comment,
+            "archived": archived,
+            "price": price,
+        },
+    )
+
+    # Verify that registered happened almost now (within 1 minute)
+    diff = abs((datetime.now(tz=timezone.utc) - datetime.fromisoformat(spool["registered"])).total_seconds())
+    assert diff < 60
+
+    # Clean up
+    httpx.delete(f"{URL}/api/v1/spool/{spool['id']}").raise_for_status()
+
+
+def test_add_spool_empty_weight(random_filament: dict[str, Any]):
+    """Test adding a spool to the database."""
+    # Execute
+    remaining_weight = 750
+    empty_weight = 200
+    location = "The Pantry"
+    lot_nr = "123456789"
+    comment = "abcdefghåäö"
+    archived = True
+    price = 25
+    result = httpx.post(
+        f"{URL}/api/v1/spool",
+        json={
+            "first_used": "2023-01-02T12:00:00+01:00",
+            "last_used": "2023-01-02T11:00:00Z",
+            "filament_id": random_filament["id"],
+            "remaining_weight": remaining_weight,
+            "empty_weight": empty_weight,
+            "location": location,
+            "lot_nr": lot_nr,
+            "comment": comment,
+            "archived": archived,
+            "price": price,
+        },
+    )
+    result.raise_for_status()
+
+    # Verify
+    used_weight = random_filament["weight"] - remaining_weight
+    used_length = length_from_weight(
+        weight=used_weight,
+        density=random_filament["density"],
+        diameter=random_filament["diameter"],
+    )
+    remaining_length = length_from_weight(
+        weight=remaining_weight,
+        density=random_filament["density"],
+        diameter=random_filament["diameter"],
+    )
+
+    spool = result.json()
+    assert_dicts_compatible(
+        spool,
+        {
+            "id": spool["id"],
+            "registered": spool["registered"],
+            "first_used": "2023-01-02T11:00:00Z",
+            "last_used": "2023-01-02T11:00:00Z",
+            "filament": random_filament,
+            "initial_weight": pytest.approx(random_filament["weight"] + empty_weight),
+            "empty_weight": pytest.approx(empty_weight),
+            "remaining_weight": pytest.approx(remaining_weight),
+            "used_weight": pytest.approx(used_weight),
+            "remaining_length": pytest.approx(remaining_length),
+            "used_length": pytest.approx(used_length),
+            "location": location,
+            "lot_nr": lot_nr,
+            "comment": comment,
+            "archived": archived,
+            "price": price,
+        },
+    )
+
+    # Verify that registered happened almost now (within 1 minute)
+    diff = abs((datetime.now(tz=timezone.utc) - datetime.fromisoformat(spool["registered"])).total_seconds())
+    assert diff < 60
+
+    # Clean up
+    httpx.delete(f"{URL}/api/v1/spool/{spool['id']}").raise_for_status()