jcoffey-dev
77e80cb9f2
119 SPDX-FileCopyrightText headers and the README's licence line. The distinction that matters here: LINUXexpert-org appears in this repository in two completely different roles. As a copyright holder in the SPDX headers, which is what changes, and as the GitHub organisation in the module path and 64 import statements, which does not -- the repository still lives at github.com/LINUXexpert-org/stalwart-migrator, and rewriting that would not be a licence change, it would break the build. Both replacements are anchored to their copyright forms, so an import path cannot match either. Import count is 64 before and after, and go.mod is untouched. LICENSE untouched: the FSF's copyright on the GPL text and the "<name of author>" placeholders are not ours to edit. go vet, go build and go test all clean.
89 lines
2.9 KiB
Go
89 lines
2.9 KiB
Go
// SPDX-FileCopyrightText: 2026 Coffey Labs
|
|
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
package backup
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestBuildPgDumpArgs(t *testing.T) {
|
|
args := BuildPgDumpArgs(SQLOptions{
|
|
Host: "db.internal", Port: "5432", Database: "stalwart", User: "stalwart", OutPath: "/backups/out.sql",
|
|
})
|
|
joined := strings.Join(args, " ")
|
|
for _, want := range []string{"-U stalwart", "-d stalwart", "-h db.internal", "-p 5432", "-f /backups/out.sql"} {
|
|
if !strings.Contains(joined, want) {
|
|
t.Errorf("BuildPgDumpArgs = %q, missing %q", joined, want)
|
|
}
|
|
}
|
|
for _, table := range criticalTables {
|
|
if !strings.Contains(joined, "-t "+table) {
|
|
t.Errorf("BuildPgDumpArgs missing critical table flag for %q: %q", table, joined)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestBuildMySQLDumpArgs(t *testing.T) {
|
|
args := BuildMySQLDumpArgs(SQLOptions{Host: "db.internal", Port: "3306", Database: "stalwart", User: "stalwart"})
|
|
joined := strings.Join(args, " ")
|
|
if !strings.Contains(joined, "-u stalwart") || !strings.Contains(joined, "-h db.internal") || !strings.Contains(joined, "-P 3306") {
|
|
t.Errorf("BuildMySQLDumpArgs = %q, missing expected flags", joined)
|
|
}
|
|
if !strings.HasSuffix(joined, strings.Join(criticalTables, " ")) {
|
|
t.Errorf("BuildMySQLDumpArgs = %q, want it to end with the critical table list", joined)
|
|
}
|
|
}
|
|
|
|
func TestRunPgDumpInvokesPgDumpWithArgs(t *testing.T) {
|
|
dir := t.TempDir()
|
|
log := argsFile(t, dir)
|
|
withFakeExecutable(t, "pg_dump", fakeScriptLoggingArgs(log, "exit 0"))
|
|
|
|
err := RunPgDump(context.Background(), SQLOptions{
|
|
User: "stalwart", Database: "stalwart", OutPath: filepath.Join(dir, "out.sql"),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("RunPgDump: %v", err)
|
|
}
|
|
got := readArgsFile(t, log)
|
|
if !strings.Contains(got, "-U stalwart") {
|
|
t.Errorf("pg_dump was invoked with %q, missing -U stalwart", got)
|
|
}
|
|
}
|
|
|
|
func TestRunPgDumpPropagatesFailure(t *testing.T) {
|
|
dir := t.TempDir()
|
|
withFakeExecutable(t, "pg_dump", "#!/bin/sh\necho 'connection refused' >&2\nexit 1\n")
|
|
|
|
err := RunPgDump(context.Background(), SQLOptions{User: "stalwart", Database: "stalwart", OutPath: filepath.Join(dir, "out.sql")})
|
|
if err == nil {
|
|
t.Fatal("RunPgDump should have returned an error when pg_dump exits non-zero")
|
|
}
|
|
if !strings.Contains(err.Error(), "connection refused") {
|
|
t.Errorf("error = %v, want it to include pg_dump's stderr", err)
|
|
}
|
|
}
|
|
|
|
func TestRunMySQLDumpWritesStdoutToOutPath(t *testing.T) {
|
|
dir := t.TempDir()
|
|
withFakeExecutable(t, "mysqldump", "#!/bin/sh\necho '-- fake dump output'\n")
|
|
|
|
outPath := filepath.Join(dir, "out.sql")
|
|
err := RunMySQLDump(context.Background(), SQLOptions{User: "stalwart", Database: "stalwart", OutPath: outPath})
|
|
if err != nil {
|
|
t.Fatalf("RunMySQLDump: %v", err)
|
|
}
|
|
data, err := os.ReadFile(outPath)
|
|
if err != nil {
|
|
t.Fatalf("ReadFile: %v", err)
|
|
}
|
|
if !strings.Contains(string(data), "fake dump output") {
|
|
t.Errorf("out.sql = %q, want fake dump output redirected into it", data)
|
|
}
|
|
}
|