| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819 |
- """Integration tests for Print Queue API endpoints."""
- import pytest
- from httpx import AsyncClient
- from sqlalchemy import select
- from backend.app.models.finance import CostCenter
- from backend.app.models.settings import Settings
- async def enable_billing(db_session):
- setting = await db_session.scalar(select(Settings).where(Settings.key == "billing_enabled"))
- if setting is None:
- db_session.add(Settings(key="billing_enabled", value="true"))
- else:
- setting.value = "true"
- await db_session.commit()
- class TestPrintQueueAPI:
- """Integration tests for /api/v1/queue endpoints."""
- @pytest.fixture
- async def printer_factory(self, db_session):
- """Factory to create test printers."""
- _counter = [0]
- async def _create_printer(**kwargs):
- from backend.app.models.printer import Printer
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "name": f"Test Printer {counter}",
- "ip_address": f"192.168.1.{100 + counter}",
- "serial_number": f"TESTSERIAL{counter:04d}",
- "access_code": "12345678",
- "model": "X1C",
- }
- defaults.update(kwargs)
- printer = Printer(**defaults)
- db_session.add(printer)
- await db_session.commit()
- await db_session.refresh(printer)
- return printer
- return _create_printer
- @pytest.fixture
- async def archive_factory(self, db_session):
- """Factory to create test archives."""
- _counter = [0]
- async def _create_archive(**kwargs):
- from backend.app.models.archive import PrintArchive
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "filename": f"test_print_{counter}.3mf",
- "print_name": f"Test Print {counter}",
- "file_path": f"/tmp/test_print_{counter}.3mf",
- "file_size": 1024,
- "content_hash": f"testhash{counter:08d}",
- "status": "completed",
- }
- defaults.update(kwargs)
- archive = PrintArchive(**defaults)
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- return archive
- return _create_archive
- @pytest.fixture
- async def queue_item_factory(self, db_session, printer_factory, archive_factory):
- """Factory to create test queue items."""
- _counter = [0]
- async def _create_queue_item(**kwargs):
- from backend.app.models.print_queue import PrintQueueItem
- _counter[0] += 1
- counter = _counter[0]
- # Create printer and archive if not provided
- if "printer_id" not in kwargs:
- printer = await printer_factory()
- kwargs["printer_id"] = printer.id
- if "archive_id" not in kwargs:
- archive = await archive_factory()
- kwargs["archive_id"] = archive.id
- defaults = {
- "status": "pending",
- "position": counter,
- }
- defaults.update(kwargs)
- item = PrintQueueItem(**defaults)
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- return item
- return _create_queue_item
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_list_queue_empty(self, async_client: AsyncClient):
- """Verify empty list when no queue items exist."""
- response = await async_client.get("/api/v1/queue/")
- assert response.status_code == 200
- assert isinstance(response.json(), list)
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue(self, async_client: AsyncClient, printer_factory, archive_factory, db_session):
- """Verify item can be added to queue."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["printer_id"] == printer.id
- assert result["archive_id"] == archive.id
- assert result["status"] == "pending"
- assert result["manual_start"] is False
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_with_cost_center_id(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify item can be added to queue with cost_center_id."""
- await enable_billing(db_session)
- printer = await printer_factory()
- archive = await archive_factory(cost=1.25, filament_used_grams=50.0)
- cost_center = CostCenter(name="Queue CC", is_active=True, is_private=False)
- db_session.add(cost_center)
- await db_session.commit()
- await db_session.refresh(cost_center)
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": archive.id,
- "cost_center_id": cost_center.id,
- "estimated_cost": 0.01,
- },
- )
- assert response.status_code == 200
- result = response.json()
- assert result["cost_center_id"] == cost_center.id
- assert result["estimated_cost"] == 1.25
- from backend.app.models.print_queue import PrintQueueItem
- row = await db_session.scalar(select(PrintQueueItem).where(PrintQueueItem.id == result["id"]))
- assert row is not None
- assert row.cost_center_id == cost_center.id
- assert row.estimated_cost == 1.25
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_derives_cost_without_client_estimate(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """The persisted budget estimate comes from the archive, not the request."""
- await enable_billing(db_session)
- printer = await printer_factory()
- archive = await archive_factory(cost=1.25, filament_used_grams=50.0)
- cost_center = CostCenter(name="Budget CC", is_active=True, is_private=False, monthly_budget=10.0)
- db_session.add(cost_center)
- await db_session.commit()
- await db_session.refresh(cost_center)
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": archive.id,
- "cost_center_id": cost_center.id,
- },
- )
- assert response.status_code == 200
- assert response.json()["estimated_cost"] == 1.25
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_model_based_queue_item_derives_cost_without_client_estimate(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Model dispatch has no printer-side estimate but remains billable."""
- await enable_billing(db_session)
- await printer_factory(model="X1C")
- archive = await archive_factory(cost=1.25, filament_used_grams=50.0, sliced_for_model="X1C")
- cost_center = CostCenter(name="Model Budget CC", is_active=True, is_private=False, monthly_budget=10.0)
- db_session.add(cost_center)
- await db_session.commit()
- await db_session.refresh(cost_center)
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "target_model": "X1C",
- "archive_id": archive.id,
- "cost_center_id": cost_center.id,
- },
- )
- assert response.status_code == 200
- assert response.json()["printer_id"] is None
- assert response.json()["estimated_cost"] == 1.25
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_rejects_tampered_client_cost_when_server_cost_exceeds_budget(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """A forged low client hint cannot bypass the server-derived budget check."""
- await enable_billing(db_session)
- printer = await printer_factory()
- archive = await archive_factory(cost=2.0, filament_used_grams=50.0)
- cost_center = CostCenter(name="Tiny Budget CC", is_active=True, is_private=False, monthly_budget=1.0)
- db_session.add(cost_center)
- await db_session.commit()
- await db_session.refresh(cost_center)
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": archive.id,
- "cost_center_id": cost_center.id,
- "estimated_cost": 0.01,
- },
- )
- assert response.status_code == 400
- assert "exceeds available cost center budget" in response.json()["detail"]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_requires_cost_center_when_billing_enabled(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Billing enforcement rejects queue jobs that omit cost center."""
- await enable_billing(db_session)
- printer = await printer_factory()
- archive = await archive_factory(cost=1.0, filament_used_grams=50.0)
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": archive.id,
- },
- )
- assert response.status_code == 400
- assert "Cost center is required" in response.json()["detail"]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_counts_pending_queue_reservations_against_budget(
- self, async_client: AsyncClient, printer_factory, archive_factory, queue_item_factory, db_session
- ):
- """Open queue items reserve budget until they leave pending/printing states."""
- await enable_billing(db_session)
- printer = await printer_factory()
- archive = await archive_factory(cost=3.0, filament_used_grams=50.0)
- cost_center = CostCenter(name="Reserved Budget CC", is_active=True, is_private=False, monthly_budget=10.0)
- db_session.add(cost_center)
- await db_session.commit()
- await db_session.refresh(cost_center)
- await queue_item_factory(
- printer_id=printer.id,
- archive_id=archive.id,
- cost_center_id=cost_center.id,
- estimated_cost=8.0,
- status="pending",
- )
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": archive.id,
- "cost_center_id": cost_center.id,
- "estimated_cost": 0.01,
- },
- )
- assert response.status_code == 400
- assert "exceeds available cost center budget" in response.json()["detail"]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_with_manual_start(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify item can be added to queue with manual_start=True."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "manual_start": True,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["printer_id"] == printer.id
- assert result["archive_id"] == archive.id
- assert result["status"] == "pending"
- assert result["manual_start"] is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_with_skip_filament_check(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """PrintModal "Print Anyway" persists skip_filament_check on creation (#1698-followup)."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "skip_filament_check": True,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["skip_filament_check"] is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_skip_filament_check_defaults_false(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Default add-to-queue has skip_filament_check=False — no silent bypass."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {"printer_id": printer.id, "archive_id": archive.id}
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["skip_filament_check"] is False
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_update_queue_item_cost_center_id(
- self, async_client: AsyncClient, printer_factory, archive_factory, queue_item_factory, db_session
- ):
- """Verify a pending queue item can be updated with cost_center_id."""
- await enable_billing(db_session)
- printer = await printer_factory()
- archive = await archive_factory(cost=1.25, filament_used_grams=50.0)
- item = await queue_item_factory(printer_id=printer.id, archive_id=archive.id)
- cost_center = CostCenter(name="Update CC", is_active=True, is_private=False)
- db_session.add(cost_center)
- await db_session.commit()
- await db_session.refresh(cost_center)
- response = await async_client.patch(
- f"/api/v1/queue/{item.id}",
- json={"cost_center_id": cost_center.id, "estimated_cost": 0.01},
- )
- assert response.status_code == 200
- assert response.json()["cost_center_id"] == cost_center.id
- assert response.json()["estimated_cost"] == 1.25
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_with_project_id(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """#932: queue items created from the project view carry project_id forward."""
- from backend.app.models.project import Project
- printer = await printer_factory()
- archive = await archive_factory()
- project = Project(name="Queue Project")
- db_session.add(project)
- await db_session.commit()
- await db_session.refresh(project)
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "project_id": project.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- # The response schema may or may not echo project_id; the stored row is
- # what matters, so verify via DB.
- from sqlalchemy import select
- from backend.app.models.print_queue import PrintQueueItem
- row = (await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.id == result["id"]))).scalar_one()
- assert row.project_id == project.id
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_invalid_project_id_returns_404(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """#932: bogus project_id must be rejected before the FK constraint fires.
- Regression guard for the pre-check added to add_to_queue. Without the
- validation, a nonexistent project_id would reach db.commit() and raise
- an IntegrityError → 500. The pre-check must convert that to a 404 so
- the UI gets a clean error it can surface.
- """
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "project_id": 999999, # nonexistent
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 404
- assert "project" in response.json()["detail"].lower()
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_with_ams_mapping(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify item can be added to queue with ams_mapping."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "ams_mapping": [5, -1, 2, -1], # Slot 1 -> tray 5, slot 3 -> tray 2
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["printer_id"] == printer.id
- assert result["archive_id"] == archive.id
- assert result["ams_mapping"] == [5, -1, 2, -1]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_falls_back_to_archive_slicer_ams_mapping_when_unset(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """When the caller sends no explicit ams_mapping, but the archive
- carries the slicer's own saved pick for this exact printer
- (extra_data.slicer_ams_mapping, written by a VP with "Save AMS
- mapping" on), the queue item should inherit it — the same
- exact-physical-spool reuse the "Mapping" button gives you, but
- automatic when nothing was hand-edited.
- """
- printer = await printer_factory()
- archive = await archive_factory(
- extra_data={"slicer_ams_mapping": {"mapping": [5, -1, 2, -1], "printer_id": printer.id}}
- )
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["ams_mapping"] == [5, -1, 2, -1]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_ignores_archive_slicer_ams_mapping_for_different_printer(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """A saved mapping's tray IDs only mean something relative to the
- printer they were resolved against. Reprinting the same archive on a
- *different* printer must not inherit it — tray 5 on printer A can
- hold a completely different spool than tray 5 on printer B.
- """
- origin_printer = await printer_factory()
- other_printer = await printer_factory()
- archive = await archive_factory(
- extra_data={"slicer_ams_mapping": {"mapping": [5, -1, 2, -1], "printer_id": origin_printer.id}}
- )
- data = {
- "printer_id": other_printer.id,
- "archive_id": archive.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["ams_mapping"] is None
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_ignores_archive_slicer_ams_mapping_for_model_based_dispatch(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """A model-based item (no fixed printer_id) can't know in advance
- which printer the scheduler will pick, so a saved mapping resolved
- against one specific printer must never be inherited here either.
- """
- origin_printer = await printer_factory()
- archive = await archive_factory(
- extra_data={"slicer_ams_mapping": {"mapping": [5, -1, 2, -1], "printer_id": origin_printer.id}}
- )
- data = {
- "target_model": "X1C",
- "archive_id": archive.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["ams_mapping"] is None
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_explicit_ams_mapping_wins_over_archive_fallback(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """An explicit ams_mapping in the request (e.g. from the filament
- mapping panel) must take priority over the archive's saved slicer
- pick — the fallback only fires when the caller sent nothing at all.
- """
- printer = await printer_factory()
- archive = await archive_factory(
- extra_data={"slicer_ams_mapping": {"mapping": [5, -1, 2, -1], "printer_id": printer.id}}
- )
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "ams_mapping": [9, -1, 1, -1],
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["ams_mapping"] == [9, -1, 1, -1]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_archive_extra_data_without_slicer_mapping_key_not_used(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """extra_data present but without a slicer_ams_mapping key (the
- common case — most archives have other metadata but no saved slicer
- mapping) must not accidentally trip the fallback."""
- printer = await printer_factory()
- archive = await archive_factory(extra_data={"filament_slots": []})
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["ams_mapping"] is None
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_force_color_match_overrides_beat_the_archive_fallback(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Force-color-match overrides are the caller asking the scheduler to
- match strictly against the printer's live trays, and they are only ever
- applied inside `_compute_ams_mapping_for_printer` — the function a
- stored mapping makes the scheduler skip. Inheriting the saved mapping
- here would silently retire the strictness that was just requested
- (#2700 review).
- """
- printer = await printer_factory()
- archive = await archive_factory(
- extra_data={"slicer_ams_mapping": {"mapping": [5, -1, 2, -1], "printer_id": printer.id}}
- )
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "filament_overrides": [
- {"slot_id": 1, "type": "PLA", "color": "#FF0000", "force_color_match": True},
- ],
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["ams_mapping"] is None
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_plain_overrides_still_allow_the_archive_fallback(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Only force_color_match stands the fallback down. A plain preference
- override is a filament swap, not a request for live colour matching, so
- the saved mapping is still the best starting point.
- """
- printer = await printer_factory()
- archive = await archive_factory(
- extra_data={"slicer_ams_mapping": {"mapping": [5, -1, 2, -1], "printer_id": printer.id}}
- )
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "filament_overrides": [{"slot_id": 1, "type": "PLA", "color": "#FF0000"}],
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["ams_mapping"] == [5, -1, 2, -1]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_queue_response_flags_saved_mapping_only_for_its_own_printer(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """`archive_has_slicer_ams_mapping` drives a badge that claims the
- print reuses the slicer's exact trays. Global tray IDs mean nothing on
- another printer, so the flag must be false for a row targeting one —
- otherwise the badge is there while nothing is reused (#2700 review).
- """
- origin_printer = await printer_factory()
- other_printer = await printer_factory()
- archive = await archive_factory(
- extra_data={"slicer_ams_mapping": {"mapping": [5, -1, 2, -1], "printer_id": origin_printer.id}}
- )
- own = await async_client.post(
- "/api/v1/queue/", json={"printer_id": origin_printer.id, "archive_id": archive.id}
- )
- assert own.status_code == 200
- assert own.json()["archive_has_slicer_ams_mapping"] is True
- foreign = await async_client.post(
- "/api/v1/queue/", json={"printer_id": other_printer.id, "archive_id": archive.id}
- )
- assert foreign.status_code == 200
- assert foreign.json()["archive_has_slicer_ams_mapping"] is False
- # Model-based: the scheduler hasn't picked a printer yet, so the
- # mapping is not reused there either.
- model_based = await async_client.post("/api/v1/queue/", json={"target_model": "X1C", "archive_id": archive.id})
- assert model_based.status_code == 200
- assert model_based.json()["archive_has_slicer_ams_mapping"] is False
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_with_plate_id(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify item can be added to queue with plate_id for multi-plate 3MF."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "plate_id": 3,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["plate_id"] == 3
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_with_print_options(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify item can be added to queue with print options."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "bed_levelling": "off",
- "flow_cali": "on",
- "vibration_cali": False,
- "layer_inspect": True,
- "timelapse": True,
- "use_ams": False,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["bed_levelling"] == "off"
- assert result["flow_cali"] == "on"
- assert result["vibration_cali"] is False
- assert result["layer_inspect"] is True
- assert result["timelapse"] is True
- assert result["use_ams"] is False
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_update_queue_item_plate_id(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify queue item plate_id can be updated."""
- item = await queue_item_factory()
- response = await async_client.patch(f"/api/v1/queue/{item.id}", json={"plate_id": 5})
- assert response.status_code == 200
- result = response.json()
- assert result["plate_id"] == 5
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_update_queue_item_print_options(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify queue item print options can be updated."""
- item = await queue_item_factory()
- response = await async_client.patch(
- f"/api/v1/queue/{item.id}",
- json={
- "bed_levelling": "off",
- "timelapse": True,
- },
- )
- assert response.status_code == 200
- result = response.json()
- assert result["bed_levelling"] == "off"
- assert result["timelapse"] is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_reassign_rejected_while_dispatching(
- self, async_client: AsyncClient, queue_item_factory, printer_factory, db_session
- ):
- """#2615: a claimed (in-flight) row rejects edits with 409, so its printer
- can't be reassigned out from under the running FTP upload."""
- from datetime import datetime, timezone
- item = await queue_item_factory(dispatching_at=datetime.now(timezone.utc))
- other = await printer_factory()
- original_printer_id = item.printer_id
- response = await async_client.patch(f"/api/v1/queue/{item.id}", json={"printer_id": other.id})
- assert response.status_code == 409
- await db_session.refresh(item)
- assert item.printer_id == original_printer_id, "printer_id must not change on a dispatching row"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bulk_update_skips_dispatching_item(
- self, async_client: AsyncClient, queue_item_factory, printer_factory, db_session
- ):
- """#2615: bulk edits skip a claimed row rather than splitting it."""
- from datetime import datetime, timezone
- item = await queue_item_factory(dispatching_at=datetime.now(timezone.utc))
- other = await printer_factory()
- original_printer_id = item.printer_id
- response = await async_client.patch("/api/v1/queue/bulk", json={"item_ids": [item.id], "printer_id": other.id})
- assert response.status_code == 200
- body = response.json()
- assert body["skipped_count"] == 1
- assert body["updated_count"] == 0
- await db_session.refresh(item)
- assert item.printer_id == original_printer_id
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_update_allowed_on_unclaimed_pending_item(
- self, async_client: AsyncClient, queue_item_factory, db_session
- ):
- """Regression guard: a normal pending row (no claim) still edits fine."""
- item = await queue_item_factory()
- response = await async_client.patch(f"/api/v1/queue/{item.id}", json={"plate_id": 7})
- assert response.status_code == 200
- assert response.json()["plate_id"] == 7
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_get_queue_item(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify single queue item can be retrieved."""
- item = await queue_item_factory()
- response = await async_client.get(f"/api/v1/queue/{item.id}")
- assert response.status_code == 200
- assert response.json()["id"] == item.id
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_get_queue_item_not_found(self, async_client: AsyncClient):
- """Verify 404 for non-existent queue item."""
- response = await async_client.get("/api/v1/queue/9999")
- assert response.status_code == 404
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_update_queue_item(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify queue item can be updated."""
- item = await queue_item_factory()
- response = await async_client.patch(f"/api/v1/queue/{item.id}", json={"auto_off_after": True})
- assert response.status_code == 200
- result = response.json()
- assert result["auto_off_after"] is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_update_queue_item_manual_start(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify queue item manual_start can be updated."""
- item = await queue_item_factory(manual_start=False)
- response = await async_client.patch(f"/api/v1/queue/{item.id}", json={"manual_start": True})
- assert response.status_code == 200
- result = response.json()
- assert result["manual_start"] is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_delete_queue_item(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify queue item can be deleted."""
- item = await queue_item_factory()
- response = await async_client.delete(f"/api/v1/queue/{item.id}")
- assert response.status_code == 200
- assert response.json()["message"] == "Queue item deleted"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_delete_queue_item_releases_reserved_budget(
- self, async_client: AsyncClient, printer_factory, archive_factory, queue_item_factory, db_session
- ):
- """Deleting a pending cost-center queue item releases its reserved budget."""
- await enable_billing(db_session)
- printer = await printer_factory()
- archive = await archive_factory(cost=3.0, filament_used_grams=50.0)
- cost_center = CostCenter(
- name="Delete Releases Budget CC", is_active=True, is_private=False, monthly_budget=10.0
- )
- db_session.add(cost_center)
- await db_session.commit()
- await db_session.refresh(cost_center)
- item = await queue_item_factory(
- printer_id=printer.id,
- archive_id=archive.id,
- cost_center_id=cost_center.id,
- estimated_cost=8.0,
- status="pending",
- )
- blocked = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": archive.id,
- "cost_center_id": cost_center.id,
- "estimated_cost": 0.01,
- },
- )
- assert blocked.status_code == 400
- deleted = await async_client.delete(f"/api/v1/queue/{item.id}")
- assert deleted.status_code == 200
- allowed = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": archive.id,
- "cost_center_id": cost_center.id,
- "estimated_cost": 0.01,
- },
- )
- assert allowed.status_code == 200
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_delete_queue_item_not_found(self, async_client: AsyncClient):
- """Verify 404 for deleting non-existent queue item."""
- response = await async_client.delete("/api/v1/queue/9999")
- assert response.status_code == 404
- class TestQueueStartEndpoint:
- """Tests for the /queue/{item_id}/start endpoint."""
- @pytest.fixture
- async def printer_factory(self, db_session):
- """Factory to create test printers."""
- _counter = [0]
- async def _create_printer(**kwargs):
- from backend.app.models.printer import Printer
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "name": f"Test Printer {counter}",
- "ip_address": f"192.168.1.{100 + counter}",
- "serial_number": f"TESTSERIAL{counter:04d}",
- "access_code": "12345678",
- "model": "X1C",
- }
- defaults.update(kwargs)
- printer = Printer(**defaults)
- db_session.add(printer)
- await db_session.commit()
- await db_session.refresh(printer)
- return printer
- return _create_printer
- @pytest.fixture
- async def archive_factory(self, db_session):
- """Factory to create test archives."""
- _counter = [0]
- async def _create_archive(**kwargs):
- from backend.app.models.archive import PrintArchive
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "filename": f"test_print_{counter}.3mf",
- "print_name": f"Test Print {counter}",
- "file_path": f"/tmp/test_print_{counter}.3mf",
- "file_size": 1024,
- "content_hash": f"testhash{counter:08d}",
- "status": "completed",
- }
- defaults.update(kwargs)
- archive = PrintArchive(**defaults)
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- return archive
- return _create_archive
- @pytest.fixture
- async def queue_item_factory(self, db_session, printer_factory, archive_factory):
- """Factory to create test queue items."""
- _counter = [0]
- async def _create_queue_item(**kwargs):
- from backend.app.models.print_queue import PrintQueueItem
- _counter[0] += 1
- counter = _counter[0]
- if "printer_id" not in kwargs:
- printer = await printer_factory()
- kwargs["printer_id"] = printer.id
- if "archive_id" not in kwargs:
- archive = await archive_factory()
- kwargs["archive_id"] = archive.id
- defaults = {
- "status": "pending",
- "position": counter,
- }
- defaults.update(kwargs)
- item = PrintQueueItem(**defaults)
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- return item
- return _create_queue_item
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_start_staged_queue_item(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify starting a staged (manual_start=True) queue item clears the flag."""
- item = await queue_item_factory(manual_start=True)
- assert item.manual_start is True
- response = await async_client.post(f"/api/v1/queue/{item.id}/start")
- assert response.status_code == 200
- result = response.json()
- assert result["manual_start"] is False
- assert result["status"] == "pending"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_start_non_staged_queue_item(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify starting a non-staged queue item still works (idempotent)."""
- item = await queue_item_factory(manual_start=False)
- assert item.manual_start is False
- response = await async_client.post(f"/api/v1/queue/{item.id}/start")
- assert response.status_code == 200
- result = response.json()
- assert result["manual_start"] is False
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_start_queue_item_with_cost_center_requires_estimated_cost(
- self, async_client: AsyncClient, queue_item_factory, db_session
- ):
- """Starting a cost-center queue item requires a stored estimate."""
- await enable_billing(db_session)
- cost_center = CostCenter(name="Start Budget CC", is_active=True, is_private=False, monthly_budget=10.0)
- db_session.add(cost_center)
- await db_session.commit()
- await db_session.refresh(cost_center)
- item = await queue_item_factory(manual_start=True, cost_center_id=cost_center.id, estimated_cost=None)
- response = await async_client.post(f"/api/v1/queue/{item.id}/start")
- assert response.status_code == 400
- assert "Estimated cost is required" in response.json()["detail"]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_start_queue_item_not_found(self, async_client: AsyncClient):
- """Verify 404 for non-existent queue item."""
- response = await async_client.post("/api/v1/queue/9999/start")
- assert response.status_code == 404
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_start_non_pending_queue_item(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify 400 error when trying to start a non-pending queue item."""
- item = await queue_item_factory(status="printing", manual_start=True)
- response = await async_client.post(f"/api/v1/queue/{item.id}/start")
- assert response.status_code == 400
- assert "pending" in response.json()["detail"].lower()
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_start_completed_queue_item(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify 400 error when trying to start a completed queue item."""
- item = await queue_item_factory(status="completed", manual_start=True)
- response = await async_client.post(f"/api/v1/queue/{item.id}/start")
- assert response.status_code == 400
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_start_returns_409_on_filament_deficit(
- self,
- async_client: AsyncClient,
- queue_item_factory,
- db_session,
- monkeypatch,
- ):
- """Filament deficit must surface as 409 + structured payload (#1496)."""
- from backend.app.services import filament_deficit as fd_module
- item = await queue_item_factory(manual_start=True)
- async def _fake_deficit(_db, _item):
- return [
- fd_module.FilamentDeficit(
- slot_id=1,
- ams_id=0,
- tray_id=0,
- filament_type="PLA",
- required_grams=270.0,
- remaining_grams=200.0,
- ),
- ]
- monkeypatch.setattr(
- "backend.app.api.routes.print_queue.compute_deficit_for_queue_item",
- _fake_deficit,
- )
- response = await async_client.post(f"/api/v1/queue/{item.id}/start")
- assert response.status_code == 409
- body = response.json()
- assert body["detail"]["code"] == "insufficient_filament"
- assert len(body["detail"]["deficit"]) == 1
- assert body["detail"]["deficit"][0]["slot_id"] == 1
- assert body["detail"]["deficit"][0]["required_grams"] == 270.0
- assert body["detail"]["deficit"][0]["remaining_grams"] == 200.0
- # Item still pending, manual_start unchanged.
- await db_session.refresh(item)
- assert item.status == "pending"
- assert item.manual_start is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_start_with_skip_flag_bypasses_deficit_check(
- self,
- async_client: AsyncClient,
- queue_item_factory,
- db_session,
- monkeypatch,
- ):
- """With skip_filament_check=true the route dispatches even when short (#1496)."""
- from backend.app.services import filament_deficit as fd_module
- item = await queue_item_factory(manual_start=True, filament_short=True)
- called_with = {}
- async def _fake_deficit(_db, _item):
- called_with["called"] = True
- return [
- fd_module.FilamentDeficit(
- slot_id=1,
- ams_id=0,
- tray_id=0,
- filament_type="PLA",
- required_grams=270.0,
- remaining_grams=200.0,
- ),
- ]
- monkeypatch.setattr(
- "backend.app.api.routes.print_queue.compute_deficit_for_queue_item",
- _fake_deficit,
- )
- response = await async_client.post(f"/api/v1/queue/{item.id}/start?skip_filament_check=true")
- assert response.status_code == 200
- body = response.json()
- assert body["manual_start"] is False
- assert body["filament_short"] is False
- # Helper not called on the bypass path — we trust the operator's
- # decision to print anyway.
- assert called_with == {}
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_start_with_skip_flag_persists_acknowledgement(
- self,
- async_client: AsyncClient,
- queue_item_factory,
- db_session,
- ):
- """skip_filament_check=true sets the persistent flag on the queue item
- so the scheduler doesn't re-flag it on the next tick (#1698-followup).
- Without persistence the route's flag-clearing only survives until the
- next scheduler tick re-runs the deficit check on identical spool
- state and re-promotes the item — the user has to click Play+Confirm
- every single tick.
- """
- item = await queue_item_factory(manual_start=True, filament_short=True)
- assert item.skip_filament_check is False
- response = await async_client.post(f"/api/v1/queue/{item.id}/start?skip_filament_check=true")
- assert response.status_code == 200
- body = response.json()
- assert body["skip_filament_check"] is True
- await db_session.refresh(item)
- assert item.skip_filament_check is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_start_without_skip_flag_does_not_set_acknowledgement(
- self,
- async_client: AsyncClient,
- queue_item_factory,
- db_session,
- ):
- """A successful Play click with no deficit must NOT silently set the
- acknowledgement flag — only an explicit Print Anyway should.
- """
- item = await queue_item_factory(manual_start=False, filament_short=False)
- assert item.skip_filament_check is False
- response = await async_client.post(f"/api/v1/queue/{item.id}/start")
- assert response.status_code == 200
- await db_session.refresh(item)
- assert item.skip_filament_check is False
- class TestQueueCancelEndpoint:
- """Tests for the /queue/{item_id}/cancel endpoint."""
- @pytest.fixture
- async def printer_factory(self, db_session):
- """Factory to create test printers."""
- async def _create_printer(**kwargs):
- from backend.app.models.printer import Printer
- defaults = {
- "name": "Cancel Test Printer",
- "ip_address": "192.168.1.200",
- "serial_number": "TESTCANCEL001",
- "access_code": "12345678",
- "model": "X1C",
- }
- defaults.update(kwargs)
- printer = Printer(**defaults)
- db_session.add(printer)
- await db_session.commit()
- await db_session.refresh(printer)
- return printer
- return _create_printer
- @pytest.fixture
- async def archive_factory(self, db_session):
- """Factory to create test archives."""
- async def _create_archive(**kwargs):
- from backend.app.models.archive import PrintArchive
- defaults = {
- "filename": "cancel_test.3mf",
- "print_name": "Cancel Test Print",
- "file_path": "/tmp/cancel_test.3mf",
- "file_size": 1024,
- "content_hash": "cancelhash001",
- "status": "completed",
- }
- defaults.update(kwargs)
- archive = PrintArchive(**defaults)
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- return archive
- return _create_archive
- @pytest.fixture
- async def queue_item_factory(self, db_session, printer_factory, archive_factory):
- """Factory to create test queue items."""
- async def _create_queue_item(**kwargs):
- from backend.app.models.print_queue import PrintQueueItem
- if "printer_id" not in kwargs:
- printer = await printer_factory()
- kwargs["printer_id"] = printer.id
- if "archive_id" not in kwargs:
- archive = await archive_factory()
- kwargs["archive_id"] = archive.id
- defaults = {
- "status": "pending",
- "position": 1,
- }
- defaults.update(kwargs)
- item = PrintQueueItem(**defaults)
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- return item
- return _create_queue_item
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_cancel_pending_queue_item(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify cancelling a pending queue item."""
- item = await queue_item_factory(status="pending")
- response = await async_client.post(f"/api/v1/queue/{item.id}/cancel")
- assert response.status_code == 200
- assert response.json()["message"] == "Queue item cancelled"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_cancel_non_pending_queue_item(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify 400 error when trying to cancel a non-pending queue item."""
- item = await queue_item_factory(status="printing")
- response = await async_client.post(f"/api/v1/queue/{item.id}/cancel")
- assert response.status_code == 400
- class TestQueueLibraryFileSupport:
- """Tests for queue items with library_file_id (instead of archive_id)."""
- @pytest.fixture
- async def printer_factory(self, db_session):
- """Factory to create test printers."""
- _counter = [0]
- async def _create_printer(**kwargs):
- from backend.app.models.printer import Printer
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "name": f"Library Test Printer {counter}",
- "ip_address": f"192.168.1.{150 + counter}",
- "serial_number": f"TESTLIB{counter:04d}",
- "access_code": "12345678",
- "model": "X1C",
- }
- defaults.update(kwargs)
- printer = Printer(**defaults)
- db_session.add(printer)
- await db_session.commit()
- await db_session.refresh(printer)
- return printer
- return _create_printer
- @pytest.fixture
- async def library_file_factory(self, db_session):
- """Factory to create test library files."""
- _counter = [0]
- async def _create_library_file(**kwargs):
- from backend.app.models.library import LibraryFile
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "filename": f"library_test_{counter}.3mf",
- "file_path": f"/test/library/library_test_{counter}.3mf",
- "file_size": 2048,
- "file_type": "3mf",
- "file_metadata": {"print_name": f"Library Print {counter}", "print_time_seconds": 3600},
- }
- defaults.update(kwargs)
- lib_file = LibraryFile(**defaults)
- db_session.add(lib_file)
- await db_session.commit()
- await db_session.refresh(lib_file)
- return lib_file
- return _create_library_file
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_with_library_file(
- self, async_client: AsyncClient, printer_factory, library_file_factory, db_session
- ):
- """Verify item can be added to queue using library_file_id instead of archive_id."""
- printer = await printer_factory()
- lib_file = await library_file_factory()
- data = {
- "printer_id": printer.id,
- "library_file_id": lib_file.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["printer_id"] == printer.id
- assert result["library_file_id"] == lib_file.id
- assert result["archive_id"] is None
- assert result["status"] == "pending"
- assert result["library_file_name"] == "Library Print 1"
- assert result["print_time_seconds"] == 3600
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_library_file_rejects_cross_model_mismatch(
- self, async_client: AsyncClient, printer_factory, library_file_factory, db_session
- ):
- """Cross-model gate (#2578) also reads sliced_for_model from library file metadata."""
- await printer_factory(model="H2D")
- lib_file = await library_file_factory(file_metadata={"print_name": "Mismatch", "sliced_for_model": "X1C"})
- response = await async_client.post(
- "/api/v1/queue/",
- json={"target_model": "H2D", "library_file_id": lib_file.id},
- )
- assert response.status_code == 400
- assert "sliced for X1C" in response.json()["detail"]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_library_file_with_options(
- self, async_client: AsyncClient, printer_factory, library_file_factory, db_session
- ):
- """Verify library file queue item can have all options set."""
- printer = await printer_factory()
- lib_file = await library_file_factory()
- data = {
- "printer_id": printer.id,
- "library_file_id": lib_file.id,
- "ams_mapping": [1, 2, -1, -1],
- "plate_id": 2,
- "bed_levelling": "off",
- "timelapse": True,
- "manual_start": True,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["library_file_id"] == lib_file.id
- assert result["ams_mapping"] == [1, 2, -1, -1]
- assert result["plate_id"] == 2
- assert result["bed_levelling"] == "off"
- assert result["timelapse"] is True
- assert result["manual_start"] is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_requires_archive_or_library_file(
- self, async_client: AsyncClient, printer_factory, db_session
- ):
- """Verify 400 error when neither archive_id nor library_file_id provided."""
- printer = await printer_factory()
- data = {
- "printer_id": printer.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 400
- assert (
- "archive_id" in response.json()["detail"].lower() or "library_file_id" in response.json()["detail"].lower()
- )
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_update_queue_item_with_library_file(
- self, async_client: AsyncClient, printer_factory, library_file_factory, db_session
- ):
- """Verify queue item with library_file_id can be updated."""
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- lib_file = await library_file_factory()
- # Create queue item directly
- item = PrintQueueItem(
- printer_id=printer.id,
- library_file_id=lib_file.id,
- status="pending",
- position=1,
- )
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- # Update the item
- response = await async_client.patch(
- f"/api/v1/queue/{item.id}",
- json={"auto_off_after": True, "plate_id": 3},
- )
- assert response.status_code == 200
- result = response.json()
- assert result["auto_off_after"] is True
- assert result["plate_id"] == 3
- assert result["library_file_id"] == lib_file.id
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_list_queue_includes_library_file_info(
- self, async_client: AsyncClient, printer_factory, library_file_factory, db_session
- ):
- """Verify queue list includes library file metadata."""
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- lib_file = await library_file_factory(
- file_metadata={"print_name": "Custom Print Name", "print_time_seconds": 7200}
- )
- item = PrintQueueItem(
- printer_id=printer.id,
- library_file_id=lib_file.id,
- status="pending",
- position=1,
- )
- db_session.add(item)
- await db_session.commit()
- response = await async_client.get("/api/v1/queue/")
- assert response.status_code == 200
- items = response.json()
- assert len(items) >= 1
- # Find our item
- our_item = next((i for i in items if i["library_file_id"] == lib_file.id), None)
- assert our_item is not None
- assert our_item["library_file_name"] == "Custom Print Name"
- assert our_item["print_time_seconds"] == 7200
- class TestBulkUpdateEndpoint:
- """Tests for the /queue/bulk endpoint."""
- @pytest.fixture
- async def printer_factory(self, db_session):
- """Factory to create test printers."""
- _counter = [0]
- async def _create_printer(**kwargs):
- from backend.app.models.printer import Printer
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "name": f"Bulk Test Printer {counter}",
- "ip_address": f"192.168.1.{150 + counter}",
- "serial_number": f"TESTBULK{counter:04d}",
- "access_code": "12345678",
- "model": "X1C",
- }
- defaults.update(kwargs)
- printer = Printer(**defaults)
- db_session.add(printer)
- await db_session.commit()
- await db_session.refresh(printer)
- return printer
- return _create_printer
- @pytest.fixture
- async def archive_factory(self, db_session):
- """Factory to create test archives."""
- _counter = [0]
- async def _create_archive(**kwargs):
- from backend.app.models.archive import PrintArchive
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "filename": f"bulk_test_{counter}.3mf",
- "print_name": f"Bulk Test Print {counter}",
- "file_path": f"/tmp/bulk_test_{counter}.3mf",
- "file_size": 1024,
- "content_hash": f"bulkhash{counter:04d}",
- "status": "completed",
- }
- defaults.update(kwargs)
- archive = PrintArchive(**defaults)
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- return archive
- return _create_archive
- @pytest.fixture
- async def queue_item_factory(self, db_session, printer_factory, archive_factory):
- """Factory to create test queue items."""
- async def _create_item(**kwargs):
- from backend.app.models.print_queue import PrintQueueItem
- if "printer_id" not in kwargs:
- printer = await printer_factory()
- kwargs["printer_id"] = printer.id
- if "archive_id" not in kwargs:
- archive = await archive_factory()
- kwargs["archive_id"] = archive.id
- defaults = {
- "status": "pending",
- "position": 1,
- "bed_levelling": "on",
- "flow_cali": "off",
- "vibration_cali": True,
- }
- defaults.update(kwargs)
- item = PrintQueueItem(**defaults)
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- return item
- return _create_item
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bulk_update_single_field(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify bulk update can change a single field on multiple items."""
- item1 = await queue_item_factory(bed_levelling="on")
- item2 = await queue_item_factory(bed_levelling="on")
- response = await async_client.patch(
- "/api/v1/queue/bulk",
- json={"item_ids": [item1.id, item2.id], "bed_levelling": "off"},
- )
- assert response.status_code == 200
- result = response.json()
- assert result["updated_count"] == 2
- assert result["skipped_count"] == 0
- # Verify items were updated
- await db_session.refresh(item1)
- await db_session.refresh(item2)
- assert item1.bed_levelling == "off"
- assert item2.bed_levelling == "off"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bulk_update_multiple_fields(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify bulk update can change multiple fields at once."""
- item1 = await queue_item_factory(bed_levelling="on", flow_cali="off", manual_start=False)
- item2 = await queue_item_factory(bed_levelling="on", flow_cali="off", manual_start=False)
- response = await async_client.patch(
- "/api/v1/queue/bulk",
- json={
- "item_ids": [item1.id, item2.id],
- "bed_levelling": "off",
- "flow_cali": "on",
- "manual_start": True,
- },
- )
- assert response.status_code == 200
- result = response.json()
- assert result["updated_count"] == 2
- await db_session.refresh(item1)
- assert item1.bed_levelling == "off"
- assert item1.flow_cali == "on"
- assert item1.manual_start is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bulk_update_skips_non_pending(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify bulk update skips non-pending items."""
- pending_item = await queue_item_factory(status="pending", bed_levelling="on")
- printing_item = await queue_item_factory(status="printing", bed_levelling="on")
- completed_item = await queue_item_factory(status="completed", bed_levelling="on")
- response = await async_client.patch(
- "/api/v1/queue/bulk",
- json={
- "item_ids": [pending_item.id, printing_item.id, completed_item.id],
- "bed_levelling": "off",
- },
- )
- assert response.status_code == 200
- result = response.json()
- assert result["updated_count"] == 1
- assert result["skipped_count"] == 2
- # Only pending item should be updated
- await db_session.refresh(pending_item)
- await db_session.refresh(printing_item)
- await db_session.refresh(completed_item)
- assert pending_item.bed_levelling == "off"
- assert printing_item.bed_levelling == "on"
- assert completed_item.bed_levelling == "on"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bulk_update_change_printer(
- self, async_client: AsyncClient, queue_item_factory, printer_factory, db_session
- ):
- """Verify bulk update can reassign items to a different printer."""
- new_printer = await printer_factory(name="New Target Printer")
- item1 = await queue_item_factory()
- item2 = await queue_item_factory()
- original_printer_id = item1.printer_id
- response = await async_client.patch(
- "/api/v1/queue/bulk",
- json={"item_ids": [item1.id, item2.id], "printer_id": new_printer.id},
- )
- assert response.status_code == 200
- await db_session.refresh(item1)
- await db_session.refresh(item2)
- assert item1.printer_id == new_printer.id
- assert item2.printer_id == new_printer.id
- assert item1.printer_id != original_printer_id
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bulk_update_empty_item_ids(self, async_client: AsyncClient):
- """Verify 400 error when item_ids is empty."""
- response = await async_client.patch(
- "/api/v1/queue/bulk",
- json={"item_ids": [], "bed_levelling": False},
- )
- assert response.status_code == 400
- assert "no item" in response.json()["detail"].lower()
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bulk_update_no_fields(self, async_client: AsyncClient, queue_item_factory):
- """Verify 400 error when no fields to update."""
- item = await queue_item_factory()
- response = await async_client.patch(
- "/api/v1/queue/bulk",
- json={"item_ids": [item.id]},
- )
- assert response.status_code == 400
- assert "no fields" in response.json()["detail"].lower()
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bulk_update_invalid_printer(self, async_client: AsyncClient, queue_item_factory):
- """Verify 400 error when printer_id doesn't exist."""
- item = await queue_item_factory()
- response = await async_client.patch(
- "/api/v1/queue/bulk",
- json={"item_ids": [item.id], "printer_id": 99999},
- )
- assert response.status_code == 400
- assert "printer not found" in response.json()["detail"].lower()
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bulk_update_cost_center_requires_estimated_cost(
- self, async_client: AsyncClient, queue_item_factory, db_session
- ):
- """Bulk assigning a cost center requires an estimate, same as single-item updates."""
- await enable_billing(db_session)
- item = await queue_item_factory()
- cost_center = CostCenter(name="Bulk Budget CC", is_active=True, is_private=False, monthly_budget=10.0)
- db_session.add(cost_center)
- await db_session.commit()
- await db_session.refresh(cost_center)
- response = await async_client.patch(
- "/api/v1/queue/bulk",
- json={"item_ids": [item.id], "cost_center_id": cost_center.id},
- )
- assert response.status_code == 400
- assert "Estimated cost is required" in response.json()["detail"]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bulk_update_cost_center_counts_pending_reservations(
- self, async_client: AsyncClient, queue_item_factory, archive_factory, db_session
- ):
- """A forged bulk-update hint cannot weaken the server-derived reservation."""
- await enable_billing(db_session)
- archive = await archive_factory(cost=3.0, filament_used_grams=50.0)
- item = await queue_item_factory(archive_id=archive.id)
- existing = await queue_item_factory()
- cost_center = CostCenter(name="Bulk Reserved CC", is_active=True, is_private=False, monthly_budget=10.0)
- db_session.add(cost_center)
- await db_session.commit()
- await db_session.refresh(cost_center)
- existing.cost_center_id = cost_center.id
- existing.estimated_cost = 8.0
- await db_session.commit()
- response = await async_client.patch(
- "/api/v1/queue/bulk",
- json={"item_ids": [item.id], "cost_center_id": cost_center.id, "estimated_cost": 0.01},
- )
- assert response.status_code == 400
- assert "exceeds available cost center budget" in response.json()["detail"]
- await db_session.refresh(item)
- assert item.cost_center_id is None
- assert item.estimated_cost is None
- class TestTargetLocationFeature:
- """Tests for queue items with target_location (Issue #220)."""
- @pytest.fixture
- async def printer_factory(self, db_session):
- """Factory to create test printers."""
- _counter = [0]
- async def _create_printer(**kwargs):
- from backend.app.models.printer import Printer
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "name": f"Location Test Printer {counter}",
- "ip_address": f"192.168.1.{50 + counter}",
- "serial_number": f"TESTLOC{counter:04d}",
- "access_code": "12345678",
- "model": "X1C",
- }
- defaults.update(kwargs)
- printer = Printer(**defaults)
- db_session.add(printer)
- await db_session.commit()
- await db_session.refresh(printer)
- return printer
- return _create_printer
- @pytest.fixture
- async def archive_factory(self, db_session):
- """Factory to create test archives."""
- _counter = [0]
- async def _create_archive(**kwargs):
- from backend.app.models.archive import PrintArchive
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "filename": f"location_test_{counter}.3mf",
- "print_name": f"Location Test Print {counter}",
- "file_path": f"/tmp/location_test_{counter}.3mf",
- "file_size": 1024,
- "content_hash": f"lochash{counter:08d}",
- "status": "completed",
- }
- defaults.update(kwargs)
- archive = PrintArchive(**defaults)
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- return archive
- return _create_archive
- @pytest.fixture
- async def queue_item_factory(self, db_session, printer_factory, archive_factory):
- """Factory to create test queue items."""
- _counter = [0]
- async def _create_queue_item(**kwargs):
- from backend.app.models.print_queue import PrintQueueItem
- _counter[0] += 1
- counter = _counter[0]
- if "printer_id" not in kwargs and "target_model" not in kwargs:
- printer = await printer_factory()
- kwargs["printer_id"] = printer.id
- if "archive_id" not in kwargs:
- archive = await archive_factory()
- kwargs["archive_id"] = archive.id
- defaults = {
- "status": "pending",
- "position": counter,
- }
- defaults.update(kwargs)
- item = PrintQueueItem(**defaults)
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- return item
- return _create_queue_item
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_with_target_location(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify item can be added with target_model and target_location."""
- # Create a printer with model X1C so the API can validate
- await printer_factory(model="X1C", location="Office")
- archive = await archive_factory()
- data = {
- "target_model": "X1C",
- "target_location": "Workbench",
- "archive_id": archive.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["target_model"] == "X1C"
- assert result["target_location"] == "Workbench"
- assert result["printer_id"] is None
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_location_without_model_ignored(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify target_location without target_model is allowed (location is just ignored)."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "target_location": "Workbench", # This gets ignored since printer_id is set
- "archive_id": archive.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- # The API accepts this but the location is only used with target_model
- assert response.status_code == 200
- result = response.json()
- assert result["printer_id"] == printer.id
- # Location may or may not be stored since it's meaningless without target_model
- # The important thing is the request succeeds
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_queue_item_target_location_in_response(
- self, async_client: AsyncClient, queue_item_factory, db_session
- ):
- """Verify target_location is returned in queue item response."""
- item = await queue_item_factory(
- printer_id=None,
- target_model="X1C",
- target_location="Workshop",
- )
- response = await async_client.get(f"/api/v1/queue/{item.id}")
- assert response.status_code == 200
- result = response.json()
- assert result["target_model"] == "X1C"
- assert result["target_location"] == "Workshop"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_queue_list_includes_target_location(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify target_location is included in queue list."""
- await queue_item_factory(
- printer_id=None,
- target_model="P1S",
- target_location="Garage",
- )
- response = await async_client.get("/api/v1/queue/")
- assert response.status_code == 200
- items = response.json()
- assert len(items) >= 1
- # Find our item
- our_item = next((i for i in items if i["target_location"] == "Garage"), None)
- assert our_item is not None
- assert our_item["target_model"] == "P1S"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_update_queue_item_target_location(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify target_location can be updated on existing queue item."""
- item = await queue_item_factory(
- printer_id=None,
- target_model="X1C",
- target_location="Office",
- )
- response = await async_client.patch(
- f"/api/v1/queue/{item.id}",
- json={"target_location": "Basement"},
- )
- assert response.status_code == 200
- result = response.json()
- assert result["target_location"] == "Basement"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_clear_target_location(self, async_client: AsyncClient, queue_item_factory, db_session):
- """Verify target_location can be cleared (set to None)."""
- item = await queue_item_factory(
- printer_id=None,
- target_model="X1C",
- target_location="Office",
- )
- # Note: Setting to empty string should clear it
- response = await async_client.patch(
- f"/api/v1/queue/{item.id}",
- json={"target_location": None},
- )
- assert response.status_code == 200
- result = response.json()
- assert result["target_location"] is None
- # ------------------------------------------------------------------
- # Cross-model dispatch gate (#2578): a G-code 3MF sliced for one model
- # must not be queued for model-based dispatch to an incompatible model.
- # ------------------------------------------------------------------
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_rejects_cross_model_mismatch(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """X1C-sliced archive + target_model=H2D must be rejected (#2578)."""
- await printer_factory(model="H2D")
- archive = await archive_factory(sliced_for_model="X1C")
- response = await async_client.post(
- "/api/v1/queue/",
- json={"target_model": "H2D", "archive_id": archive.id},
- )
- assert response.status_code == 400
- assert "sliced for X1C" in response.json()["detail"]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_allows_gcode_family_target(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """X1C-sliced G-code on a P1S is an intentional mixed-farm workflow —
- same kinematics/volume family, must stay allowed."""
- await printer_factory(model="P1S")
- archive = await archive_factory(sliced_for_model="X1C")
- response = await async_client.post(
- "/api/v1/queue/",
- json={"target_model": "P1S", "archive_id": archive.id},
- )
- assert response.status_code == 200
- assert response.json()["target_model"] == "P1S"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_without_sliced_metadata_not_blocked(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Legacy archives without sliced_for_model can't be validated — must keep working."""
- await printer_factory(model="H2D")
- archive = await archive_factory() # no sliced_for_model
- response = await async_client.post(
- "/api/v1/queue/",
- json={"target_model": "H2D", "archive_id": archive.id},
- )
- assert response.status_code == 200
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_update_rejects_cross_model_mismatch(
- self, async_client: AsyncClient, printer_factory, archive_factory, queue_item_factory, db_session
- ):
- """Editing an item must not be able to introduce an incompatible target either."""
- await printer_factory(model="X1C")
- await printer_factory(model="H2D")
- archive = await archive_factory(sliced_for_model="X1C")
- item = await queue_item_factory(printer_id=None, target_model="X1C", archive_id=archive.id)
- response = await async_client.patch(f"/api/v1/queue/{item.id}", json={"target_model": "H2D"})
- assert response.status_code == 400
- assert "sliced for X1C" in response.json()["detail"]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_update_can_fix_stale_mismatched_target(
- self, async_client: AsyncClient, printer_factory, archive_factory, queue_item_factory, db_session
- ):
- """A pre-fix DB row with a wrong target (the reporter's rows 78-82) must be
- repairable by editing the target back to the sliced-for model."""
- await printer_factory(model="X1C")
- archive = await archive_factory(sliced_for_model="X1C")
- # Stale mismatched row written directly to the DB (bypasses the API gate)
- item = await queue_item_factory(printer_id=None, target_model="H2D", archive_id=archive.id)
- response = await async_client.patch(f"/api/v1/queue/{item.id}", json={"target_model": "X1C"})
- assert response.status_code == 200
- assert response.json()["target_model"] == "X1C"
- class TestAbortedStatusNormalisation:
- """Tests for issue #558: 'aborted' queue status causes 500 error."""
- @pytest.fixture
- async def printer_factory(self, db_session):
- """Factory to create test printers."""
- _counter = [0]
- async def _create_printer(**kwargs):
- from backend.app.models.printer import Printer
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "name": f"Abort Test Printer {counter}",
- "ip_address": f"192.168.1.{60 + counter}",
- "serial_number": f"TESTABORT{counter:04d}",
- "access_code": "12345678",
- "model": "P1S",
- }
- defaults.update(kwargs)
- printer = Printer(**defaults)
- db_session.add(printer)
- await db_session.commit()
- await db_session.refresh(printer)
- return printer
- return _create_printer
- @pytest.fixture
- async def archive_factory(self, db_session):
- """Factory to create test archives."""
- _counter = [0]
- async def _create_archive(**kwargs):
- from backend.app.models.archive import PrintArchive
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "filename": f"abort_test_{counter}.3mf",
- "print_name": f"Abort Test Print {counter}",
- "file_path": f"/tmp/abort_test_{counter}.3mf",
- "file_size": 1024,
- "content_hash": f"aborthash{counter:06d}",
- "status": "completed",
- }
- defaults.update(kwargs)
- archive = PrintArchive(**defaults)
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- return archive
- return _create_archive
- @pytest.fixture
- async def queue_item_factory(self, db_session, printer_factory, archive_factory):
- """Factory to create test queue items."""
- _counter = [0]
- async def _create_queue_item(**kwargs):
- from backend.app.models.print_queue import PrintQueueItem
- _counter[0] += 1
- counter = _counter[0]
- if "printer_id" not in kwargs:
- printer = await printer_factory()
- kwargs["printer_id"] = printer.id
- if "archive_id" not in kwargs:
- archive = await archive_factory()
- kwargs["archive_id"] = archive.id
- defaults = {
- "status": "pending",
- "position": counter,
- }
- defaults.update(kwargs)
- item = PrintQueueItem(**defaults)
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- return item
- return _create_queue_item
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_on_print_complete_normalises_aborted_to_cancelled(self, queue_item_factory, db_session):
- """Verify the completion handler maps 'aborted' → 'cancelled' for queue items."""
- import asyncio
- from unittest.mock import AsyncMock, MagicMock, patch
- item = await queue_item_factory(status="printing")
- # Build a mock session whose execute returns our item
- mock_result = MagicMock()
- mock_result.scalars.return_value.all.return_value = [item]
- mock_session = AsyncMock()
- mock_session.__aenter__ = AsyncMock(return_value=mock_session)
- mock_session.__aexit__ = AsyncMock(return_value=False)
- mock_session.execute = AsyncMock(return_value=mock_result)
- mock_session.commit = AsyncMock()
- tasks_before = set(asyncio.all_tasks())
- with (
- patch("backend.app.main.async_session", return_value=mock_session),
- patch("backend.app.core.database.async_session", return_value=mock_session),
- patch("backend.app.main.ws_manager") as mock_ws,
- patch("backend.app.main.mqtt_relay") as mock_relay,
- patch("backend.app.main.notification_service") as mock_notif,
- patch("backend.app.main.smart_plug_manager") as mock_plug,
- patch("backend.app.main.printer_manager") as mock_pm,
- ):
- mock_ws.send_print_complete = AsyncMock()
- mock_ws.broadcast = AsyncMock()
- mock_relay.on_print_complete = AsyncMock()
- mock_relay.on_queue_job_completed = AsyncMock()
- mock_notif.on_print_complete = AsyncMock()
- mock_plug.on_print_complete = AsyncMock()
- mock_pm.get_printer.return_value = None
- from backend.app.main import on_print_complete
- await on_print_complete(
- item.printer_id,
- {
- "status": "aborted",
- "filename": "test.gcode",
- "subtask_name": "Test",
- "timelapse_was_active": False,
- },
- )
- # Cancel background tasks before leaving mock context
- for task in asyncio.all_tasks() - tasks_before:
- task.cancel()
- try:
- await task
- except (asyncio.CancelledError, Exception):
- pass
- # The item status should be normalised to 'cancelled', not 'aborted'
- assert item.status == "cancelled"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_startup_fixup_converts_aborted_to_cancelled(self, queue_item_factory, db_session):
- """Verify the startup fixup converts existing 'aborted' rows to 'cancelled'."""
- from sqlalchemy import select
- from backend.app.models.print_queue import PrintQueueItem
- # Create items with various statuses including 'aborted'
- item_aborted = await queue_item_factory(status="pending")
- item_pending = await queue_item_factory(status="pending")
- # Manually set the invalid status
- item_aborted.status = "aborted"
- db_session.add(item_aborted)
- await db_session.commit()
- # Run the fixup query (same logic as lifespan)
- result = await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.status == "aborted"))
- aborted_items = result.scalars().all()
- for i in aborted_items:
- i.status = "cancelled"
- await db_session.commit()
- # Verify: no more 'aborted' items
- result = await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.status == "aborted"))
- assert len(result.scalars().all()) == 0
- # The previously aborted item should now be 'cancelled'
- await db_session.refresh(item_aborted)
- assert item_aborted.status == "cancelled"
- # The pending item should be unchanged
- await db_session.refresh(item_pending)
- assert item_pending.status == "pending"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_completed_status_passes_through_unchanged(self, queue_item_factory, db_session):
- """Verify normal statuses like 'completed' are not affected by normalisation."""
- import asyncio
- from unittest.mock import AsyncMock, MagicMock, patch
- item = await queue_item_factory(status="printing")
- mock_result = MagicMock()
- mock_result.scalars.return_value.all.return_value = [item]
- mock_session = AsyncMock()
- mock_session.__aenter__ = AsyncMock(return_value=mock_session)
- mock_session.__aexit__ = AsyncMock(return_value=False)
- mock_session.execute = AsyncMock(return_value=mock_result)
- mock_session.commit = AsyncMock()
- tasks_before = set(asyncio.all_tasks())
- with (
- patch("backend.app.main.async_session", return_value=mock_session),
- patch("backend.app.core.database.async_session", return_value=mock_session),
- patch("backend.app.main.ws_manager") as mock_ws,
- patch("backend.app.main.mqtt_relay") as mock_relay,
- patch("backend.app.main.notification_service") as mock_notif,
- patch("backend.app.main.smart_plug_manager") as mock_plug,
- patch("backend.app.main.printer_manager") as mock_pm,
- ):
- mock_ws.send_print_complete = AsyncMock()
- mock_ws.broadcast = AsyncMock()
- mock_relay.on_print_complete = AsyncMock()
- mock_relay.on_queue_job_completed = AsyncMock()
- mock_notif.on_print_complete = AsyncMock()
- mock_plug.on_print_complete = AsyncMock()
- mock_pm.get_printer.return_value = None
- from backend.app.main import on_print_complete
- await on_print_complete(
- item.printer_id,
- {
- "status": "completed",
- "filename": "test.gcode",
- "subtask_name": "Test",
- "timelapse_was_active": False,
- },
- )
- # Cancel background tasks before leaving mock context
- for task in asyncio.all_tasks() - tasks_before:
- task.cancel()
- try:
- await task
- except (asyncio.CancelledError, Exception):
- pass
- assert item.status == "completed"
- # ========================================================================
- # Library file usage tracking on print completion (#1008)
- #
- # These exercise the _bump_library_file_usage_if_completed helper directly
- # rather than invoking the whole on_print_complete handler — that path
- # spawns background asyncio tasks (notifications, MQTT relay, smart-plug)
- # that are expensive to mock and have nothing to do with the bump logic.
- # ========================================================================
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bump_library_file_usage_on_completed(self, printer_factory, db_session):
- """Successful completion increments print_count and stamps last_printed_at."""
- from datetime import datetime, timezone
- from backend.app.main import _bump_library_file_usage_if_completed
- from backend.app.models.library import LibraryFile
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- lib_file = LibraryFile(
- filename="benchy.gcode.3mf",
- file_path="/data/library/benchy.gcode.3mf",
- file_type="gcode.3mf",
- file_size=1024,
- print_count=0,
- last_printed_at=None,
- )
- db_session.add(lib_file)
- await db_session.commit()
- await db_session.refresh(lib_file)
- item = PrintQueueItem(
- printer_id=printer.id,
- library_file_id=lib_file.id,
- status="printing",
- position=1,
- )
- before = datetime.now(timezone.utc).replace(tzinfo=None)
- await _bump_library_file_usage_if_completed(db_session, item, "completed")
- await db_session.commit()
- await db_session.refresh(lib_file)
- assert lib_file.print_count == 1
- assert lib_file.last_printed_at is not None
- assert lib_file.last_printed_at >= before
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bump_library_file_usage_repeated_prints_increment_count(self, printer_factory, db_session):
- """Each successful completion bumps print_count cumulatively."""
- from backend.app.main import _bump_library_file_usage_if_completed
- from backend.app.models.library import LibraryFile
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- lib_file = LibraryFile(
- filename="repeat.gcode.3mf",
- file_path="/data/library/repeat.gcode.3mf",
- file_type="gcode.3mf",
- file_size=1024,
- print_count=0,
- )
- db_session.add(lib_file)
- await db_session.commit()
- await db_session.refresh(lib_file)
- item = PrintQueueItem(
- printer_id=printer.id,
- library_file_id=lib_file.id,
- status="printing",
- position=1,
- )
- for _ in range(3):
- await _bump_library_file_usage_if_completed(db_session, item, "completed")
- await db_session.commit()
- await db_session.refresh(lib_file)
- assert lib_file.print_count == 3
- @pytest.mark.asyncio
- @pytest.mark.integration
- @pytest.mark.parametrize("terminal_status", ["failed", "cancelled"])
- async def test_bump_library_file_usage_skips_non_completed(self, printer_factory, db_session, terminal_status):
- """Failed and cancelled prints must NOT count as usage."""
- from backend.app.main import _bump_library_file_usage_if_completed
- from backend.app.models.library import LibraryFile
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- lib_file = LibraryFile(
- filename="broken.gcode.3mf",
- file_path="/data/library/broken.gcode.3mf",
- file_type="gcode.3mf",
- file_size=1024,
- print_count=0,
- last_printed_at=None,
- )
- db_session.add(lib_file)
- await db_session.commit()
- await db_session.refresh(lib_file)
- item = PrintQueueItem(
- printer_id=printer.id,
- library_file_id=lib_file.id,
- status="printing",
- position=1,
- )
- await _bump_library_file_usage_if_completed(db_session, item, terminal_status)
- await db_session.commit()
- await db_session.refresh(lib_file)
- assert lib_file.print_count == 0
- assert lib_file.last_printed_at is None
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_bump_library_file_usage_skips_when_no_library_file_id(
- self, printer_factory, archive_factory, db_session
- ):
- """Queue items without library_file_id (e.g. archive reprints) are a no-op."""
- from backend.app.main import _bump_library_file_usage_if_completed
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- archive = await archive_factory()
- item = PrintQueueItem(
- printer_id=printer.id,
- library_file_id=None,
- archive_id=archive.id,
- status="printing",
- position=1,
- )
- # Must not raise.
- await _bump_library_file_usage_if_completed(db_session, item, "completed")
- # ========================================================================
- # Batch quantity tests
- # ========================================================================
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_quantity_default(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify quantity=1 (default) creates a single item with no batch."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["batch_id"] is None
- assert result["batch_name"] is None
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_quantity_one_explicit(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify quantity=1 explicitly creates a single item with no batch."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "quantity": 1,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- assert result["batch_id"] is None
- assert result["batch_name"] is None
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_quantity_creates_batch(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify quantity > 1 creates a batch and multiple queue items."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "quantity": 3,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- result = response.json()
- # First item is returned, linked to a batch
- assert result["batch_id"] is not None
- assert result["batch_name"] is not None
- assert "×3" in result["batch_name"]
- # Verify all 3 items were created
- list_response = await async_client.get("/api/v1/queue/")
- items = list_response.json()
- batch_items = [i for i in items if i["batch_id"] == result["batch_id"]]
- assert len(batch_items) == 3
- # All items should have the same settings
- for item in batch_items:
- assert item["printer_id"] == printer.id
- assert item["archive_id"] == archive.id
- assert item["status"] == "pending"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_quantity_sequential_positions(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify batch items get sequential positions."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "quantity": 3,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- batch_id = response.json()["batch_id"]
- list_response = await async_client.get("/api/v1/queue/")
- items = list_response.json()
- batch_items = sorted(
- [i for i in items if i["batch_id"] == batch_id],
- key=lambda i: i["position"],
- )
- positions = [i["position"] for i in batch_items]
- assert positions == [positions[0], positions[0] + 1, positions[0] + 2]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_insert_position_shifts_existing_items(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify priority insertion shifts existing pending items in the same printer queue."""
- printer = await printer_factory()
- first = await archive_factory(print_name="First")
- second = await archive_factory(print_name="Second")
- priority = await archive_factory(print_name="Priority")
- assert (
- await async_client.post("/api/v1/queue/", json={"printer_id": printer.id, "archive_id": first.id})
- ).status_code == 200
- assert (
- await async_client.post("/api/v1/queue/", json={"printer_id": printer.id, "archive_id": second.id})
- ).status_code == 200
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": priority.id,
- "insert_position": 1,
- },
- )
- assert response.status_code == 200
- list_response = await async_client.get(f"/api/v1/queue/?printer_id={printer.id}")
- items = sorted(list_response.json(), key=lambda item: item["position"])
- assert [item["archive_id"] for item in items[:3]] == [priority.id, first.id, second.id]
- assert [item["position"] for item in items[:3]] == [1, 2, 3]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_insert_position_quantity_shifts_existing_by_quantity(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """ASAP batch insertion shifts existing pending items by the inserted quantity."""
- printer = await printer_factory()
- first = await archive_factory(print_name="First")
- second = await archive_factory(print_name="Second")
- priority = await archive_factory(print_name="Priority")
- assert (
- await async_client.post("/api/v1/queue/", json={"printer_id": printer.id, "archive_id": first.id})
- ).status_code == 200
- assert (
- await async_client.post("/api/v1/queue/", json={"printer_id": printer.id, "archive_id": second.id})
- ).status_code == 200
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": priority.id,
- "quantity": 3,
- "insert_position": 1,
- },
- )
- assert response.status_code == 200
- batch_id = response.json()["batch_id"]
- list_response = await async_client.get(f"/api/v1/queue/?printer_id={printer.id}")
- items = sorted(list_response.json(), key=lambda item: item["position"])
- assert [item["archive_id"] for item in items] == [
- priority.id,
- priority.id,
- priority.id,
- first.id,
- second.id,
- ]
- assert [item["position"] for item in items] == [1, 2, 3, 4, 5]
- assert [item["batch_id"] for item in items[:3]] == [batch_id, batch_id, batch_id]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_insert_position_scopes_unassigned_items(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Unassigned inserts shift only the unassigned queue scope."""
- printer = await printer_factory()
- unassigned_first = await archive_factory(print_name="Unassigned First")
- unassigned_second = await archive_factory(print_name="Unassigned Second")
- assigned = await archive_factory(print_name="Assigned")
- priority = await archive_factory(print_name="Unassigned Priority")
- assert (await async_client.post("/api/v1/queue/", json={"archive_id": unassigned_first.id})).status_code == 200
- assert (await async_client.post("/api/v1/queue/", json={"archive_id": unassigned_second.id})).status_code == 200
- assigned_response = await async_client.post(
- "/api/v1/queue/",
- json={"printer_id": printer.id, "archive_id": assigned.id},
- )
- assert assigned_response.status_code == 200
- assert assigned_response.json()["position"] == 1
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "archive_id": priority.id,
- "insert_position": 1,
- },
- )
- assert response.status_code == 200
- unassigned_response = await async_client.get("/api/v1/queue/?printer_id=-1")
- unassigned_items = sorted(unassigned_response.json(), key=lambda item: item["position"])
- assert [item["archive_id"] for item in unassigned_items] == [
- priority.id,
- unassigned_first.id,
- unassigned_second.id,
- ]
- assert [item["position"] for item in unassigned_items] == [1, 2, 3]
- assigned_scope_response = await async_client.get(f"/api/v1/queue/?printer_id={printer.id}&target_model=NONE")
- assigned_items = sorted(assigned_scope_response.json(), key=lambda item: item["position"])
- assert [item["archive_id"] for item in assigned_items] == [assigned.id]
- assert [item["position"] for item in assigned_items] == [1]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_insert_position_greater_than_max_appends_without_gap(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Oversized explicit insert_position appends at max+1 instead of creating sparse positions."""
- printer = await printer_factory()
- first = await archive_factory(print_name="First")
- second = await archive_factory(print_name="Second")
- appended = await archive_factory(print_name="Append")
- assert (
- await async_client.post("/api/v1/queue/", json={"printer_id": printer.id, "archive_id": first.id})
- ).status_code == 200
- assert (
- await async_client.post("/api/v1/queue/", json={"printer_id": printer.id, "archive_id": second.id})
- ).status_code == 200
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": appended.id,
- "insert_position": 99,
- },
- )
- assert response.status_code == 200
- assert response.json()["position"] == 3
- list_response = await async_client.get(f"/api/v1/queue/?printer_id={printer.id}")
- items = sorted(list_response.json(), key=lambda item: item["position"])
- assert [item["archive_id"] for item in items] == [first.id, second.id, appended.id]
- assert [item["position"] for item in items] == [1, 2, 3]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_consecutive_asap_inserts_stack_in_submission_order(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Consecutive ASAP inserts to the same printer preserve the client submission order."""
- printer = await printer_factory()
- existing = await archive_factory(print_name="Existing")
- first_asap = await archive_factory(print_name="First ASAP")
- second_asap = await archive_factory(print_name="Second ASAP")
- assert (
- await async_client.post("/api/v1/queue/", json={"printer_id": printer.id, "archive_id": existing.id})
- ).status_code == 200
- first_response = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": first_asap.id,
- "insert_position": 1,
- },
- )
- assert first_response.status_code == 200
- second_response = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": second_asap.id,
- "insert_position": 2,
- },
- )
- assert second_response.status_code == 200
- list_response = await async_client.get(f"/api/v1/queue/?printer_id={printer.id}")
- items = sorted(list_response.json(), key=lambda item: item["position"])
- assert [item["archive_id"] for item in items] == [first_asap.id, second_asap.id, existing.id]
- assert [item["position"] for item in items] == [1, 2, 3]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_quantity_with_print_options(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Verify print options are applied to all batch items."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "quantity": 2,
- "bed_levelling": "off",
- "timelapse": True,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- assert response.status_code == 200
- batch_id = response.json()["batch_id"]
- list_response = await async_client.get("/api/v1/queue/")
- batch_items = [i for i in list_response.json() if i["batch_id"] == batch_id]
- assert len(batch_items) == 2
- for item in batch_items:
- assert item["bed_levelling"] == "off"
- assert item["timelapse"] is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_get_batch(self, async_client: AsyncClient, printer_factory, archive_factory, db_session):
- """Verify batch can be retrieved with progress stats."""
- printer = await printer_factory()
- archive = await archive_factory()
- # Create a batch of 3
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "quantity": 3,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- batch_id = response.json()["batch_id"]
- # Get batch
- response = await async_client.get(f"/api/v1/queue/batches/{batch_id}")
- assert response.status_code == 200
- result = response.json()
- assert result["id"] == batch_id
- assert result["quantity"] == 3
- assert result["status"] == "active"
- assert result["pending_count"] == 3
- assert result["printing_count"] == 0
- assert result["completed_count"] == 0
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_list_batches(self, async_client: AsyncClient, printer_factory, archive_factory, db_session):
- """Verify batches can be listed."""
- printer = await printer_factory()
- archive = await archive_factory()
- # Create two batches
- for qty in [2, 3]:
- await async_client.post(
- "/api/v1/queue/",
- json={"printer_id": printer.id, "archive_id": archive.id, "quantity": qty},
- )
- response = await async_client.get("/api/v1/queue/batches")
- assert response.status_code == 200
- batches = response.json()
- assert len(batches) >= 2
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_cancel_batch(self, async_client: AsyncClient, printer_factory, archive_factory, db_session):
- """Verify cancelling a batch cancels all pending items."""
- printer = await printer_factory()
- archive = await archive_factory()
- data = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "quantity": 3,
- }
- response = await async_client.post("/api/v1/queue/", json=data)
- batch_id = response.json()["batch_id"]
- # Cancel the batch
- response = await async_client.delete(f"/api/v1/queue/batches/{batch_id}")
- assert response.status_code == 200
- # Verify all items are cancelled
- list_response = await async_client.get("/api/v1/queue/")
- batch_items = [i for i in list_response.json() if i["batch_id"] == batch_id]
- for item in batch_items:
- assert item["status"] == "cancelled"
- # Verify batch status
- batch_response = await async_client.get(f"/api/v1/queue/batches/{batch_id}")
- assert batch_response.json()["status"] == "cancelled"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_get_batch_not_found(self, async_client: AsyncClient):
- """Verify 404 for non-existent batch."""
- response = await async_client.get("/api/v1/queue/batches/9999")
- assert response.status_code == 404
- # ========================================================================
- # Queue redesign: create-empty + group-existing + ungroup
- # ========================================================================
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_create_empty_batch_for_client_side_grouping(
- self, async_client: AsyncClient, printer_factory, archive_factory
- ):
- """Verify POST /queue/batches without item_ids creates an empty batch
- whose id can be passed on subsequent /queue/ POSTs (the multi-plate
- auto-batch flow). Subsequent items must end up with the same batch_id."""
- printer = await printer_factory()
- archive = await archive_factory()
- # 1. Pre-create batch
- batch_resp = await async_client.post(
- "/api/v1/queue/batches",
- json={"name": "Plates · 2 plates", "archive_id": archive.id},
- )
- assert batch_resp.status_code == 200
- batch = batch_resp.json()
- assert batch["status"] == "active"
- batch_id = batch["id"]
- # 2. Add two items referencing that batch
- for plate_id in (1, 2):
- item_resp = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": archive.id,
- "plate_id": plate_id,
- "batch_id": batch_id,
- },
- )
- assert item_resp.status_code == 200
- assert item_resp.json()["batch_id"] == batch_id
- # 3. Verify batch now has 2 pending children
- list_resp = await async_client.get("/api/v1/queue/")
- siblings = [i for i in list_resp.json() if i["batch_id"] == batch_id]
- assert len(siblings) == 2
- assert {i["plate_id"] for i in siblings} == {1, 2}
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_group_existing_items_as_batch(
- self, async_client: AsyncClient, printer_factory, archive_factory, queue_item_factory
- ):
- """Verify POST /queue/batches with item_ids assigns batch_id to
- existing pending items (the 'Group as batch' UI action)."""
- printer = await printer_factory()
- archive = await archive_factory()
- item_a = await queue_item_factory(printer_id=printer.id, archive_id=archive.id, status="pending")
- item_b = await queue_item_factory(printer_id=printer.id, archive_id=archive.id, status="pending")
- resp = await async_client.post(
- "/api/v1/queue/batches",
- json={"name": "Manual group", "item_ids": [item_a.id, item_b.id]},
- )
- assert resp.status_code == 200
- batch_id = resp.json()["id"]
- list_resp = await async_client.get("/api/v1/queue/")
- grouped = [i for i in list_resp.json() if i["batch_id"] == batch_id]
- assert {i["id"] for i in grouped} == {item_a.id, item_b.id}
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_group_skips_non_pending_items(
- self, async_client: AsyncClient, printer_factory, archive_factory, queue_item_factory
- ):
- """Verify grouping doesn't pull in already-completed/cancelled items."""
- printer = await printer_factory()
- archive = await archive_factory()
- pending = await queue_item_factory(printer_id=printer.id, archive_id=archive.id, status="pending")
- completed = await queue_item_factory(printer_id=printer.id, archive_id=archive.id, status="completed")
- resp = await async_client.post(
- "/api/v1/queue/batches",
- json={"name": "Mixed", "item_ids": [pending.id, completed.id]},
- )
- assert resp.status_code == 200
- batch_id = resp.json()["id"]
- list_resp = await async_client.get("/api/v1/queue/")
- grouped = [i for i in list_resp.json() if i["batch_id"] == batch_id]
- assert {i["id"] for i in grouped} == {pending.id}
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_create_batch_requires_name(self, async_client: AsyncClient):
- """Verify empty / whitespace-only name is rejected with 400."""
- resp = await async_client.post("/api/v1/queue/batches", json={"name": " "})
- assert resp.status_code == 400
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_ungroup_batch_clears_batch_id_and_deletes_row(
- self, async_client: AsyncClient, printer_factory, archive_factory
- ):
- """Verify POST /queue/batches/{id}/ungroup clears batch_id from all
- members and deletes the batch row when nothing remains assigned."""
- printer = await printer_factory()
- archive = await archive_factory()
- # Create batch with two items via the existing quantity flow
- add_resp = await async_client.post(
- "/api/v1/queue/",
- json={"printer_id": printer.id, "archive_id": archive.id, "quantity": 2},
- )
- batch_id = add_resp.json()["batch_id"]
- # Ungroup
- ungroup_resp = await async_client.post(f"/api/v1/queue/batches/{batch_id}/ungroup")
- assert ungroup_resp.status_code == 200
- assert ungroup_resp.json()["ungrouped_count"] == 2
- # Verify items still exist but no longer batched
- list_resp = await async_client.get("/api/v1/queue/")
- ex_members = [i for i in list_resp.json() if i["batch_id"] == batch_id]
- assert ex_members == []
- # Batch row was deleted
- get_resp = await async_client.get(f"/api/v1/queue/batches/{batch_id}")
- assert get_resp.status_code == 404
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_add_to_queue_with_unknown_batch_id_404(
- self, async_client: AsyncClient, printer_factory, archive_factory
- ):
- """Verify addToQueue with a non-existent batch_id is rejected."""
- printer = await printer_factory()
- archive = await archive_factory()
- resp = await async_client.post(
- "/api/v1/queue/",
- json={
- "printer_id": printer.id,
- "archive_id": archive.id,
- "batch_id": 99999,
- },
- )
- assert resp.status_code == 404
- # ========================================================================
- # Soft-deleted archive handling (#1348 follow-up)
- # ========================================================================
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_soft_delete_archive_deletes_all_related_queue_items(
- self, async_client: AsyncClient, printer_factory, archive_factory, queue_item_factory, db_session
- ):
- """Soft-deleting an archive removes every related queue item, regardless
- of status (#1734). Pre-#1734 only ``pending`` rows were flipped to
- ``cancelled`` and stayed in the DB, surprising users who expected the
- queue lines to disappear with the archive — especially on multi-plate
- Send All uploads (#1733), where ONE archive backed N queue items and
- soft-deleting the archive left N "cancelled" rows behind. The change
- keeps the printing guard (a row with ``status='printing'`` blocks the
- delete one layer up at the API route), so we never delete the row of
- an actively-running print here.
- Print history lives in ``PrintLogEntry`` (FK ``ON DELETE SET NULL``) —
- the audit trail survives independently of the queue rows.
- """
- from sqlalchemy import select
- from backend.app.models.print_queue import PrintQueueItem
- from backend.app.services.archive import ArchiveService
- printer = await printer_factory()
- archive = await archive_factory(thumbnail_path="archives/test/test/thumbnail.png")
- pending = await queue_item_factory(printer_id=printer.id, archive_id=archive.id, status="pending")
- completed = await queue_item_factory(printer_id=printer.id, archive_id=archive.id, status="completed")
- service = ArchiveService(db_session)
- assert await service.soft_delete_archive(archive.id) is True
- # Every queue row that referenced this archive is gone — both the
- # pending and the completed rows. Print history (PrintLogEntry) is
- # the authoritative record and is preserved by the FK SET NULL.
- remaining = (
- (await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.id.in_([pending.id, completed.id]))))
- .scalars()
- .all()
- )
- assert remaining == [], (
- "Soft-deleting the archive must delete every related queue row, "
- f"got {[(r.id, r.status) for r in remaining]} still present"
- )
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_queue_api_hides_archive_surface_when_soft_deleted(
- self, async_client: AsyncClient, printer_factory, archive_factory, queue_item_factory, db_session
- ):
- """Queue serializer must NOT populate archive_thumbnail / archive_name
- when the archive is soft-deleted — otherwise the frontend renders a
- broken <img> and 404-storms the thumbnail / plates / plate-thumbnail
- endpoints. archive_deleted=True signals the soft-deleted state so
- the UI can render a 'source deleted' badge."""
- from datetime import datetime, timezone
- printer = await printer_factory()
- archive = await archive_factory(
- print_name="Test Print",
- thumbnail_path="archives/test/test/thumbnail.png",
- deleted_at=datetime.now(timezone.utc), # Pre-soft-deleted
- )
- item = await queue_item_factory(printer_id=printer.id, archive_id=archive.id, status="cancelled")
- resp = await async_client.get("/api/v1/queue/")
- assert resp.status_code == 200
- body = resp.json()
- row = next((r for r in body if r["id"] == item.id), None)
- assert row is not None
- assert row["archive_deleted"] is True
- assert row["archive_thumbnail"] is None, "must not expose stale thumbnail path for soft-deleted archive"
- assert row["archive_name"] is None
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_queue_api_still_exposes_archive_surface_when_live(
- self, async_client: AsyncClient, printer_factory, archive_factory, queue_item_factory, db_session
- ):
- """Sanity guard: the soft-delete suppression must not affect live
- archives. archive_name / archive_thumbnail still flow through and
- archive_deleted stays False."""
- printer = await printer_factory()
- archive = await archive_factory(
- print_name="Live Archive",
- thumbnail_path="archives/test/live/thumbnail.png",
- )
- item = await queue_item_factory(printer_id=printer.id, archive_id=archive.id, status="pending")
- resp = await async_client.get("/api/v1/queue/")
- assert resp.status_code == 200
- row = next((r for r in resp.json() if r["id"] == item.id), None)
- assert row is not None
- assert row["archive_deleted"] is False
- assert row["archive_name"] == "Live Archive"
- assert row["archive_thumbnail"] == "archives/test/live/thumbnail.png"
- class TestResumeQueueAfterFailure:
- """Integration tests for POST /api/v1/queue/printer/{id}/resume (#1818)."""
- @pytest.fixture
- async def printer_factory(self, db_session):
- _counter = [0]
- async def _create_printer(**kwargs):
- from backend.app.models.printer import Printer
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "name": f"Resume Printer {counter}",
- "ip_address": f"192.168.42.{100 + counter}",
- "serial_number": f"RESUMESERIAL{counter:04d}",
- "access_code": "12345678",
- "model": "P1S",
- }
- defaults.update(kwargs)
- printer = Printer(**defaults)
- db_session.add(printer)
- await db_session.commit()
- await db_session.refresh(printer)
- return printer
- return _create_printer
- @pytest.fixture
- async def archive_factory(self, db_session):
- _counter = [0]
- async def _create_archive(**kwargs):
- from backend.app.models.archive import PrintArchive
- _counter[0] += 1
- counter = _counter[0]
- defaults = {
- "filename": f"resume_print_{counter}.3mf",
- "print_name": f"Resume Print {counter}",
- "file_path": f"/tmp/resume_print_{counter}.3mf", # nosec B108
- "file_size": 1024,
- "content_hash": f"resumehash{counter:08d}",
- "status": "completed",
- }
- defaults.update(kwargs)
- archive = PrintArchive(**defaults)
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- return archive
- return _create_archive
- async def _add_item(self, db_session, printer, archive_factory, **kwargs):
- from backend.app.models.print_queue import PrintQueueItem
- archive = await archive_factory()
- defaults = {
- "printer_id": printer.id,
- "archive_id": archive.id,
- "status": "pending",
- "require_previous_success": True,
- }
- defaults.update(kwargs)
- item = PrintQueueItem(**defaults)
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- return item
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_resume_unknown_printer_returns_404(self, async_client: AsyncClient):
- resp = await async_client.post("/api/v1/queue/printer/999999/resume")
- assert resp.status_code == 404
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_resume_no_op_on_clean_queue(self, async_client: AsyncClient, printer_factory):
- """Calling resume on a printer with no failures and no skipped items
- returns zero counts — endpoint is idempotent and safe to spam."""
- printer = await printer_factory()
- resp = await async_client.post(f"/api/v1/queue/printer/{printer.id}/resume")
- assert resp.status_code == 200
- assert resp.json() == {"acknowledged": 0, "restored": 0}
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_resume_acknowledges_failed_and_restores_skipped(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Reporter's scenario: failed predecessor + N skipped downstream items.
- Resume sets gate_acknowledged on the failure and flips skipped → pending."""
- from sqlalchemy import select
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- failed = await self._add_item(db_session, printer, archive_factory, status="failed")
- skipped_1 = await self._add_item(
- db_session,
- printer,
- archive_factory,
- status="skipped",
- error_message="Previous print failed or was aborted",
- )
- skipped_2 = await self._add_item(
- db_session,
- printer,
- archive_factory,
- status="skipped",
- error_message="Previous print failed or was aborted",
- )
- resp = await async_client.post(f"/api/v1/queue/printer/{printer.id}/resume")
- assert resp.status_code == 200
- assert resp.json() == {"acknowledged": 1, "restored": 2}
- failed_id = failed.id
- skipped_ids = [skipped_1.id, skipped_2.id]
- db_session.expire_all()
- result = await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.id == failed_id))
- assert result.scalar_one().gate_acknowledged is True
- for sid in skipped_ids:
- result = await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.id == sid))
- row = result.scalar_one()
- assert row.status == "pending"
- assert row.error_message is None
- assert row.completed_at is None
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_resume_preserves_skipped_items_with_other_reasons(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Skipped items whose error_message is something OTHER than the
- gate string (e.g. filament-deficit promotion, future skip reasons)
- must not be touched — they encode different user intent."""
- from sqlalchemy import select
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- gate_skip = await self._add_item(
- db_session,
- printer,
- archive_factory,
- status="skipped",
- error_message="Previous print failed or was aborted",
- )
- other_skip = await self._add_item(
- db_session,
- printer,
- archive_factory,
- status="skipped",
- error_message="User skipped via UI",
- )
- gate_id = gate_skip.id
- other_id = other_skip.id
- resp = await async_client.post(f"/api/v1/queue/printer/{printer.id}/resume")
- assert resp.json() == {"acknowledged": 0, "restored": 1}
- db_session.expire_all()
- result = await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.id == gate_id))
- assert result.scalar_one().status == "pending"
- result = await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.id == other_id))
- assert result.scalar_one().status == "skipped"
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_resume_scoped_to_printer(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """A resume on printer A must not clear printer B's gate — farms run
- each printer's queue independently."""
- from sqlalchemy import select
- from backend.app.models.print_queue import PrintQueueItem
- p1 = await printer_factory()
- p2 = await printer_factory()
- failed_p1 = await self._add_item(db_session, p1, archive_factory, status="failed")
- failed_p2 = await self._add_item(db_session, p2, archive_factory, status="failed")
- failed_p1_id = failed_p1.id
- failed_p2_id = failed_p2.id
- resp = await async_client.post(f"/api/v1/queue/printer/{p1.id}/resume")
- assert resp.json() == {"acknowledged": 1, "restored": 0}
- db_session.expire_all()
- result = await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.id == failed_p1_id))
- assert result.scalar_one().gate_acknowledged is True
- result = await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.id == failed_p2_id))
- assert result.scalar_one().gate_acknowledged is False
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_resume_handles_aborted_status(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Aborted prints (printer-detected mid-print failure) gate the same
- way failed prints do and must also be acknowledgeable."""
- from sqlalchemy import select
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- aborted = await self._add_item(db_session, printer, archive_factory, status="aborted")
- aborted_id = aborted.id
- resp = await async_client.post(f"/api/v1/queue/printer/{printer.id}/resume")
- assert resp.json() == {"acknowledged": 1, "restored": 0}
- db_session.expire_all()
- result = await db_session.execute(select(PrintQueueItem).where(PrintQueueItem.id == aborted_id))
- assert result.scalar_one().gate_acknowledged is True
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_resume_idempotent_second_call_is_no_op(
- self, async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Calling resume twice on the same printer doesn't re-acknowledge
- the same failure — the second call sees acknowledged=0, restored=0."""
- printer = await printer_factory()
- await self._add_item(db_session, printer, archive_factory, status="failed")
- await self._add_item(
- db_session,
- printer,
- archive_factory,
- status="skipped",
- error_message="Previous print failed or was aborted",
- )
- first = await async_client.post(f"/api/v1/queue/printer/{printer.id}/resume")
- assert first.json() == {"acknowledged": 1, "restored": 1}
- second = await async_client.post(f"/api/v1/queue/printer/{printer.id}/resume")
- assert second.json() == {"acknowledged": 0, "restored": 0}
- class TestReorderEndpoint:
- """Tests for the /queue/reorder endpoint (#1625-followup duplicate-position validator)."""
- @pytest.fixture
- async def printer_factory(self, db_session):
- async def _create(**kwargs):
- from backend.app.models.printer import Printer
- defaults = {
- "name": "Reorder Test Printer",
- "ip_address": "192.168.1.220",
- "serial_number": "TESTREORDER001",
- "access_code": "12345678",
- "model": "X1C",
- }
- defaults.update(kwargs)
- printer = Printer(**defaults)
- db_session.add(printer)
- await db_session.commit()
- await db_session.refresh(printer)
- return printer
- return _create
- @pytest.fixture
- async def archive_factory(self, db_session):
- _counter = [0]
- async def _create(**kwargs):
- from backend.app.models.archive import PrintArchive
- _counter[0] += 1
- defaults = {
- "filename": f"reorder_{_counter[0]}.3mf",
- "print_name": f"Reorder {_counter[0]}",
- "file_path": f"/tmp/reorder_{_counter[0]}.3mf", # nosec B108
- "file_size": 1024,
- "content_hash": f"reorderhash{_counter[0]:06d}",
- "status": "completed",
- }
- defaults.update(kwargs)
- archive = PrintArchive(**defaults)
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- return archive
- return _create
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_reorder_rejects_duplicate_positions(
- self, async_client: AsyncClient, db_session, printer_factory, archive_factory
- ):
- """Reorder payload with duplicate positions → 422 at schema layer.
- Regression guard: pre-fix, a buggy client sending two items at the
- same position would leave the queue in an inconsistent state (the
- scheduler's ORDER BY (printer_id, position) tie would be broken by
- physical row order — non-deterministic dispatch order).
- """
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- a1 = await archive_factory()
- a2 = await archive_factory()
- item1 = PrintQueueItem(printer_id=printer.id, archive_id=a1.id, status="pending", position=1)
- item2 = PrintQueueItem(printer_id=printer.id, archive_id=a2.id, status="pending", position=2)
- db_session.add_all([item1, item2])
- await db_session.commit()
- await db_session.refresh(item1)
- await db_session.refresh(item2)
- response = await async_client.post(
- "/api/v1/queue/reorder",
- json={
- "items": [
- {"id": item1.id, "position": 1},
- {"id": item2.id, "position": 1}, # duplicate
- ]
- },
- )
- assert response.status_code == 422
- body = response.json()
- # Pydantic v2 wraps custom validator errors; the message must mention "Duplicate"
- # so the FE can surface the actionable detail.
- assert any("duplicate" in str(err).lower() for err in body.get("detail", []))
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_reorder_accepts_unique_positions(
- self, async_client: AsyncClient, db_session, printer_factory, archive_factory
- ):
- """Reorder with unique positions succeeds and updates them in DB."""
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory()
- a1 = await archive_factory()
- a2 = await archive_factory()
- item1 = PrintQueueItem(printer_id=printer.id, archive_id=a1.id, status="pending", position=1)
- item2 = PrintQueueItem(printer_id=printer.id, archive_id=a2.id, status="pending", position=2)
- db_session.add_all([item1, item2])
- await db_session.commit()
- await db_session.refresh(item1)
- await db_session.refresh(item2)
- response = await async_client.post(
- "/api/v1/queue/reorder",
- json={
- "items": [
- {"id": item1.id, "position": 2},
- {"id": item2.id, "position": 1},
- ]
- },
- )
- assert response.status_code == 200
- await db_session.refresh(item1)
- await db_session.refresh(item2)
- assert item1.position == 2
- assert item2.position == 1
- class TestForceColorOverridesAreScopedToThePlate:
- """Queueing several plates of one 3MF must not make each plate wait on the
- colours of its siblings (#2551).
- The print dialog builds one override list from every selected plate and posts
- that same list with each plate's item, so the API is what has to keep only the
- slots the plate prints -- a ``force_color_match`` entry blocks dispatch until
- the printer has that exact colour loaded.
- """
- THREE_PLATES = """<?xml version="1.0" encoding="UTF-8"?>
- <config>
- <plate>
- <metadata key="index" value="1"/>
- <filament id="1" used_g="50.0" type="PLA" color="#0B2C7A"/>
- </plate>
- <plate>
- <metadata key="index" value="2"/>
- <filament id="2" used_g="40.0" type="PLA" color="#9B9EA0"/>
- </plate>
- <plate>
- <metadata key="index" value="3"/>
- <filament id="3" used_g="30.0" type="PLA" color="#F4EE2A"/>
- </plate>
- </config>
- """
- # What the dialog posts for every plate: the union of all three plates'
- # filaments, each one force-matched.
- ALL_THREE_COLORS = [
- {"slot_id": 1, "type": "PLA", "color": "#0B2C7A", "color_name": "Army Blue", "force_color_match": True},
- {"slot_id": 2, "type": "PLA", "color": "#9B9EA0", "color_name": "Ash Grey", "force_color_match": True},
- {"slot_id": 3, "type": "PLA", "color": "#F4EE2A", "color_name": "Sunshine Yellow", "force_color_match": True},
- ]
- @pytest.fixture
- async def multi_plate_archive(self, db_session, tmp_path):
- """An archive whose 3MF really exists on disk, one colour per plate."""
- import zipfile
- from backend.app.models.archive import PrintArchive
- file_path = tmp_path / "three_plates.gcode.3mf"
- with zipfile.ZipFile(file_path, "w") as zf:
- zf.writestr("Metadata/slice_info.config", self.THREE_PLATES)
- archive = PrintArchive(
- filename="three_plates.gcode.3mf",
- print_name="Three Plates",
- file_path=str(file_path),
- file_size=file_path.stat().st_size,
- content_hash="platehash0001",
- status="completed",
- )
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- return archive
- @pytest.fixture
- async def x1c(self, db_session):
- from backend.app.models.printer import Printer
- printer = Printer(
- name="Force Color X1C",
- ip_address="192.168.1.210",
- serial_number="FORCECOLOR01",
- access_code="12345678",
- model="X1C",
- )
- db_session.add(printer)
- await db_session.commit()
- await db_session.refresh(printer)
- return printer
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_each_plate_keeps_only_the_colour_it_prints(
- self, async_client: AsyncClient, multi_plate_archive, x1c
- ):
- """The bug: plate 1 prints Army Blue only, but was stored demanding all three."""
- stored = {}
- for plate_id in (1, 2, 3):
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "target_model": "X1C",
- "archive_id": multi_plate_archive.id,
- "plate_id": plate_id,
- "filament_overrides": self.ALL_THREE_COLORS,
- },
- )
- assert response.status_code == 200
- stored[plate_id] = response.json()["filament_overrides"]
- assert [o["color_name"] for o in stored[1]] == ["Army Blue"]
- assert [o["color_name"] for o in stored[2]] == ["Ash Grey"]
- assert [o["color_name"] for o in stored[3]] == ["Sunshine Yellow"]
- # The slot each entry maps to has to survive narrowing untouched, or the
- # dispatch-time AMS mapping would key the override onto the wrong slot.
- assert [o["slot_id"] for o in stored[2]] == [2]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_whole_file_queue_keeps_every_colour(self, async_client: AsyncClient, multi_plate_archive, x1c):
- """No plate_id means the job prints the whole file, so every colour is needed."""
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "target_model": "X1C",
- "archive_id": multi_plate_archive.id,
- "filament_overrides": self.ALL_THREE_COLORS,
- },
- )
- assert response.status_code == 200
- assert len(response.json()["filament_overrides"]) == 3
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_unreadable_3mf_keeps_every_colour(self, async_client: AsyncClient, db_session, tmp_path, x1c):
- """When the plate's slots can't be read, keep the overrides rather than drop them.
- An item waiting on a colour it doesn't need is visible and fixable; one that
- silently lost its forced colour would dispatch in the wrong filament.
- """
- from backend.app.models.archive import PrintArchive
- file_path = tmp_path / "not_a_zip.gcode.3mf"
- file_path.write_text("this is not a 3mf")
- archive = PrintArchive(
- filename="not_a_zip.gcode.3mf",
- print_name="Corrupt",
- file_path=str(file_path),
- file_size=file_path.stat().st_size,
- content_hash="platehash0002",
- status="completed",
- )
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "target_model": "X1C",
- "archive_id": archive.id,
- "plate_id": 1,
- "filament_overrides": self.ALL_THREE_COLORS,
- },
- )
- assert response.status_code == 200
- assert len(response.json()["filament_overrides"]) == 3
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_required_types_stay_scoped_to_the_plate(self, async_client: AsyncClient, db_session, tmp_path, x1c):
- """Override types are merged into required_filament_types, so a shared list
- also widened the type gate -- a PLA-only plate demanded PETG as well."""
- import zipfile
- from backend.app.models.archive import PrintArchive
- xml = """<?xml version="1.0" encoding="UTF-8"?>
- <config>
- <plate>
- <metadata key="index" value="1"/>
- <filament id="1" used_g="50.0" type="PLA" color="#0B2C7A"/>
- </plate>
- <plate>
- <metadata key="index" value="2"/>
- <filament id="2" used_g="40.0" type="PETG" color="#9B9EA0"/>
- </plate>
- </config>
- """
- file_path = tmp_path / "mixed_types.gcode.3mf"
- with zipfile.ZipFile(file_path, "w") as zf:
- zf.writestr("Metadata/slice_info.config", xml)
- archive = PrintArchive(
- filename="mixed_types.gcode.3mf",
- print_name="Mixed",
- file_path=str(file_path),
- file_size=file_path.stat().st_size,
- content_hash="platehash0003",
- status="completed",
- )
- db_session.add(archive)
- await db_session.commit()
- await db_session.refresh(archive)
- response = await async_client.post(
- "/api/v1/queue/",
- json={
- "target_model": "X1C",
- "archive_id": archive.id,
- "plate_id": 1,
- "filament_overrides": [
- {"slot_id": 1, "type": "PLA", "color": "#0B2C7A", "force_color_match": True},
- {"slot_id": 2, "type": "PETG", "color": "#9B9EA0", "force_color_match": True},
- ],
- },
- )
- assert response.status_code == 200
- assert response.json()["required_filament_types"] == ["PLA"]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_editing_an_item_narrows_the_overrides_too(
- self, async_client: AsyncClient, db_session, multi_plate_archive, x1c
- ):
- """The edit dialog posts the same shared list, so PATCH narrows it as well."""
- from backend.app.models.print_queue import PrintQueueItem
- item = PrintQueueItem(
- target_model="X1C",
- archive_id=multi_plate_archive.id,
- plate_id=2,
- status="pending",
- position=1,
- )
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- response = await async_client.patch(
- f"/api/v1/queue/{item.id}",
- json={"filament_overrides": self.ALL_THREE_COLORS},
- )
- assert response.status_code == 200
- assert [o["color_name"] for o in response.json()["filament_overrides"]] == ["Ash Grey"]
- @pytest.mark.asyncio
- @pytest.mark.integration
- async def test_editing_the_plate_renarrows_against_the_new_plate(
- self, async_client: AsyncClient, db_session, multi_plate_archive, x1c
- ):
- """Moving an item to another plate must re-scope its colours to that plate."""
- from backend.app.models.print_queue import PrintQueueItem
- item = PrintQueueItem(
- target_model="X1C",
- archive_id=multi_plate_archive.id,
- plate_id=1,
- status="pending",
- position=1,
- )
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- response = await async_client.patch(
- f"/api/v1/queue/{item.id}",
- json={"plate_id": 3, "filament_overrides": self.ALL_THREE_COLORS},
- )
- assert response.status_code == 200
- assert [o["color_name"] for o in response.json()["filament_overrides"]] == ["Sunshine Yellow"]
- @pytest.mark.asyncio
- async def test_stop_offline_reconciles_linked_archive_status_2603(
- async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """Stopping a printing item while the printer is offline must also close out its
- archive (#2603).
- When the stop command reaches the printer, the later MQTT completion event flips
- the archive to cancelled. When the printer is offline no such event ever arrives,
- so without this the archive stays "printing" forever while the queue row is
- already cancelled — the reporter's archive 436. The offline branch reconciles the
- archive directly.
- """
- from unittest.mock import MagicMock, patch
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory(name="Offline printer")
- archive = await archive_factory(
- printer.id, status="printing", plate_id=22, filename="heart 3.gcode.3mf", with_run=False
- )
- item = PrintQueueItem(printer_id=printer.id, archive_id=archive.id, status="printing")
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- # stop_print returns False => printer offline / not connected.
- with patch(
- "backend.app.services.printer_manager.printer_manager.stop_print",
- MagicMock(return_value=False),
- ):
- resp = await async_client.post(f"/api/v1/queue/{item.id}/stop")
- assert resp.status_code == 200
- await db_session.refresh(item)
- await db_session.refresh(archive)
- assert item.status == "cancelled"
- assert archive.status == "cancelled", "an offline stop must reconcile the archive, not leave it 'printing'"
- assert archive.completed_at is not None
- assert archive.failure_reason == "Stopped by user (printer was offline)"
- @pytest.mark.asyncio
- async def test_stop_online_leaves_archive_for_mqtt_to_reconcile_2603(
- async_client: AsyncClient, printer_factory, archive_factory, db_session
- ):
- """When the stop command reaches the printer, the archive is left to the MQTT
- completion path — the offline reconcile must NOT fire and pre-empt it."""
- from unittest.mock import MagicMock, patch
- from backend.app.models.print_queue import PrintQueueItem
- printer = await printer_factory(name="Online printer")
- archive = await archive_factory(printer.id, status="printing", filename="heart 3.gcode.3mf", with_run=False)
- item = PrintQueueItem(printer_id=printer.id, archive_id=archive.id, status="printing")
- db_session.add(item)
- await db_session.commit()
- await db_session.refresh(item)
- with patch(
- "backend.app.services.printer_manager.printer_manager.stop_print",
- MagicMock(return_value=True),
- ):
- resp = await async_client.post(f"/api/v1/queue/{item.id}/stop")
- assert resp.status_code == 200
- await db_session.refresh(item)
- await db_session.refresh(archive)
- assert item.status == "cancelled"
- assert archive.status == "printing", "an online stop must leave the archive for the MQTT completion path"
|