39 lines
2.4 KiB
Python
39 lines
2.4 KiB
Python
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'])
|