jcoffey-dev
719a945d64
In-place upgrade tool for Stalwart Mail Server (0.15.5 -> latest) with checkpointed rollback and post-migration validation. Design stage; see ARCHITECTURE.md.
57 lines
2.2 KiB
Go
57 lines
2.2 KiB
Go
package backup
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestBuildVandelayImportArgs(t *testing.T) {
|
|
args := BuildVandelayImportArgs(VandelayOptions{URL: "https://mail.example.com", AuthBasic: "alice:app-pass"}, "[email protected]", "/backups/alice.sqlite")
|
|
joined := strings.Join(args, " ")
|
|
for _, want := range []string{"import", "jmap", "--url https://mail.example.com", "--auth-basic alice:app-pass", "--account-name [email protected]", "/backups/alice.sqlite"} {
|
|
if !strings.Contains(joined, want) {
|
|
t.Errorf("BuildVandelayImportArgs = %q, missing %q", joined, want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestExportAccountsRunsEachAccount(t *testing.T) {
|
|
dir := t.TempDir()
|
|
log := argsFile(t, dir)
|
|
withFakeExecutable(t, "vandelay", fakeScriptLoggingArgs(log, "exit 0"))
|
|
|
|
outDir := filepath.Join(dir, "out")
|
|
files, err := ExportAccounts(context.Background(), VandelayOptions{URL: "https://mail.example.com", AuthBasic: "a:b", OutDir: outDir}, []string{"[email protected]", "[email protected]"})
|
|
if err != nil {
|
|
t.Fatalf("ExportAccounts: %v", err)
|
|
}
|
|
if len(files) != 2 {
|
|
t.Fatalf("got %d files, want 2: %v", len(files), files)
|
|
}
|
|
got := readArgsFile(t, log)
|
|
if !strings.Contains(got, "[email protected]") || !strings.Contains(got, "[email protected]") {
|
|
t.Errorf("vandelay invocations = %q, want both accounts", got)
|
|
}
|
|
}
|
|
|
|
func TestExportAccountsStopsAtFirstFailure(t *testing.T) {
|
|
dir := t.TempDir()
|
|
// Fails on the second invocation (bob), succeeds on the first (alice).
|
|
script := "#!/bin/sh\ncase \"$*\" in\n *bob*) echo 'account not found' >&2; exit 1 ;;\n *) exit 0 ;;\nesac\n"
|
|
withFakeExecutable(t, "vandelay", script)
|
|
|
|
outDir := filepath.Join(dir, "out")
|
|
files, err := ExportAccounts(context.Background(), VandelayOptions{URL: "https://mail.example.com", AuthBasic: "a:b", OutDir: outDir}, []string{"[email protected]", "[email protected]", "[email protected]"})
|
|
if err == nil {
|
|
t.Fatal("ExportAccounts should have failed on [email protected]")
|
|
}
|
|
if len(files) != 1 {
|
|
t.Errorf("got %d successful exports before failure, want 1 (alice only)", len(files))
|
|
}
|
|
if !strings.Contains(err.Error(), "1/3") {
|
|
t.Errorf("error = %v, want it to report 1/3 accounts exported before failing", err)
|
|
}
|
|
}
|