nan
f48bbb3d05
- Implement tests for image layer functionality including alpha curve and depth layer behavior. - Create tests for image plane evaluation and modifier input handling. - Add tests to verify the saved asset inventory and its persistence. - Introduce tests for the documentation command to ensure correct project version deployment. - Validate the extension manifest structure and required permissions. - Develop tests for the packaging process, ensuring correct platform handling and dependency management. - Enhance image depth plane group functionality with new split threshold input and surface processing.
135 lines
4.6 KiB
Python
135 lines
4.6 KiB
Python
import importlib
|
|
from types import SimpleNamespace
|
|
from unittest.mock import patch
|
|
from tests.support.blender import BlenderTestCase
|
|
|
|
|
|
class ServerPanelTest(BlenderTestCase):
|
|
|
|
|
|
def test_server_panel_shows_controls_for_every_running_state(self):
|
|
for state, label, icon in (
|
|
("READY", "Running", "RADIOBUT_ON"),
|
|
("BUSY", "Running (Busy)", "TIME"),
|
|
):
|
|
with self.subTest(state=state):
|
|
events = self._draw_server_panel(state)
|
|
operators = [identifier for identifier, _enabled in events.operators]
|
|
self.assertIn(("row", label, icon), events.labels)
|
|
self.assertIn("anyimage.restart_server", operators)
|
|
self.assertIn("anyimage.stop_server", operators)
|
|
|
|
for state in ("STOPPED", "STARTING", "ERROR"):
|
|
with self.subTest(state=state):
|
|
events = self._draw_server_panel(state)
|
|
operators = [identifier for identifier, _enabled in events.operators]
|
|
self.assertNotIn("anyimage.restart_server", operators)
|
|
self.assertNotIn("anyimage.stop_server", operators)
|
|
|
|
|
|
def test_server_panel_wraps_multiple_cached_models(self):
|
|
for cached_names, expected_labels in (
|
|
((), [("row", "Cached: None")]),
|
|
(("Model A",), [("row", "Cached: Model A")]),
|
|
(
|
|
("Model A", "Model B"),
|
|
[
|
|
("column", "Cached:"),
|
|
("column", "Model A"),
|
|
("column", "Model B"),
|
|
],
|
|
),
|
|
):
|
|
with self.subTest(cached_names=cached_names):
|
|
events = self._draw_server_panel("READY", cached_names)
|
|
cached_labels = [
|
|
(kind, text)
|
|
for kind, text, _icon in events.labels
|
|
if text != "Running"
|
|
]
|
|
self.assertEqual(cached_labels, expected_labels)
|
|
operators = [identifier for identifier, _enabled in events.operators]
|
|
self.assertIn("anyimage.clear_models", operators)
|
|
|
|
|
|
def test_status_bar_reads_runtime_state(self):
|
|
calls = []
|
|
|
|
class Layout:
|
|
def row(self, **_options):
|
|
return self
|
|
|
|
def progress(self, **options):
|
|
calls.append(("progress", options))
|
|
|
|
def operator(self, identifier, **_options):
|
|
calls.append(("operator", identifier))
|
|
|
|
runtime = self.anyimage.runtime
|
|
runtime.active_job = object()
|
|
runtime.progress = 0.4
|
|
runtime.message = "Working"
|
|
runtime._draw_status_bar(SimpleNamespace(layout=Layout()), None)
|
|
runtime.active_job = None
|
|
|
|
self.assertEqual(calls[0][1]["factor"], 0.4)
|
|
self.assertEqual(calls[0][1]["text"], "Working")
|
|
self.assertEqual(calls[1], ("operator", "anyimage.cancel_job"))
|
|
|
|
|
|
def _draw_server_panel(self, state, cached_names=()):
|
|
events = SimpleNamespace(labels=[], operators=[])
|
|
|
|
class Layout:
|
|
def __init__(self, kind="layout"):
|
|
self.kind = kind
|
|
self.enabled = True
|
|
|
|
def row(self, **_options):
|
|
return Layout("row")
|
|
|
|
def column(self, **_options):
|
|
return Layout("column")
|
|
|
|
def label(self, **options):
|
|
events.labels.append(
|
|
(self.kind, options.get("text"), options.get("icon"))
|
|
)
|
|
|
|
def operator(self, identifier, **_options):
|
|
events.operators.append((identifier, self.enabled))
|
|
return SimpleNamespace()
|
|
|
|
panel_module = importlib.import_module("anyimage.panel")
|
|
panel = panel_module.ServerPanel()
|
|
panel.layout = Layout()
|
|
loaded = {
|
|
str(index): name
|
|
for index, name in enumerate(cached_names)
|
|
}
|
|
with (
|
|
patch.object(
|
|
panel_module.runtime,
|
|
"environment_ready",
|
|
return_value=True,
|
|
),
|
|
patch.object(
|
|
panel_module.runtime,
|
|
"server_busy",
|
|
return_value=state == "BUSY",
|
|
),
|
|
patch.object(
|
|
panel_module.runtime,
|
|
"server_status",
|
|
return_value={
|
|
"state": state,
|
|
"message": state.title(),
|
|
"resources": {
|
|
"model_manager": {"loaded": loaded},
|
|
},
|
|
},
|
|
),
|
|
):
|
|
panel.draw(None)
|
|
return events
|