restic: RunUnlock + RunStats (raw-data mode)

Add RunUnlock (delegates straight to runWithPump) and RunStats which runs `restic stats --json --mode raw-data`, captures the single JSON line from stdout into RepoStats, and returns an error if no JSON arrives. Tests cover arg plumbing for unlock, JSON parsing, and the no-JSON error path.
2026-05-03 22:22:19 +01:00
parent 4cc6962ec6
commit 3336485d02
2 changed files with 100 additions and 0 deletions
@@ -96,3 +96,56 @@ func TestRunCheckSubsetArg(t *testing.T) {
 	}
 	t.Fatalf("expected %q in captured output; got: %v", want, *lines)
 }
+
+// --- B3: RunUnlock + RunStats ---
+
+func TestRunUnlockInvokesUnlock(t *testing.T) {
+	bin := setupScriptBin(t, `echo "$@"`)
+	env := Env{Bin: bin}
+	lines, h := captureLines()
+	if err := env.RunUnlock(context.Background(), h); err != nil {
+		t.Fatalf("RunUnlock: %v", err)
+	}
+	for _, l := range *lines {
+		if strings.Contains(l, "unlock") {
+			return
+		}
+	}
+	t.Fatalf("expected 'unlock' in captured output; got: %v", *lines)
+}
+
+func TestRunStatsParsesJSON(t *testing.T) {
+	bin := setupScriptBin(t, `echo '{"total_size":1234,"total_uncompressed_size":5678,"snapshots_count":3,"total_file_count":100,"total_blob_count":50}'`)
+	env := Env{Bin: bin}
+	stats, err := env.RunStats(context.Background(), nil)
+	if err != nil {
+		t.Fatalf("RunStats: %v", err)
+	}
+	if stats.TotalSize != 1234 {
+		t.Fatalf("TotalSize: got %d, want 1234", stats.TotalSize)
+	}
+	if stats.TotalUncompressed != 5678 {
+		t.Fatalf("TotalUncompressed: got %d, want 5678", stats.TotalUncompressed)
+	}
+	if stats.SnapshotsCount != 3 {
+		t.Fatalf("SnapshotsCount: got %d, want 3", stats.SnapshotsCount)
+	}
+	if stats.TotalFileCount != 100 {
+		t.Fatalf("TotalFileCount: got %d, want 100", stats.TotalFileCount)
+	}
+	if stats.TotalBlobCount != 50 {
+		t.Fatalf("TotalBlobCount: got %d, want 50", stats.TotalBlobCount)
+	}
+}
+
+func TestRunStatsErrorsWithoutJSON(t *testing.T) {
+	bin := setupScriptBin(t, `echo "no json here"`)
+	env := Env{Bin: bin}
+	_, err := env.RunStats(context.Background(), nil)
+	if err == nil {
+		t.Fatal("expected error when no JSON in output")
+	}
+	if !strings.Contains(err.Error(), "no JSON in output") {
+		t.Fatalf("unexpected error: %v", err)
+	}
+}