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.
61 lines
2.1 KiB
Go
61 lines
2.1 KiB
Go
package recovery
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestApplyInvokesCLIWithEnvCredentials(t *testing.T) {
|
|
dir := t.TempDir()
|
|
log := argsFile(t, dir)
|
|
withFakeExecutable(t, "stalwart-cli", fmt.Sprintf("#!/bin/sh\necho \"$@ URL=$STALWART_URL USER=$STALWART_USER\" >> %q\nexit 0\n", log))
|
|
|
|
err := Apply(context.Background(), ApplyOptions{URL: "http://127.0.0.1:8080", User: "admin", Password: "secret"}, "/tmp/export.json")
|
|
if err != nil {
|
|
t.Fatalf("Apply: %v", err)
|
|
}
|
|
got := readArgsFile(t, log)
|
|
for _, want := range []string{"apply", "--file /tmp/export.json", "URL=http://127.0.0.1:8080", "USER=admin"} {
|
|
if !strings.Contains(got, want) {
|
|
t.Errorf("stalwart-cli invoked with %q, missing %q", got, want)
|
|
}
|
|
}
|
|
if strings.Contains(got, "secret") {
|
|
t.Error("password should not appear in argv")
|
|
}
|
|
}
|
|
|
|
func TestApplyPropagatesFailure(t *testing.T) {
|
|
withFakeExecutable(t, "stalwart-cli", "#!/bin/sh\necho 'invalid object' >&2\nexit 1\n")
|
|
err := Apply(context.Background(), ApplyOptions{URL: "http://127.0.0.1:8080", User: "admin", Password: "x"}, "/tmp/export.json")
|
|
if err == nil {
|
|
t.Fatal("Apply should error when stalwart-cli exits non-zero")
|
|
}
|
|
if !strings.Contains(err.Error(), "invalid object") {
|
|
t.Errorf("error = %v, want it to include stderr", err)
|
|
}
|
|
}
|
|
|
|
func TestApplyAllStopsAtFirstFailure(t *testing.T) {
|
|
dir := t.TempDir()
|
|
log := argsFile(t, dir)
|
|
script := fmt.Sprintf("#!/bin/sh\necho \"$@\" >> %q\ncase \"$*\" in\n *bad.json*) exit 1 ;;\n *) exit 0 ;;\nesac\n", log)
|
|
withFakeExecutable(t, "stalwart-cli", script)
|
|
|
|
err := ApplyAll(context.Background(), ApplyOptions{URL: "http://127.0.0.1:8080", User: "admin", Password: "x"},
|
|
[]string{filepath.Join(dir, "good1.json"), filepath.Join(dir, "bad.json"), filepath.Join(dir, "good2.json")})
|
|
if err == nil {
|
|
t.Fatal("ApplyAll should fail on bad.json")
|
|
}
|
|
got := readArgsFile(t, log)
|
|
if strings.Contains(got, "good2.json") {
|
|
t.Error("ApplyAll should stop before applying good2.json after bad.json failed")
|
|
}
|
|
if !strings.Contains(err.Error(), "1/3") {
|
|
t.Errorf("error = %v, want it to report 1/3 files applied before failing", err)
|
|
}
|
|
}
|