Files
bambuddy/backend/tests/unit/test_github_backup_spool_fields.py

48 lines
1.5 KiB
Python

"""The spool collector's field whitelist for Git backup (#2870).
``_collect_spools`` builds an explicit dict per spool, so anything missing
from it is silently absent from the backup and therefore lost on restore.
The owner's own bookkeeping — purchasing number (#2870), category and
low-stock override (#729), free-text storage — belongs in the file.
"""
import pytest
from backend.app.models.spool import Spool
from backend.app.services.github_backup import GitHubBackupService
@pytest.mark.asyncio
async def test_collects_the_owners_own_bookkeeping_fields(db_session):
db_session.add(
Spool(
material="PLA",
material_number="15",
category="Production",
low_stock_threshold_pct=40,
storage_location="Shelf B",
)
)
await db_session.commit()
files: dict = {}
await GitHubBackupService()._collect_spools(db_session, files)
entry = files["spools/inventory.json"]["spools"][0]
assert entry["material_number"] == "15"
assert entry["category"] == "Production"
assert entry["low_stock_threshold_pct"] == 40
assert entry["storage_location"] == "Shelf B"
@pytest.mark.asyncio
async def test_location_id_is_left_out(db_session):
"""The locations table is not backed up, so the ID has nothing to mean."""
db_session.add(Spool(material="PLA"))
await db_session.commit()
files: dict = {}
await GitHubBackupService()._collect_spools(db_session, files)
assert "location_id" not in files["spools/inventory.json"]["spools"][0]