2024-06-05 16:37:31 +00:00
|
|
|
(import ../lib/container-test.nix) ({
|
|
|
|
name = "postgresql";
|
|
|
|
|
|
|
|
nodes.machine =
|
|
|
|
{ self, config, ... }:
|
|
|
|
{
|
|
|
|
imports = [
|
|
|
|
self.nixosModules.clanCore
|
|
|
|
self.clanModules.postgresql
|
|
|
|
self.clanModules.localbackup
|
|
|
|
];
|
|
|
|
clan.postgresql.users.test = { };
|
|
|
|
clan.postgresql.databases.test.create.options.OWNER = "test";
|
2024-06-06 13:17:47 +00:00
|
|
|
clan.postgresql.databases.test.restore.stopOnRestore = [ "sample-service" ];
|
2024-06-05 16:37:31 +00:00
|
|
|
clan.localbackup.targets.hdd.directory = "/mnt/external-disk";
|
|
|
|
|
2024-06-06 13:17:47 +00:00
|
|
|
systemd.services.sample-service = {
|
|
|
|
wantedBy = [ "multi-user.target" ];
|
|
|
|
script = ''
|
|
|
|
while true; do
|
|
|
|
echo "Hello, world!"
|
|
|
|
sleep 5
|
|
|
|
done
|
|
|
|
'';
|
|
|
|
};
|
|
|
|
|
2024-06-05 16:37:31 +00:00
|
|
|
environment.systemPackages = [ config.services.postgresql.package ];
|
|
|
|
};
|
|
|
|
testScript =
|
|
|
|
{ nodes, ... }:
|
|
|
|
''
|
2024-05-31 14:36:37 +00:00
|
|
|
start_all()
|
|
|
|
machine.wait_for_unit("postgresql")
|
2024-06-06 13:17:47 +00:00
|
|
|
machine.wait_for_unit("sample-service")
|
2024-06-05 16:37:31 +00:00
|
|
|
# Create a test table
|
|
|
|
machine.succeed("runuser -u postgres -- /run/current-system/sw/bin/psql -c 'CREATE TABLE test (id serial PRIMARY KEY);' test")
|
|
|
|
|
2024-06-04 10:56:39 +00:00
|
|
|
machine.succeed("/run/current-system/sw/bin/localbackup-create >&2")
|
2024-06-06 13:17:47 +00:00
|
|
|
timestamp_before = int(machine.succeed("systemctl show --property=ExecMainStartTimestampMonotonic sample-service | cut -d= -f2").strip())
|
2024-06-05 16:37:31 +00:00
|
|
|
|
2024-06-06 13:17:47 +00:00
|
|
|
machine.succeed("test -e /mnt/external-disk/snapshot.0/machine/var/backup/postgres/test/pg-dump || { echo 'pg-dump not found'; exit 1; }")
|
2024-06-05 16:37:31 +00:00
|
|
|
machine.succeed("runuser -u postgres -- /run/current-system/sw/bin/psql -d test -c 'INSERT INTO test DEFAULT VALUES;'")
|
|
|
|
machine.succeed("runuser -u postgres -- /run/current-system/sw/bin/psql -d test -c 'DROP TABLE test;'")
|
2024-06-06 13:17:47 +00:00
|
|
|
machine.succeed("test -e /var/backup/postgres/test/pg-dump || { echo 'pg-dump not found'; exit 1; }")
|
2024-06-05 16:37:31 +00:00
|
|
|
|
|
|
|
machine.succeed("rm -rf /var/backup/postgres")
|
|
|
|
|
|
|
|
machine.succeed("NAME=/mnt/external-disk/snapshot.0 FOLDERS=/var/backup/postgres/test /run/current-system/sw/bin/localbackup-restore >&2")
|
2024-06-06 13:17:47 +00:00
|
|
|
machine.succeed("test -e /var/backup/postgres/test/pg-dump || { echo 'pg-dump not found'; exit 1; }")
|
2024-06-05 16:37:31 +00:00
|
|
|
|
|
|
|
machine.succeed("""
|
|
|
|
set -x
|
2024-06-17 10:42:28 +00:00
|
|
|
${nodes.machine.clan.core.state.postgresql-test.postRestoreCommand}
|
2024-06-05 16:37:31 +00:00
|
|
|
""")
|
|
|
|
machine.succeed("runuser -u postgres -- /run/current-system/sw/bin/psql -l >&2")
|
|
|
|
machine.succeed("runuser -u postgres -- /run/current-system/sw/bin/psql -d test -c '\dt' >&2")
|
|
|
|
|
2024-06-06 13:17:47 +00:00
|
|
|
timestamp_after = int(machine.succeed("systemctl show --property=ExecMainStartTimestampMonotonic sample-service | cut -d= -f2").strip())
|
|
|
|
assert timestamp_before < timestamp_after, f"{timestamp_before} >= {timestamp_after}: expected sample-service to be restarted after restore"
|
|
|
|
|
2024-06-05 16:37:31 +00:00
|
|
|
# Check that the table is still there
|
|
|
|
machine.succeed("runuser -u postgres -- /run/current-system/sw/bin/psql -d test -c 'SELECT * FROM test;'")
|
|
|
|
output = machine.succeed("runuser -u postgres -- /run/current-system/sw/bin/psql --csv -c \"SELECT datdba::regrole FROM pg_database WHERE datname = 'test'\"")
|
|
|
|
owner = output.split("\n")[1]
|
|
|
|
assert owner == "test", f"Expected database owner to be 'test', got '{owner}'"
|
2024-06-14 09:48:42 +00:00
|
|
|
|
|
|
|
# check if restore works if the database does not exist
|
|
|
|
machine.succeed("runuser -u postgres -- dropdb test")
|
|
|
|
machine.succeed("${nodes.machine.clanCore.state.postgresql-test.postRestoreCommand}")
|
|
|
|
machine.succeed("runuser -u postgres -- /run/current-system/sw/bin/psql -d test -c '\dt' >&2")
|
2024-05-31 14:36:37 +00:00
|
|
|
'';
|
|
|
|
})
|