unkinben
1656664dfa
ci/woodpecker/tag/docker Pipeline was successful
Repository types now live under dedicated top-level keys instead of a shared remotes: block distinguished by a type field: remotes: caching proxy remotes (no type field needed) virtuals: virtual merged-index repositories locals: local upload repositories Routes for local repos move from /api/v1/remote/ to /api/v1/local/. config.py gains get_virtual_config() and get_local_config() lookups. Root endpoint now reports all three sections. Drop root conf.d/ (was an exact duplicate of examples/conf.d-method/). Reviewed-on: #31
597 lines
24 KiB
Python
597 lines
24 KiB
Python
"""Unit tests for the virtual repository handler (artifact/virtual.py)."""
|
|
|
|
from datetime import UTC, date, datetime
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import pytest
|
|
import yaml
|
|
|
|
from artifactapi.artifact.virtual import (
|
|
_HANDLERS,
|
|
_get_member_index,
|
|
_HelmDumper,
|
|
_HelmHandler,
|
|
_merge_helm_indexes,
|
|
_VirtualHandler,
|
|
)
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Shared sample data
|
|
# ---------------------------------------------------------------------------
|
|
|
|
_INDEX_A = b"""\
|
|
apiVersion: v1
|
|
entries:
|
|
vault:
|
|
- name: vault
|
|
version: "0.27.0"
|
|
urls:
|
|
- https://helm.releases.hashicorp.com/vault-0.27.0.tgz
|
|
consul:
|
|
- name: consul
|
|
version: "1.2.0"
|
|
urls:
|
|
- https://helm.releases.hashicorp.com/consul-1.2.0.tgz
|
|
generated: "2023-01-01T00:00:00.000Z"
|
|
"""
|
|
|
|
_INDEX_B = b"""\
|
|
apiVersion: v1
|
|
entries:
|
|
nginx:
|
|
- name: nginx
|
|
version: "15.0.0"
|
|
urls:
|
|
- https://charts.example.com/nginx-15.0.0.tgz
|
|
vault:
|
|
- name: vault
|
|
version: "0.27.0"
|
|
urls:
|
|
- https://charts.example.com/vault-0.27.0.tgz
|
|
- name: vault
|
|
version: "0.26.0"
|
|
urls:
|
|
- https://charts.example.com/vault-0.26.0.tgz
|
|
generated: "2023-01-01T00:00:00.000Z"
|
|
"""
|
|
|
|
_INDEX_SIMPLE = b"""\
|
|
apiVersion: v1
|
|
entries:
|
|
mychart:
|
|
- name: mychart
|
|
version: "1.0.0"
|
|
urls:
|
|
- https://helm.releases.hashicorp.com/mychart-1.0.0.tgz
|
|
generated: "2023-01-01T00:00:00.000Z"
|
|
"""
|
|
|
|
_CFG_A = {"base_url": "https://helm.releases.hashicorp.com", "cache": {"mutable_ttl": 3600}}
|
|
_CFG_B = {"base_url": "https://charts.example.com", "cache": {"mutable_ttl": 1800}}
|
|
|
|
|
|
def _identity_resolve(data, *args, **kwargs):
|
|
return data, None
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# _HelmDumper — datetime/date YAML serialization
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TestHelmDumper:
|
|
def _dump(self, value):
|
|
return yaml.dump({"v": value}, Dumper=_HelmDumper)
|
|
|
|
def test_datetime_with_tz_includes_Z_suffix(self):
|
|
dt = datetime(2023, 6, 15, 12, 0, 0, tzinfo=UTC)
|
|
assert "Z" in self._dump(dt)
|
|
|
|
def test_datetime_without_tz_has_no_Z_suffix(self):
|
|
dt = datetime(2023, 6, 15, 12, 0, 0)
|
|
assert "Z" not in self._dump(dt)
|
|
|
|
def test_datetime_uses_T_separator_not_space(self):
|
|
dt = datetime(2023, 6, 15, 12, 30, 0, tzinfo=UTC)
|
|
assert "T12:30:00" in self._dump(dt)
|
|
|
|
def test_date_serialized_as_iso_string(self):
|
|
assert "2023-01-15" in self._dump(date(2023, 1, 15))
|
|
|
|
def test_datetime_round_trips_as_string_not_python_datetime(self):
|
|
dt = datetime(2023, 6, 15, 12, 0, 0, tzinfo=UTC)
|
|
parsed = yaml.safe_load(self._dump(dt))
|
|
# yaml.safe_load must not re-parse this as a datetime object
|
|
assert isinstance(parsed["v"], str)
|
|
|
|
def test_date_round_trips_as_string_not_python_date(self):
|
|
parsed = yaml.safe_load(self._dump(date(2023, 1, 15)))
|
|
assert isinstance(parsed["v"], str)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# _HelmHandler
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TestHelmHandler:
|
|
def setup_method(self):
|
|
self.handler = _HelmHandler()
|
|
|
|
def test_accepts_index_yaml(self):
|
|
assert self.handler.accepts_path("index.yaml") is True
|
|
|
|
def test_rejects_tgz_path(self):
|
|
assert self.handler.accepts_path("vault-0.27.0.tgz") is False
|
|
|
|
def test_rejects_subdirectory_index(self):
|
|
assert self.handler.accepts_path("charts/index.yaml") is False
|
|
|
|
def test_rejects_empty_path(self):
|
|
assert self.handler.accepts_path("") is False
|
|
|
|
def test_path_error_is_non_empty_string(self):
|
|
msg = self.handler.path_error()
|
|
assert isinstance(msg, str) and len(msg) > 0
|
|
|
|
def test_merge_returns_bytes(self):
|
|
with patch("artifactapi.artifact.virtual._helm.resolve_content", side_effect=_identity_resolve):
|
|
result = self.handler.merge([_INDEX_A], ["member-a"], [_CFG_A], "http://proxy.example.com")
|
|
assert isinstance(result, bytes)
|
|
|
|
def test_merge_delegates_to_merge_helm_indexes(self):
|
|
with patch("artifactapi.artifact.virtual._merge_helm_indexes", return_value=b"merged") as mock_fn:
|
|
result = self.handler.merge([b"data"], ["m"], [{}], "http://proxy")
|
|
mock_fn.assert_called_once_with([b"data"], ["m"], [{}], "http://proxy")
|
|
assert result == b"merged"
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# _HANDLERS registry
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TestHandlersRegistry:
|
|
def test_helm_handler_is_registered(self):
|
|
assert "helm" in _HANDLERS
|
|
assert isinstance(_HANDLERS["helm"], _HelmHandler)
|
|
|
|
def test_helm_handler_satisfies_protocol(self):
|
|
assert isinstance(_HANDLERS["helm"], _VirtualHandler)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# _merge_helm_indexes
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TestMergeHelmIndexes:
|
|
def _merge(self, raw_indexes, member_names, member_configs, proxy_base="http://proxy.example.com"):
|
|
with patch("artifactapi.artifact.virtual._helm.resolve_content", side_effect=_identity_resolve):
|
|
return _merge_helm_indexes(raw_indexes, member_names, member_configs, proxy_base)
|
|
|
|
def _parse(self, raw):
|
|
return yaml.safe_load(raw)
|
|
|
|
def test_single_member_all_charts_present(self):
|
|
index = self._parse(self._merge([_INDEX_A], ["member-a"], [_CFG_A]))
|
|
assert "vault" in index["entries"]
|
|
assert "consul" in index["entries"]
|
|
|
|
def test_two_members_non_overlapping_charts_all_present(self):
|
|
index = self._parse(self._merge([_INDEX_A, _INDEX_B], ["member-a", "member-b"], [_CFG_A, _CFG_B]))
|
|
assert "vault" in index["entries"]
|
|
assert "consul" in index["entries"]
|
|
assert "nginx" in index["entries"]
|
|
|
|
def test_first_member_wins_on_duplicate_name_and_version(self):
|
|
index = self._parse(self._merge([_INDEX_A, _INDEX_B], ["member-a", "member-b"], [_CFG_A, _CFG_B]))
|
|
v027 = next(e for e in index["entries"]["vault"] if e["version"] == "0.27.0")
|
|
assert "helm.releases.hashicorp.com" in v027["urls"][0]
|
|
|
|
def test_different_versions_of_same_chart_both_included(self):
|
|
index = self._parse(self._merge([_INDEX_A, _INDEX_B], ["member-a", "member-b"], [_CFG_A, _CFG_B]))
|
|
versions = {e["version"] for e in index["entries"]["vault"]}
|
|
assert "0.27.0" in versions
|
|
assert "0.26.0" in versions
|
|
|
|
def test_malformed_yaml_from_member_is_skipped(self):
|
|
index = self._parse(self._merge([_INDEX_A, b"{bad yaml"], ["member-a", "bad"], [_CFG_A, _CFG_B]))
|
|
assert "vault" in index["entries"]
|
|
assert "consul" in index["entries"]
|
|
|
|
def test_output_has_apiVersion_v1(self):
|
|
index = self._parse(self._merge([_INDEX_A], ["member-a"], [_CFG_A]))
|
|
assert index["apiVersion"] == "v1"
|
|
|
|
def test_output_has_generated_field(self):
|
|
index = self._parse(self._merge([_INDEX_A], ["member-a"], [_CFG_A]))
|
|
assert "generated" in index
|
|
|
|
def test_output_is_valid_yaml(self):
|
|
raw = self._merge([_INDEX_A, _INDEX_B], ["member-a", "member-b"], [_CFG_A, _CFG_B])
|
|
assert isinstance(yaml.safe_load(raw), dict)
|
|
|
|
def test_empty_index_from_member_produces_no_entries(self):
|
|
empty = b"apiVersion: v1\nentries: {}\ngenerated: '2023-01-01T00:00:00.000Z'\n"
|
|
index = self._parse(self._merge([empty], ["member-a"], [_CFG_A]))
|
|
assert index["entries"] == {}
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# _get_member_index (async)
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
class TestGetMemberIndex:
|
|
@pytest.fixture
|
|
def storage(self):
|
|
m = MagicMock()
|
|
m.get_object_key.return_value = "member/key/index.yaml"
|
|
m.exists.return_value = False
|
|
m.download_object.return_value = b"cached bytes"
|
|
return m
|
|
|
|
@pytest.fixture
|
|
def cache(self):
|
|
m = MagicMock()
|
|
m.is_index_valid.return_value = False
|
|
return m
|
|
|
|
@pytest.fixture
|
|
def member_cfg(self):
|
|
return {"base_url": "https://helm.releases.hashicorp.com", "cache": {"mutable_ttl": 3600}}
|
|
|
|
def _fake_response(self, content=b"upstream bytes"):
|
|
r = MagicMock()
|
|
r.content = content
|
|
r.raise_for_status = MagicMock()
|
|
return r
|
|
|
|
def _patch_httpx(self, response):
|
|
mock_client_cls = patch("artifactapi.artifact.virtual.httpx.AsyncClient")
|
|
p = mock_client_cls.start()
|
|
mock_client = AsyncMock()
|
|
p.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.return_value = response
|
|
return mock_client_cls, mock_client
|
|
|
|
async def test_cache_hit_returns_stored_bytes(self, storage, cache, member_cfg):
|
|
storage.exists.return_value = True
|
|
cache.is_index_valid.return_value = True
|
|
|
|
_, _, _, raw_data = await _get_member_index("m", member_cfg, "index.yaml", storage, cache)
|
|
|
|
assert raw_data == b"cached bytes"
|
|
|
|
async def test_cache_hit_does_not_fetch_upstream(self, storage, cache, member_cfg):
|
|
storage.exists.return_value = True
|
|
cache.is_index_valid.return_value = True
|
|
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
await _get_member_index("m", member_cfg, "index.yaml", storage, cache)
|
|
|
|
mock_cls.assert_not_called()
|
|
|
|
async def test_cache_hit_storage_error_falls_through_to_upstream(self, storage, cache, member_cfg):
|
|
storage.exists.return_value = True
|
|
cache.is_index_valid.return_value = True
|
|
storage.download_object.side_effect = Exception("S3 read error")
|
|
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
mock_client = AsyncMock()
|
|
mock_cls.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.return_value = self._fake_response(b"fresh bytes")
|
|
|
|
_, _, _, raw_data = await _get_member_index("m", member_cfg, "index.yaml", storage, cache)
|
|
|
|
assert raw_data == b"fresh bytes"
|
|
|
|
async def test_cache_miss_fetches_from_upstream(self, storage, cache, member_cfg):
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
mock_client = AsyncMock()
|
|
mock_cls.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.return_value = self._fake_response()
|
|
|
|
_, _, _, raw_data = await _get_member_index("m", member_cfg, "index.yaml", storage, cache)
|
|
|
|
assert raw_data == b"upstream bytes"
|
|
|
|
async def test_cache_miss_stores_result_in_s3(self, storage, cache, member_cfg):
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
mock_client = AsyncMock()
|
|
mock_cls.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.return_value = self._fake_response()
|
|
|
|
await _get_member_index("m", member_cfg, "index.yaml", storage, cache)
|
|
|
|
storage.upload.assert_called_once()
|
|
|
|
async def test_cache_miss_marks_cache_with_configured_ttl(self, storage, cache, member_cfg):
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
mock_client = AsyncMock()
|
|
mock_cls.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.return_value = self._fake_response()
|
|
|
|
await _get_member_index("m", member_cfg, "index.yaml", storage, cache)
|
|
|
|
cache.mark_index_cached.assert_called_once_with("m", "index.yaml", 3600)
|
|
|
|
async def test_cache_miss_with_auth_sends_basic_auth_header(self, storage, cache):
|
|
cfg = {
|
|
"base_url": "https://private.example.com",
|
|
"username": "user",
|
|
"password": "pass",
|
|
"cache": {"mutable_ttl": 3600},
|
|
}
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
mock_client = AsyncMock()
|
|
mock_cls.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.return_value = self._fake_response()
|
|
|
|
await _get_member_index("m", cfg, "index.yaml", storage, cache)
|
|
|
|
headers = mock_client.get.call_args.kwargs["headers"]
|
|
assert "Authorization" in headers
|
|
assert headers["Authorization"].startswith("Basic ")
|
|
|
|
async def test_no_credentials_sends_no_auth_header(self, storage, cache, member_cfg):
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
mock_client = AsyncMock()
|
|
mock_cls.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.return_value = self._fake_response()
|
|
|
|
await _get_member_index("m", member_cfg, "index.yaml", storage, cache)
|
|
|
|
headers = mock_client.get.call_args.kwargs["headers"]
|
|
assert "Authorization" not in headers
|
|
|
|
async def test_upstream_fetch_failure_returns_none(self, storage, cache, member_cfg):
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
mock_client = AsyncMock()
|
|
mock_cls.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.side_effect = Exception("connection refused")
|
|
|
|
_, _, _, raw_data = await _get_member_index("m", member_cfg, "index.yaml", storage, cache)
|
|
|
|
assert raw_data is None
|
|
|
|
async def test_s3_upload_failure_still_returns_data(self, storage, cache, member_cfg):
|
|
storage.upload.side_effect = Exception("S3 write error")
|
|
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
mock_client = AsyncMock()
|
|
mock_cls.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.return_value = self._fake_response()
|
|
|
|
_, _, _, raw_data = await _get_member_index("m", member_cfg, "index.yaml", storage, cache)
|
|
|
|
assert raw_data == b"upstream bytes"
|
|
|
|
async def test_returns_ttl_from_config(self, storage, cache):
|
|
cfg = {"base_url": "https://example.com", "cache": {"mutable_ttl": 900}}
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
mock_client = AsyncMock()
|
|
mock_cls.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.return_value = self._fake_response()
|
|
|
|
_, _, ttl, _ = await _get_member_index("m", cfg, "index.yaml", storage, cache)
|
|
|
|
assert ttl == 900
|
|
|
|
async def test_defaults_ttl_to_3600_when_not_configured(self, storage, cache):
|
|
cfg = {"base_url": "https://example.com"}
|
|
with patch("artifactapi.artifact.virtual.httpx.AsyncClient") as mock_cls:
|
|
mock_client = AsyncMock()
|
|
mock_cls.return_value.__aenter__.return_value = mock_client
|
|
mock_client.get.return_value = self._fake_response()
|
|
|
|
_, _, ttl, _ = await _get_member_index("m", cfg, "index.yaml", storage, cache)
|
|
|
|
assert ttl == 3600
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Virtual route GET /api/v1/virtual/{name}/{path}
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_storage_v():
|
|
m = MagicMock()
|
|
m.get_object_key.return_value = "virtual/helm-virtual-test/index.yaml"
|
|
m.exists.return_value = False
|
|
m.download_object.return_value = b"apiVersion: v1\nentries: {}\n"
|
|
return m
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_cache_v():
|
|
m = MagicMock()
|
|
m.is_index_valid.return_value = False
|
|
m.available = False
|
|
m.client = None
|
|
return m
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_virtual_deps(mock_storage_v, mock_cache_v):
|
|
import artifactapi.main as main_mod
|
|
|
|
with (
|
|
patch.object(main_mod, "storage", mock_storage_v),
|
|
patch.object(main_mod, "cache", mock_cache_v),
|
|
):
|
|
yield {"storage": mock_storage_v, "cache": mock_cache_v}
|
|
|
|
|
|
class TestVirtualRoute:
|
|
def test_unknown_virtual_name_returns_404(self, client, patched_virtual_deps):
|
|
response = client.get("/api/v1/virtual/no-such-virtual/index.yaml")
|
|
assert response.status_code == 404
|
|
|
|
def test_non_virtual_name_returns_404(self, client, patched_virtual_deps):
|
|
# helm-test is in remotes, not virtuals
|
|
response = client.get("/api/v1/virtual/helm-test/index.yaml")
|
|
assert response.status_code == 404
|
|
|
|
def test_unsupported_package_returns_400(self, client, patched_virtual_deps):
|
|
# unsupported-virtual-test has package "rpm"
|
|
response = client.get("/api/v1/virtual/unsupported-virtual-test/index.yaml")
|
|
assert response.status_code == 400
|
|
|
|
def test_non_index_path_returns_404(self, client, patched_virtual_deps):
|
|
response = client.get("/api/v1/virtual/helm-virtual-test/vault-0.27.0.tgz")
|
|
assert response.status_code == 404
|
|
|
|
def test_no_members_returns_500(self, client, patched_virtual_deps):
|
|
response = client.get("/api/v1/virtual/empty-virtual-test/index.yaml")
|
|
assert response.status_code == 500
|
|
|
|
def test_virtual_cache_hit_returns_200(self, client, patched_virtual_deps):
|
|
deps = patched_virtual_deps
|
|
deps["storage"].exists.return_value = True
|
|
deps["cache"].is_index_valid.return_value = True
|
|
|
|
response = client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
assert response.status_code == 200
|
|
|
|
def test_virtual_cache_hit_content_type_is_yaml(self, client, patched_virtual_deps):
|
|
deps = patched_virtual_deps
|
|
deps["storage"].exists.return_value = True
|
|
deps["cache"].is_index_valid.return_value = True
|
|
|
|
response = client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
assert "text/yaml" in response.headers["content-type"]
|
|
|
|
def test_virtual_cache_hit_returns_stored_content(self, client, patched_virtual_deps):
|
|
deps = patched_virtual_deps
|
|
deps["storage"].exists.return_value = True
|
|
deps["cache"].is_index_valid.return_value = True
|
|
deps["storage"].download_object.return_value = b"apiVersion: v1\nentries: {}\n"
|
|
|
|
response = client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
assert response.content == b"apiVersion: v1\nentries: {}\n"
|
|
|
|
def test_virtual_cache_hit_skips_member_fetch(self, client, patched_virtual_deps):
|
|
deps = patched_virtual_deps
|
|
deps["storage"].exists.return_value = True
|
|
deps["cache"].is_index_valid.return_value = True
|
|
|
|
with patch("artifactapi.artifact.virtual._get_member_index", new_callable=AsyncMock) as mock_get:
|
|
client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
|
|
mock_get.assert_not_called()
|
|
|
|
def test_cache_miss_returns_200_with_yaml_content_type(self, client, patched_virtual_deps):
|
|
with (
|
|
patch("artifactapi.artifact.virtual._get_member_index", new_callable=AsyncMock) as mock_get,
|
|
patch("artifactapi.artifact.virtual._helm.resolve_content", side_effect=_identity_resolve),
|
|
):
|
|
mock_get.return_value = ("helm-test", _CFG_A, 3600, _INDEX_SIMPLE)
|
|
response = client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
|
|
assert response.status_code == 200
|
|
assert "text/yaml" in response.headers["content-type"]
|
|
|
|
def test_cache_miss_response_contains_merged_entries(self, client, patched_virtual_deps):
|
|
with (
|
|
patch("artifactapi.artifact.virtual._get_member_index", new_callable=AsyncMock) as mock_get,
|
|
patch("artifactapi.artifact.virtual._helm.resolve_content", side_effect=_identity_resolve),
|
|
):
|
|
mock_get.return_value = ("helm-test", _CFG_A, 3600, _INDEX_SIMPLE)
|
|
response = client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
|
|
index = yaml.safe_load(response.content)
|
|
assert "mychart" in index["entries"]
|
|
|
|
def test_cache_miss_stores_result_in_s3(self, client, patched_virtual_deps):
|
|
deps = patched_virtual_deps
|
|
with (
|
|
patch("artifactapi.artifact.virtual._get_member_index", new_callable=AsyncMock) as mock_get,
|
|
patch("artifactapi.artifact.virtual._helm.resolve_content", side_effect=_identity_resolve),
|
|
):
|
|
mock_get.return_value = ("helm-test", _CFG_A, 3600, _INDEX_SIMPLE)
|
|
client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
|
|
deps["storage"].upload.assert_called_once()
|
|
|
|
def test_cache_miss_marks_index_cached(self, client, patched_virtual_deps):
|
|
deps = patched_virtual_deps
|
|
with (
|
|
patch("artifactapi.artifact.virtual._get_member_index", new_callable=AsyncMock) as mock_get,
|
|
patch("artifactapi.artifact.virtual._helm.resolve_content", side_effect=_identity_resolve),
|
|
):
|
|
mock_get.return_value = ("helm-test", _CFG_A, 3600, _INDEX_SIMPLE)
|
|
client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
|
|
deps["cache"].mark_index_cached.assert_called_once()
|
|
|
|
def test_cache_miss_uses_min_ttl_across_members(self, client, patched_virtual_deps):
|
|
deps = patched_virtual_deps
|
|
with (
|
|
patch("artifactapi.artifact.virtual._get_member_index", new_callable=AsyncMock) as mock_get,
|
|
patch("artifactapi.artifact.virtual._helm.resolve_content", side_effect=_identity_resolve),
|
|
):
|
|
mock_get.side_effect = [
|
|
("helm-test", _CFG_A, 3600, _INDEX_SIMPLE),
|
|
("helm-member-2", _CFG_B, 1800, _INDEX_SIMPLE),
|
|
]
|
|
client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
|
|
_, _, ttl = deps["cache"].mark_index_cached.call_args[0]
|
|
assert ttl == 1800
|
|
|
|
def test_all_members_unreachable_returns_502(self, client, patched_virtual_deps):
|
|
with patch("artifactapi.artifact.virtual._get_member_index", new_callable=AsyncMock) as mock_get:
|
|
mock_get.return_value = ("helm-test", _CFG_A, 3600, None)
|
|
response = client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
|
|
assert response.status_code == 502
|
|
|
|
def test_one_member_unreachable_still_returns_200(self, client, patched_virtual_deps):
|
|
with (
|
|
patch("artifactapi.artifact.virtual._get_member_index", new_callable=AsyncMock) as mock_get,
|
|
patch("artifactapi.artifact.virtual._helm.resolve_content", side_effect=_identity_resolve),
|
|
):
|
|
mock_get.side_effect = [
|
|
("helm-test", _CFG_A, 3600, _INDEX_SIMPLE),
|
|
("helm-member-2", _CFG_B, 1800, None),
|
|
]
|
|
response = client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
|
|
assert response.status_code == 200
|
|
|
|
def test_member_not_in_config_is_skipped(self, client, patched_virtual_deps):
|
|
import artifactapi.main as main_mod
|
|
|
|
real_get = main_mod.config.get_remote_config
|
|
|
|
def patched_get(name):
|
|
return None if name == "helm-member-2" else real_get(name)
|
|
|
|
with (
|
|
patch("artifactapi.artifact.virtual._get_member_index", new_callable=AsyncMock) as mock_get,
|
|
patch("artifactapi.artifact.virtual._helm.resolve_content", side_effect=_identity_resolve),
|
|
patch.object(main_mod.config, "get_remote_config", side_effect=patched_get),
|
|
):
|
|
mock_get.return_value = ("helm-test", _CFG_A, 3600, _INDEX_SIMPLE)
|
|
response = client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
|
|
# only helm-test was available — should succeed
|
|
assert response.status_code == 200
|
|
mock_get.assert_called_once()
|
|
|
|
def test_s3_store_failure_still_returns_200(self, client, patched_virtual_deps):
|
|
deps = patched_virtual_deps
|
|
deps["storage"].upload.side_effect = Exception("S3 write error")
|
|
|
|
with (
|
|
patch("artifactapi.artifact.virtual._get_member_index", new_callable=AsyncMock) as mock_get,
|
|
patch("artifactapi.artifact.virtual._helm.resolve_content", side_effect=_identity_resolve),
|
|
):
|
|
mock_get.return_value = ("helm-test", _CFG_A, 3600, _INDEX_SIMPLE)
|
|
response = client.get("/api/v1/virtual/helm-virtual-test/index.yaml")
|
|
|
|
assert response.status_code == 200
|