prole/tests/installer/test_stream_exec.py
chrisfu 5618b662dd Remove prole-db-manager; simplify deployment via prole-authority; fix pg18 downgrade & cluster name
Summary:
Removed the prole-db-manager microservice and simplified deployment to use
prole-authority as the internal management and authorization point. Fixed two
blocking bugs that prevented silent install from completing on knoe-dev-cluster.

Removed: prole-db-manager
- Deleted db-manager-deployment.yaml and db-manager-service.yaml from opentofu manifests
- Deleted src/db-manager/ (Dockerfile, server.js, package.json, tests)
- Removed prole-db-manager port-forward mapping from installer/core/env.py
- Removed init_db_manager.sh from Initialization Scripts (milestones.py, actions.py)
- Removed init_certmgr.sh and init_db_manager.sh tabs from services screen (services.py)
- Removed live k8s Deployment/Service from knoe-dev-cluster

Fixed: PostgreSQL version downgrade error (pg17 -> pg18)
- Created conf/postgresql/.version with value 18
- Updated k8s/prole/prole-db.yaml and prole-db-recovery.yaml.tpl imageName to prole-db:18-089
- Fixed _init_database_options_state() to restore saved version_type from prole.cfg
  so db_version_type defaults to v18 (pg18) instead of silently reverting to pg17
- Added database_options.* keys to _collect_input_snapshot() in cfg.py so
  distribution, version_type, and all extension toggles persist to prole.cfg

Fixed: Cluster name inconsistency
- Removed stale prole-dev-cluster references; all scripts now use knoe-dev-cluster
- Added knoe-dev-cluster to mode-detection case in etc/prole_cfg.sh

Config: conf/prole.cfg
- Set kerberos_config.enabled = False, KERBEROS_AUTO_ENABLED = False
- Added database_options.distribution = percona, version_type = v18
- Added all 13 extension flags set to True (postgis, pgvector, pgcrypto, pgaudit,
  pg_repack, pg_stat_statements, pg_buffercache, pg_freespacemap, pgrowlocks,
  postgres_fdw, dblink, pg_stat_monitor, pgbadger)

Verification:
./install.py -s -l -v -c conf/prole.cfg completed successfully.
CNPG deployed prole-db:18-089 to knoe-dev-cluster; all milestones passed.

Co-authored-by: Junie <junie@jetbrains.com>
2026-03-01 20:40:44 -08:00

95 lines
2.7 KiB
Python

import sys
import time
from installer.core.stream_exec import run_streaming_cmd
def test_run_streaming_cmd_streams_stdout_incrementally():
chunks: list[str] = []
first_chunk_at: list[float] = []
def _on_stdout(text: str):
chunks.append(text)
if not first_chunk_at:
first_chunk_at.append(time.time())
started_at = time.time()
rc = run_streaming_cmd(
[
sys.executable,
"-c",
"import sys,time;"
"sys.stdout.write('first\\n');sys.stdout.flush();"
"time.sleep(0.35);"
"sys.stdout.write('second\\n');sys.stdout.flush()",
],
on_stdout=_on_stdout,
)
finished_at = time.time()
assert rc == 0
assert first_chunk_at
# The first callback should fire while the process is still running.
assert first_chunk_at[0] < finished_at - 0.15
output = "".join(chunks)
assert "first" in output
assert "second" in output
assert started_at < first_chunk_at[0]
def test_run_streaming_cmd_nonzero_exit():
rc = run_streaming_cmd([sys.executable, "-c", "import sys; sys.exit(42)"])
assert rc == 42
def test_run_streaming_cmd_string_form():
"""A plain string command is wrapped in bash -lc."""
chunks: list[str] = []
rc = run_streaming_cmd(
f"{sys.executable} -c \"print('strtest')\"",
on_stdout=lambda t: chunks.append(t),
)
assert rc == 0
assert "strtest" in "".join(chunks)
def test_run_streaming_cmd_stdin_text():
"""stdin_text is written to the subprocess stdin."""
chunks: list[str] = []
rc = run_streaming_cmd(
[sys.executable, "-c", "import sys; print(sys.stdin.read().strip())"],
stdin_text="hello_from_stdin",
on_stdout=lambda t: chunks.append(t),
)
assert rc == 0
assert "hello_from_stdin" in "".join(chunks)
def test_run_streaming_cmd_no_callbacks():
"""Running without callbacks should not crash."""
rc = run_streaming_cmd([sys.executable, "-c", "print('silent')"])
assert rc == 0
def test_run_streaming_cmd_keeps_stderr_and_ansi_sequences():
stdout_chunks: list[str] = []
stderr_chunks: list[str] = []
rc = run_streaming_cmd(
[
sys.executable,
"-c",
"import sys;"
"sys.stdout.write('OUT\\n');sys.stdout.flush();"
"sys.stderr.write('\\x1b[31mERR\\x1b[0m\\n');sys.stderr.flush()",
],
on_stdout=lambda text: stdout_chunks.append(text),
on_stderr=lambda text: stderr_chunks.append(text),
)
assert rc == 0
assert "OUT" in "".join(stdout_chunks)
stderr_out = "".join(stderr_chunks)
assert "ERR" in stderr_out
assert "\x1b[31m" in stderr_out