Benjamin Lu
9c069c13c5
Desktop runs on Linux now, but get_form_factor mapped a Desktop install on Linux to 'other', losing the platform in registry data.
37 lines
1013 B
Python
37 lines
1013 B
Python
"""Form factor the CNR client reports to the registry."""
|
|
|
|
import pytest
|
|
|
|
from _install_flags_testutil import import_context
|
|
|
|
|
|
@pytest.fixture
|
|
def cnr_utils():
|
|
import_context()
|
|
from comfyui_manager.common import cnr_utils
|
|
|
|
return cnr_utils
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
('system', 'desktop', 'expected'),
|
|
[
|
|
('Windows', True, 'desktop-win'),
|
|
('Darwin', True, 'desktop-mac'),
|
|
('Linux', True, 'desktop-linux'),
|
|
('FreeBSD', True, 'other'),
|
|
('Windows', False, 'git-windows'),
|
|
('Darwin', False, 'git-mac'),
|
|
('Linux', False, 'git-linux'),
|
|
('FreeBSD', False, 'other'),
|
|
],
|
|
)
|
|
def test_form_factor(cnr_utils, monkeypatch, system, desktop, expected):
|
|
monkeypatch.setattr(cnr_utils.platform, 'system', lambda: system)
|
|
if desktop:
|
|
monkeypatch.setenv('__COMFYUI_DESKTOP_VERSION__', '1.0.0')
|
|
else:
|
|
monkeypatch.delenv('__COMFYUI_DESKTOP_VERSION__', raising=False)
|
|
|
|
assert cnr_utils.get_form_factor() == expected
|