mirror of
https://github.com/RunLit/Bambu-Run.git
synced 2026-10-21 15:37:38 +01:00
100 lines
3.8 KiB
Python
100 lines
3.8 KiB
Python
import pytest
|
|
from django.utils import timezone
|
|
|
|
from bambu_run.models import Filament, Printer, PrintJob, FilamentUsage
|
|
from bambu_run.mcp.filaments import list_filaments
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_list_filaments_filters_material_sub_type_color():
|
|
Filament.objects.create(type="PLA", sub_type="PLA Matte", brand="Bambu Lab", color="Black", remaining_percent=50)
|
|
Filament.objects.create(type="PLA", sub_type="PLA Basic", brand="Bambu Lab", color="Black", remaining_percent=50)
|
|
Filament.objects.create(type="ABS", sub_type="ABS", brand="Bambu Lab", color="Black", remaining_percent=50)
|
|
|
|
result = list_filaments(material="PLA", sub_type="PLA Matte", color="black")
|
|
|
|
assert len(result["filaments"]) == 1
|
|
assert result["filaments"][0]["sub_type"] == "PLA Matte"
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_list_filaments_min_remaining_grams_excludes_unread_sentinel():
|
|
Filament.objects.create(type="ABS", brand="Bambu Lab", color="Mint", remaining_percent=-1, remaining_weight_grams=-10)
|
|
Filament.objects.create(type="ABS", brand="Bambu Lab", color="Orange", remaining_percent=68, remaining_weight_grams=680)
|
|
|
|
result = list_filaments(material="ABS", min_remaining_grams=250)
|
|
|
|
assert len(result["filaments"]) == 1
|
|
assert result["filaments"][0]["color_name"] == "Orange"
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_list_filaments_min_remaining_grams_excludes_unread_even_with_low_threshold():
|
|
# The existing test only checks min_remaining_grams=250, which happens to
|
|
# exclude a -10g sentinel-derived value anyway. This test uses a threshold
|
|
# low enough that a naive filter would let the sentinel spool through.
|
|
Filament.objects.create(
|
|
type="ABS", brand="Bambu Lab", color="Mint",
|
|
remaining_percent=-1, remaining_weight_grams=-10,
|
|
)
|
|
Filament.objects.create(
|
|
type="ABS", brand="Bambu Lab", color="Orange",
|
|
remaining_percent=68, remaining_weight_grams=680,
|
|
)
|
|
|
|
result = list_filaments(material="ABS", min_remaining_grams=-20)
|
|
|
|
assert len(result["filaments"]) == 1
|
|
assert result["filaments"][0]["color_name"] == "Orange"
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_list_filaments_unread_sentinel_status_not_low():
|
|
Filament.objects.create(type="ABS", brand="Bambu Lab", color="Mint", remaining_percent=-1)
|
|
|
|
result = list_filaments(material="ABS")
|
|
|
|
spool = result["filaments"][0]
|
|
assert spool["remaining_percent"] is None
|
|
assert spool["remaining_status"] == "unread"
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_list_filaments_limit_hard_ceiling_100():
|
|
Filament.objects.bulk_create([
|
|
Filament(type="PLA", brand="Bambu Lab", color=f"Color {i}", remaining_percent=50)
|
|
for i in range(150)
|
|
])
|
|
|
|
result = list_filaments(limit=1000)
|
|
|
|
assert len(result["filaments"]) == 100
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_list_filaments_include_history_adds_recent_usage():
|
|
f = Filament.objects.create(type="PLA", brand="Bambu Lab", color="White", initial_weight_grams=1000, remaining_percent=80)
|
|
p = Printer.objects.create(name="RNL-H2C", model="H2C", is_active=True)
|
|
job = PrintJob.objects.create(device=p, project_name="Vase", start_time=timezone.now(), final_status="FINISH")
|
|
FilamentUsage.objects.create(print_job=job, filament=f, tray_id=0, starting_percent=90, ending_percent=80)
|
|
|
|
result = list_filaments(material="PLA", include_history=True)
|
|
|
|
history = result["filaments"][0]["recent_usage"]
|
|
assert history[0]["job_name"] == "Vase"
|
|
assert history[0]["consumed_grams"] == 100
|
|
|
|
|
|
@pytest.mark.django_db
|
|
def test_list_filaments_never_exposes_rfid_identifiers():
|
|
Filament.objects.create(
|
|
type="PLA", brand="Bambu Lab", color="White",
|
|
tray_uuid="SECRET-UUID", tag_uid="SECRET-TAG",
|
|
)
|
|
|
|
result = list_filaments(material="PLA")
|
|
|
|
entry_str = repr(result["filaments"][0])
|
|
assert "SECRET-UUID" not in entry_str
|
|
assert "SECRET-TAG" not in entry_str
|