MajoorWaldi
68bc7b74ba
- Added `GeneratedFeedTab.js` for managing and displaying generated assets in a bottom panel. - Implemented asset selection logic with support for shift and ctrl/meta key interactions. - Integrated loading of recent assets and handling of pending assets. - Enhanced UI with new styles for compact asset cards in the bottom panel. - Introduced event handling for runtime status updates in `StatusDot.js`. - Updated `floatingViewerManager.js` to utilize the last selected grid for viewer actions. - Created unit tests for comfy API bridge and dialog functionalities. - Added backend support for model path discovery and asset indexing with prompt correlation. - Enhanced scan routes to emit progress and asset indexing events.
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
from mjr_am_backend import config
|
|
|
|
|
|
def test_get_registered_model_paths_reads_folder_paths(monkeypatch):
|
|
class _FolderPaths:
|
|
@staticmethod
|
|
def get_folder_paths(category):
|
|
if category == "checkpoints":
|
|
return ["./models/checkpoints"]
|
|
if category == "loras":
|
|
return ["./models/loras"]
|
|
return []
|
|
|
|
monkeypatch.setitem(__import__("sys").modules, "folder_paths", _FolderPaths)
|
|
|
|
result = config.get_registered_model_paths()
|
|
|
|
assert "checkpoints" in result
|
|
assert "loras" in result
|
|
assert all(isinstance(v, list) for v in result.values())
|
|
|
|
|
|
def test_get_registered_model_paths_handles_missing_module(monkeypatch):
|
|
monkeypatch.delitem(__import__("sys").modules, "folder_paths", raising=False)
|
|
import builtins
|
|
|
|
real_import = builtins.__import__
|
|
|
|
def _fake_import(name, *args, **kwargs):
|
|
if name == "folder_paths":
|
|
raise ImportError("missing folder_paths")
|
|
return real_import(name, *args, **kwargs)
|
|
|
|
monkeypatch.setattr(builtins, "__import__", _fake_import)
|
|
|
|
result = config.get_registered_model_paths()
|
|
|
|
assert result == {}
|