77 lines
3.4 KiB
Go
77 lines
3.4 KiB
Go
package restic
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/backupper-unraid/backupper/internal/model"
|
|
)
|
|
|
|
type fakeSecrets map[string]string
|
|
|
|
func (s fakeSecrets) Get(id string) (string, error) { return s[id], nil }
|
|
|
|
func TestBackupUsesPasswordFileAndStructuredArguments(t *testing.T) {
|
|
dir := t.TempDir()
|
|
argsPath := filepath.Join(dir, "args")
|
|
passwordPath := filepath.Join(dir, "password")
|
|
script := filepath.Join(dir, "restic")
|
|
body := fmt.Sprintf("#!/bin/sh\nprintf '%%s\\n' \"$@\" > '%s'\ncat \"$RESTIC_PASSWORD_FILE\" > '%s'\nprintf '%%s\\n' '{\"message_type\":\"status\",\"percent_done\":0.5,\"total_files\":12,\"files_done\":6,\"total_bytes\":1024,\"bytes_done\":512,\"seconds_remaining\":4,\"current_files\":[\"/source path/file.txt\"]}'\nprintf '%%s\\n' '{\"message_type\":\"summary\",\"snapshot_id\":\"abc123\",\"data_added\":42,\"files_new\":3,\"files_changed\":2,\"total_bytes_processed\":1024,\"total_files_processed\":12}'\n", argsPath, passwordPath)
|
|
if err := os.WriteFile(script, []byte(body), 0700); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
runner := &Runner{Binary: script, RuntimeDir: dir, Secrets: fakeSecrets{"password": "top-secret"}}
|
|
repo := model.Repository{Type: model.RepositoryLocal, Location: "/repo path", PasswordRef: "password"}
|
|
job := model.Job{ID: "job", Compression: "auto", Tags: []string{"daily"}, Excludes: []string{"*.tmp"}}
|
|
var progress Progress
|
|
summary, err := runner.Backup(context.Background(), repo, job, []string{"/source path"}, func(value Progress) { progress = value })
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if summary.SnapshotID != "abc123" || summary.DataAdded != 42 || summary.FilesChanged != 2 || summary.TotalBytesProcessed != 1024 || summary.TotalFilesProcessed != 12 {
|
|
t.Fatalf("unexpected summary: %+v", summary)
|
|
}
|
|
if progress.PercentDone != 0.5 || progress.FilesDone != 6 || progress.BytesDone != 512 || len(progress.CurrentFiles) != 1 {
|
|
t.Fatalf("unexpected progress: %+v", progress)
|
|
}
|
|
args, _ := os.ReadFile(argsPath)
|
|
for _, expected := range []string{"/repo path", "backup", "--json", "/source path"} {
|
|
if !strings.Contains(string(args), expected) {
|
|
t.Fatalf("arguments missing %q: %s", expected, args)
|
|
}
|
|
}
|
|
password, _ := os.ReadFile(passwordPath)
|
|
if string(password) != "top-secret" {
|
|
t.Fatal("password file was not provided")
|
|
}
|
|
}
|
|
|
|
func TestForgetUsesAgeAndCalendarRetention(t *testing.T) {
|
|
dir := t.TempDir()
|
|
argsPath := filepath.Join(dir, "args")
|
|
script := filepath.Join(dir, "restic")
|
|
body := fmt.Sprintf("#!/bin/sh\nprintf '%%s\\n' \"$@\" > '%s'\n", argsPath)
|
|
if err := os.WriteFile(script, []byte(body), 0700); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
runner := &Runner{Binary: script, RuntimeDir: dir, Secrets: fakeSecrets{"password": "secret"}}
|
|
repo := model.Repository{Type: model.RepositoryLocal, Location: "/repo", PasswordRef: "password"}
|
|
job := model.Job{ID: "job", Retention: model.Retention{KeepWithinDays: 30, Weekly: 4, Monthly: 12}}
|
|
if err := runner.Forget(context.Background(), repo, job); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
args, _ := os.ReadFile(argsPath)
|
|
for _, expected := range []string{"--keep-within", "30d", "--keep-weekly", "4", "--keep-monthly", "12"} {
|
|
if !strings.Contains(string(args), expected) {
|
|
t.Fatalf("retention arguments missing %q: %s", expected, args)
|
|
}
|
|
}
|
|
if strings.Contains(string(args), "--keep-daily") {
|
|
t.Fatalf("disabled daily retention was included: %s", args)
|
|
}
|
|
}
|