mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-11-13 04:50:21 +00:00
91 lines
3.8 KiB
Python
91 lines
3.8 KiB
Python
"""Test add custom repository."""
|
|
import json
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from supervisor.addons.addon import Addon
|
|
from supervisor.exceptions import StoreError
|
|
from supervisor.resolution.const import SuggestionType
|
|
from supervisor.store import BUILTIN_REPOSITORIES
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_add_valid_repository(coresys, store_manager):
|
|
"""Test add custom repository."""
|
|
current = coresys.config.addons_repositories
|
|
with patch("supervisor.store.repository.Repository.load", return_value=None), patch(
|
|
"supervisor.utils.common.read_yaml_file",
|
|
return_value={"name": "Awesome repository"},
|
|
), patch("pathlib.Path.exists", return_value=True):
|
|
|
|
await store_manager.update_repositories(current + ["http://example.com"])
|
|
assert store_manager.get_from_url("http://example.com").validate()
|
|
assert "http://example.com" in coresys.config.addons_repositories
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_add_valid_repository_url(coresys, store_manager):
|
|
"""Test add custom repository."""
|
|
current = coresys.config.addons_repositories
|
|
with patch("supervisor.store.repository.Repository.load", return_value=None), patch(
|
|
"supervisor.utils.common.read_yaml_file",
|
|
return_value={"name": "Awesome repository"},
|
|
), patch("pathlib.Path.exists", return_value=True):
|
|
await store_manager.update_repositories(current + ["http://example.com"])
|
|
assert store_manager.get_from_url("http://example.com").validate()
|
|
assert "http://example.com" in coresys.config.addons_repositories
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_add_invalid_repository(coresys, store_manager):
|
|
"""Test add custom repository."""
|
|
current = coresys.config.addons_repositories
|
|
with patch("supervisor.store.repository.Repository.load", return_value=None), patch(
|
|
"pathlib.Path.read_text",
|
|
return_value="",
|
|
):
|
|
await store_manager.update_repositories(current + ["http://example.com"])
|
|
assert not store_manager.get_from_url("http://example.com").validate()
|
|
|
|
assert "http://example.com" in coresys.config.addons_repositories
|
|
assert coresys.resolution.suggestions[-1].type == SuggestionType.EXECUTE_REMOVE
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_add_invalid_repository_file(coresys, store_manager):
|
|
"""Test add custom repository."""
|
|
current = coresys.config.addons_repositories
|
|
with patch("supervisor.store.repository.Repository.load", return_value=None), patch(
|
|
"pathlib.Path.read_text",
|
|
return_value=json.dumps({"name": "Awesome repository"}),
|
|
), patch("pathlib.Path.exists", return_value=False):
|
|
await store_manager.update_repositories(current + ["http://example.com"])
|
|
assert not store_manager.get_from_url("http://example.com").validate()
|
|
|
|
assert "http://example.com" in coresys.config.addons_repositories
|
|
assert coresys.resolution.suggestions[-1].type == SuggestionType.EXECUTE_REMOVE
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_preinstall_valid_repository(coresys, store_manager):
|
|
"""Test add core repository valid."""
|
|
with patch("supervisor.store.repository.Repository.load", return_value=None):
|
|
await store_manager.update_repositories(BUILTIN_REPOSITORIES)
|
|
assert store_manager.get("core").validate()
|
|
assert store_manager.get("local").validate()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_remove_used_repository(coresys, store_manager, store_addon):
|
|
"""Test removing used custom repository."""
|
|
coresys.addons.data.install(store_addon)
|
|
addon = Addon(coresys, store_addon.slug)
|
|
coresys.addons.local[addon.slug] = addon
|
|
|
|
with pytest.raises(
|
|
StoreError,
|
|
match="Can't remove 'https://github.com/awesome-developer/awesome-repo'. It's used by installed add-ons",
|
|
):
|
|
await store_manager.update_repositories([])
|