prole/tests/installer/test_k3s_registry_hostport_resolution.py
chrisfu 4ee2b259c9 Checkpoint: rename installer to knoe + harden db build context
- Add build-context helper to copy Docker context safely (ignore runtime data, keep symlinks)

- Update UI and core actions to use ~/.prole/build and shared copy helper

- Add/adjust tests and scripts; introduce knoe ops helpers and update manifests

Co-authored-by: Junie <junie@jetbrains.com>
2026-03-22 01:45:21 -07:00

75 lines
2.0 KiB
Python

from __future__ import annotations
from pathlib import Path
import pytest
import knoe.ui.screens.docker as docker_mod
from knoe.ui.screens.docker import DockerScreenMixin
class _DummyDockerScreen(DockerScreenMixin):
def __init__(self, prole_cfg_data: dict):
self.prole_cfg_data = prole_cfg_data
@pytest.mark.parametrize(
"server_url",
[
"https://0.0.0.0:6443",
"https://localhost:6443",
"https://127.0.0.1:6443",
],
)
def test_k3s_registry_hostport_avoids_unusable_server_hosts_and_falls_back_to_kubeconfig(
monkeypatch, tmp_path: Path, server_url: str
):
# Regression: k3s server URLs sometimes end up as bind addresses like 0.0.0.0.
# Those are not usable registry client endpoints; we should fall back to kubeconfig.
kubeconfig = tmp_path / "kubeconfig.yaml"
kubeconfig.write_text(
"""
apiVersion: v1
clusters:
- cluster:
server: https://myrddin.prole.org:6443
name: default
contexts: []
current-context: ""
kind: Config
preferences: {}
users: []
""".lstrip(),
encoding="utf-8",
)
monkeypatch.delenv("K3S_REGISTRY_HOST", raising=False)
monkeypatch.delenv("K3S_REGISTRY_PORT", raising=False)
monkeypatch.setattr(docker_mod, "_find_kubeconfig_file", lambda *_a, **_k: str(kubeconfig))
screen = _DummyDockerScreen(
{
"Global": {"DEPLOYMENT_MODE": "k3s", "PROLE_K3S_SERVER": server_url},
"Docker Build": {},
}
)
assert screen._k3s_registry_hostport() == "myrddin.prole.org:5000"
def test_k3s_registry_hostport_prefers_explicit_env_host_port(monkeypatch):
monkeypatch.setenv("K3S_REGISTRY_HOST", "registry.example.com")
monkeypatch.setenv("K3S_REGISTRY_PORT", "5443")
screen = _DummyDockerScreen(
{
"Global": {"DEPLOYMENT_MODE": "k3s", "PROLE_K3S_SERVER": "https://0.0.0.0:6443"},
"Docker Build": {},
}
)
assert screen._k3s_registry_hostport() == "registry.example.com:5443"