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.
56 lines
2.5 KiB
56 lines
2.5 KiB
import pytest
|
|
import textwrap
|
|
|
|
# Add the current cookbook's tests directory to the path so we can import helpers.py.
|
|
from pathlib import Path
|
|
import sys
|
|
sys.path.insert(0, str(Path(__file__).parent))
|
|
import helpers # noqa: E402
|
|
|
|
# Major version of PostgreSQL to install by default on a fresh VM boot.
|
|
PG_MAJOR_DEFAULT = 18
|
|
|
|
# Extra files to inject into the FCOS image for the tests in this file.
|
|
# The config.env is used to configure the PostgreSQL Quadlet.
|
|
PYTEST_FCOS_EXTRA_FILES = {
|
|
"/etc/quadlets/postgresql/config.env": (
|
|
textwrap.dedent(f"""
|
|
# This file is generated by conftest.py for testing purposes.
|
|
POSTGRES_USER=postgres
|
|
POSTGRES_PASSWORD=postgres
|
|
POSTGRES_DB=postgres
|
|
POSTGRES_HOST_AUTH_METHOD=scram-sha-256
|
|
POSTGRES_INITDB_ARGS=--auth-host=scram-sha-256
|
|
POSTGRES_ARGS=-h 127.0.0.1
|
|
PGPORT=5432
|
|
PG_MAJOR={PG_MAJOR_DEFAULT}
|
|
POSTGRES_BACKUP_RETENTION=7
|
|
"""),
|
|
0,
|
|
0,
|
|
0o600,
|
|
),
|
|
}
|
|
|
|
"""
|
|
Verify that the postgresql Quadlet correctly restores a database from a backup
|
|
on a fresh VM with the backup data present in the virtiofs.
|
|
"""
|
|
class TestPostgresqlQuadletRestore(helpers.TestPostgresqlQuadlet):
|
|
expected_pg_major = PG_MAJOR_DEFAULT
|
|
|
|
def test_data_is_still_there_after_restore(self, fcos_host):
|
|
"""Data created before the restore must still be there after the restore."""
|
|
# Check that the old data is still there after the restore
|
|
output = self._run_sql(fcos_host, "SELECT datname FROM pg_database WHERE datname = 'upgrade_path_db'")
|
|
assert output == "upgrade_path_db", f"Unexpected output from SQL query: {output}"
|
|
output = self._run_sql(fcos_host, "SELECT datname FROM pg_database WHERE datname = 'testdb'")
|
|
assert output == "testdb", f"Unexpected output from SQL query: {output}"
|
|
result = fcos_host.run(
|
|
"podman exec postgresql-server psql -U test -d testdb --csv -t -c %s", "SELECT 1 AS probe"
|
|
)
|
|
assert result.exit_status == 0, f"SQL query failed with exit code {result.exit_status}: {result.stderr}"
|
|
|
|
# Check that the upgrade_path table contains the initial postgresql version (14)
|
|
output = self._run_sql(fcos_host, "SELECT LEFT(version, 14) FROM upgrade_path ORDER BY version ASC LIMIT 1", database="upgrade_path_db")
|
|
assert output.startswith("PostgreSQL 14."), f"Unexpected output from SQL query: {output}"
|
|
|