| import argparse
|
| import pytest
|
| from requests.exceptions import HTTPError
|
|
|
| from app.frontend_management import (
|
| FrontendManager,
|
| FrontEndProvider,
|
| Release,
|
| )
|
| from comfy.cli_args import DEFAULT_VERSION_STRING
|
|
|
|
|
| @pytest.fixture
|
| def mock_releases():
|
| return [
|
| Release(
|
| id=1,
|
| tag_name="1.0.0",
|
| name="Release 1.0.0",
|
| prerelease=False,
|
| created_at="2022-01-01T00:00:00Z",
|
| published_at="2022-01-01T00:00:00Z",
|
| body="Release notes for 1.0.0",
|
| assets=[{"name": "dist.zip", "url": "https://example.com/dist.zip"}],
|
| ),
|
| Release(
|
| id=2,
|
| tag_name="2.0.0",
|
| name="Release 2.0.0",
|
| prerelease=False,
|
| created_at="2022-02-01T00:00:00Z",
|
| published_at="2022-02-01T00:00:00Z",
|
| body="Release notes for 2.0.0",
|
| assets=[{"name": "dist.zip", "url": "https://example.com/dist.zip"}],
|
| ),
|
| ]
|
|
|
|
|
| @pytest.fixture
|
| def mock_provider(mock_releases):
|
| provider = FrontEndProvider(
|
| owner="test-owner",
|
| repo="test-repo",
|
| )
|
| provider.all_releases = mock_releases
|
| provider.latest_release = mock_releases[1]
|
| FrontendManager.PROVIDERS = [provider]
|
| return provider
|
|
|
|
|
| def test_get_release(mock_provider, mock_releases):
|
| version = "1.0.0"
|
| release = mock_provider.get_release(version)
|
| assert release == mock_releases[0]
|
|
|
|
|
| def test_get_release_latest(mock_provider, mock_releases):
|
| version = "latest"
|
| release = mock_provider.get_release(version)
|
| assert release == mock_releases[1]
|
|
|
|
|
| def test_get_release_invalid_version(mock_provider):
|
| version = "invalid"
|
| with pytest.raises(ValueError):
|
| mock_provider.get_release(version)
|
|
|
|
|
| def test_init_frontend_default():
|
| version_string = DEFAULT_VERSION_STRING
|
| frontend_path = FrontendManager.init_frontend(version_string)
|
| assert frontend_path == FrontendManager.DEFAULT_FRONTEND_PATH
|
|
|
|
|
| def test_init_frontend_invalid_version():
|
| version_string = "test-owner/test-repo@1.100.99"
|
| with pytest.raises(HTTPError):
|
| FrontendManager.init_frontend_unsafe(version_string)
|
|
|
|
|
| def test_init_frontend_invalid_provider():
|
| version_string = "invalid/invalid@latest"
|
| with pytest.raises(HTTPError):
|
| FrontendManager.init_frontend_unsafe(version_string)
|
|
|
|
|
| def test_parse_version_string():
|
| version_string = "owner/repo@1.0.0"
|
| repo_owner, repo_name, version = FrontendManager.parse_version_string(
|
| version_string
|
| )
|
| assert repo_owner == "owner"
|
| assert repo_name == "repo"
|
| assert version == "1.0.0"
|
|
|
|
|
| def test_parse_version_string_invalid():
|
| version_string = "invalid"
|
| with pytest.raises(argparse.ArgumentTypeError):
|
| FrontendManager.parse_version_string(version_string)
|
|
|
