Reuse shared ComfyUI runtime for Deck-owned LTX video

2026-09-30 12:30:08 +02:00
parent ef80b2a0ca
commit fe296cebb1
16 changed files with 350 additions and 34 deletions
@@ -0,0 +1,38 @@
+import tempfile
+import unittest
+from pathlib import Path
+from unittest.mock import Mock
+from video_comfy import VideoComfy
+from profiles import LTX_REPO,LTX_FILE,LTX_COMPONENTS,Profiles
+
+class VideoComfyTests(unittest.TestCase):
+ def setUp(self):
+  self.tmp=tempfile.TemporaryDirectory();self.root=Path(self.tmp.name)
+  self.model=dict(id='a'*64,repo=LTX_REPO,file=LTX_FILE,kind='video',revision='1'*40)
+  self.entries=[self.model]+[dict(id=str(i)*64,repo=LTX_REPO,file=info['files'][0],kind='video',revision='1'*40) for i,info in enumerate(LTX_COMPONENTS.values(),2)]
+  self.catalog=Mock();self.catalog.status.side_effect=lambda:dict(entries=self.entries)
+  def entry(ident):
+   row=next((x for x in self.entries if x['id']==ident),None)
+   if not row:raise ValueError('missing')
+   return row
+  self.catalog.entry.side_effect=entry;self.profiles=Profiles(self.root/'profiles.json',self.catalog)
+  self.runtime=Mock();python=self.root/'python';python.touch();comfy=self.root/'comfy';(comfy/'comfy/ldm/lightricks').mkdir(parents=True)
+  self.runtime.paths.return_value=(python,comfy);self.runtime.status.return_value=dict(installed=True)
+  self.scheduler=Mock();self.scheduler.gpu_mode='llm'
+  helper=Mock();helper.call.return_value={'services':[]}
+  self.video=VideoComfy(self.scheduler,Mock(),helper,self.root/'video',self.catalog,self.profiles,self.runtime)
+ def tearDown(self):self.tmp.cleanup()
+ def test_ready_reuses_runtime_without_install(self):
+  row=self.video.models()[0];self.assertTrue(row['runnable']);self.assertEqual(len(row['components']),4)
+  self.video.select(self.model['id']);self.assertEqual(self.video.status()['selected'],self.model['id']);self.runtime.install.assert_not_called()
+ def test_missing_component_is_not_ready(self):
+  self.entries.pop();row=self.video.models()[0];self.assertFalse(row['runnable']);self.assertIn('Spatial',row['blockers'][0])
+ def test_wrong_revision_is_not_silently_used(self):
+  self.entries[1]['revision']='2'*40;self.assertFalse(self.video.models()[0]['runnable'])
+ def test_missing_runtime_and_model_cannot_start(self):
+  self.runtime.paths.return_value=(self.root/'missing',self.root/'missing-comfy')
+  self.video.select(self.model['id'])
+  with self.assertRaises(ValueError):self.video.switch('video')
+ def test_unknown_models_not_advertised(self):
+  self.model['repo']='unknown/model';self.assertEqual(self.video.models(),[])
+  with self.assertRaises(ValueError):self.video.select(self.model['id'])