stats: refactor ui progress printer into ui/stats

internal/ui/stats/progress_test.go

@@ -0,0 +1,44 @@
+package stats
+
+import (
+	"testing"
+	"time"
+
+	rtest "github.com/restic/restic/internal/test"
+	"github.com/restic/restic/internal/ui"
+)
+
+func TestStatsProgress(t *testing.T) {
+	term := &ui.MockTerminal{}
+
+	progress := newProgress(term, true, 2)
+	progress.printProgress(0*time.Second, false)
+	rtest.Equals(t, []string{"[0:00] 0.00%  0 / 2 snapshots, 0 B"}, term.Output)
+
+	progress.ProcessSnapshot()
+	progress.Update(1, 2, 3)
+	progress.printProgress(5*time.Second, false)
+	// Output differs from the previous one because the progress is based on the number of processed snapshots,
+	// 1/2 snapshots means processing the snapshot 1 currently
+	rtest.Equals(t, []string{"[0:05] 0.00%  1 / 2 snapshots, 1 files, 2 blobs, 3 B"}, term.Output)
+
+	progress.ProcessSnapshot()
+	progress.printProgress(10*time.Second, false)
+	rtest.Equals(t, []string{"[0:10] 50.00%  2 / 2 snapshots, 0 B"}, term.Output)
+
+	progress.Update(4, 5, 6)
+	progress.printProgress(15*time.Second, false)
+	rtest.Equals(t, []string{"[0:15] 50.00%  2 / 2 snapshots, 4 files, 5 blobs, 6 B"}, term.Output)
+
+	progress.printProgress(20*time.Second, true)
+	rtest.Equals(t, []string{"[0:20] 100.00%  2 / 2 snapshots, 4 files, 5 blobs, 6 B"}, term.Output)
+}
+
+func TestStatsProgressJSON(t *testing.T) {
+	term := &ui.MockTerminal{}
+
+	progress := newProgress(term, false, 2)
+	progress.printProgress(0*time.Second, false)
+	// JSON output is not available yet, so just make sure to not break normal json output
+	rtest.Equals(t, nil, term.Output)
+}