3x-ui/internal/web/service/server_pg_restore_test.go

package service

import "testing"

func TestPgRestoreReadFailureError(t *testing.T) {
	cases := []struct {
		name         string
		probeOutput  string
		localVersion string
		want         string
	}{
		{
			name:         "dump from postgres 17 on older client",
			probeOutput:  "pg_restore: error: unsupported version (1.16) in file header",
			localVersion: "16.4",
			want:         "This backup was created by pg_dump from PostgreSQL 17 or newer, but the server's pg_restore is version 16.4 and cannot read it; upgrade the postgresql-client package to version 17 or newer and retry the import",
		},
		{
			name:         "dump from postgres 16 on older client",
			probeOutput:  "pg_restore: error: unsupported version (1.15) in file header",
			localVersion: "15.8",
			want:         "This backup was created by pg_dump from PostgreSQL 16 or newer, but the server's pg_restore is version 15.8 and cannot read it; upgrade the postgresql-client package to version 16 or newer and retry the import",
		},
		{
			name:         "archive version newer than any known mapping",
			probeOutput:  "pg_restore: error: unsupported version (1.17) in file header",
			localVersion: "17.2",
			want:         "This backup was created by a newer pg_dump than the server's pg_restore (version 17.2) can read; upgrade the postgresql-client package and retry the import",
		},
		{
			name:         "local version could not be determined",
			probeOutput:  "pg_restore: error: unsupported version (1.16) in file header",
			localVersion: "",
			want:         "This backup was created by pg_dump from PostgreSQL 17 or newer, but the server's pg_restore is version unknown and cannot read it; upgrade the postgresql-client package to version 17 or newer and retry the import",
		},
		{
			name:         "unrelated read failure passes through",
			probeOutput:  "pg_restore: error: could not read from input file: end of file",
			localVersion: "16.4",
			want:         "pg_restore cannot read this dump file: pg_restore: error: could not read from input file: end of file",
		},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			err := pgRestoreReadFailureError(tc.probeOutput, tc.localVersion)
			if err == nil {
				t.Fatal("pgRestoreReadFailureError returned nil, want error")
			}
			if err.Error() != tc.want {
				t.Errorf("pgRestoreReadFailureError(%q, %q) = %q, want %q", tc.probeOutput, tc.localVersion, err.Error(), tc.want)
			}
		})
	}
}

func TestParsePgToolVersion(t *testing.T) {
	cases := []struct {
		name string
		in   string
		want string
	}{
		{"plain", "pg_restore (PostgreSQL) 17.2\n", "17.2"},
		{"debian packaging suffix", "pg_restore (PostgreSQL) 16.10 (Debian 16.10-1.pgdg120+1)\n", "16.10"},
		{"three component version", "pg_restore (PostgreSQL) 9.6.24\n", "9.6.24"},
		{"no version present", "pg_restore malfunction\n", ""},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			got := parsePgToolVersion(tc.in)
			if got != tc.want {
				t.Errorf("parsePgToolVersion(%q) = %q, want %q", tc.in, got, tc.want)
			}
		})
	}
}