You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
101 lines
4.9 KiB
101 lines
4.9 KiB
import sys
|
|
import pytest
|
|
import testinfra
|
|
|
|
from pathlib import Path
|
|
THIS_COOKBOOK_DIR = Path(__file__).parent.parent
|
|
COOKBOOKS_DIR = THIS_COOKBOOK_DIR.parent
|
|
TOP_LEVEL_DIR = COOKBOOKS_DIR.parent
|
|
THIS_COOKBOOK_NAME = THIS_COOKBOOK_DIR.name
|
|
|
|
sys.path.insert(0, str(TOP_LEVEL_DIR / "tests"))
|
|
import test_quadlet # noqa: E402
|
|
|
|
"""
|
|
Verify that the postgresql Quadlet is correctly installed and configured on a fresh VM boot.
|
|
"""
|
|
class TestPostgresqlQuadlet(test_quadlet.TestQuadlet):
|
|
|
|
expected_services = [
|
|
{ "name": "postgresql-server.service", "state": "active", "exists": True },
|
|
{ "name": "postgresql-init.service", "state": "inactive", "exists": True },
|
|
{ "name": "postgresql-upgrade.service", "state": "inactive", "exists": True },
|
|
{ "name": "postgresql-backup.service", "state": "inactive", "exists": True },
|
|
{ "name": "postgresql-set-major.service", "state": "inactive", "exists": True },
|
|
{ "name": "postgresql-backup.timer", "state": "active", "exists": True },
|
|
{ "name": "postgresql.target", "state": "active", "exists": True },
|
|
]
|
|
|
|
expected_sockets = [
|
|
{ "uri": "tcp://127.0.0.1:5432", "state": "listening" },
|
|
{ "uri": "tcp://0.0.0.0:5432", "state": "closed" },
|
|
{ "uri": "tcp://:::5432", "state": "closed" },
|
|
]
|
|
|
|
expected_ports = [
|
|
{ "number": 5432, "protocol": "tcp", "state": "closed" },
|
|
{ "number": 22, "protocol": "tcp", "state": "open" },
|
|
]
|
|
|
|
expected_files = [
|
|
{ "path": "/var/lib/quadlets/postgresql", "type": "directory", "owner": "postgresql", "group": "itix-svc", "mode": 0o755 },
|
|
{ "path": "/etc/quadlets/postgresql/config.env", "type": "file", "owner": "root", "group": "root", "mode": 0o600 },
|
|
{ "path": "/var/lib/virtiofs/data/postgresql", "type": "directory", "owner": "postgresql", "group": "itix-svc", "mode": 0o700 },
|
|
{ "path": "/var/lib/virtiofs/data/postgresql/backup", "type": "directory", "owner": "postgresql", "group": "itix-svc", "mode": 0o700 },
|
|
{ "path": "/var/lib/quadlets/postgresql/.initialized", "type": "file", "owner": "root", "group": "root", "mode": 0o644 },
|
|
]
|
|
|
|
expected_podman_images = [
|
|
]
|
|
|
|
expected_podman_containers = [
|
|
{ "name": "postgresql-server", "state": "present", "pid1": { "owner": "10004", "group": "10000", "commandline": "postgres -h 127.0.0.1" } },
|
|
]
|
|
|
|
expected_main_service = "postgresql.target"
|
|
expected_main_service_timeout = 300
|
|
expected_pg_major = 0 # TODO: set this variable in subclasses
|
|
|
|
def _run_sql(self, fcos_host, query: str, check: bool = True, database: str = "postgres") -> str:
|
|
"""Execute *sql* via ``podman exec`` on the running postgresql-server container.
|
|
|
|
Uses the Unix socket at /var/run/postgresql inside the container (mapped
|
|
from /run/quadlets/postgresql on the host). The pg_hba.conf generated by
|
|
the official postgres image grants trust access on local sockets, so no
|
|
password is required.
|
|
|
|
Returns:
|
|
Stripped stdout of the psql command.
|
|
"""
|
|
result = fcos_host.run(
|
|
f"podman exec postgresql-server psql -U postgres -d {database} --csv -t -c %s", query
|
|
)
|
|
if check:
|
|
assert result.exit_status == 0, f"SQL query \"{query}\" failed with exit code {result.exit_status}: {result.stderr}"
|
|
return result.stdout.strip()
|
|
|
|
def test_postgresql_major_version_items(self, fcos_host):
|
|
"""The major version from the config must be reflected in the filesystem and in the running Podman image."""
|
|
self.check_expected_files(fcos_host, [
|
|
{ "path": f"/var/lib/quadlets/postgresql/{self.expected_pg_major}", "type": "directory", "owner": "postgresql", "group": "itix-svc", "mode": 0o755 },
|
|
])
|
|
self.check_expected_podman_images(fcos_host, [
|
|
{ "name": "docker.io/library/postgres", "tag": f"{self.expected_pg_major}-alpine", "state": "present" },
|
|
])
|
|
|
|
def test_latest_symlink_exists(self, fcos_host):
|
|
"""The 'latest' symlink must point to the active major-version directory."""
|
|
link = fcos_host.file("/var/lib/quadlets/postgresql/latest")
|
|
assert link.exists
|
|
assert link.is_symlink
|
|
assert link.linked_to == f"/var/lib/quadlets/postgresql/{self.expected_pg_major}"
|
|
|
|
def test_postgresql_accepts_connections(self, fcos_host):
|
|
"""PostgreSQL must respond to a trivial SQL query."""
|
|
output = self._run_sql(fcos_host, "SELECT 1 AS probe")
|
|
assert output == "1", f"Unexpected output from SQL query: {output}"
|
|
|
|
def test_postgresql_version_matches_config(self, fcos_host):
|
|
"""The running PostgreSQL server must report the version from PG_MAJOR_DEFAULT."""
|
|
output = self._run_sql(fcos_host, "SHOW server_version")
|
|
assert output.startswith(f"{self.expected_pg_major}."), f"Expected PostgreSQL server version to start with {self.expected_pg_major}, but got {output}"
|
|
|