progress tracker: SetValue() to set a custom value (#73)

jedib0t · Dec 4, 2018 · f6b258e · f6b258e
1 parent b8cb96c
commit f6b258e
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/progress/tracker.go b/progress/tracker.go
@@ -80,6 +80,13 @@ func (t *Tracker) Reset() {
 	t.value = 0
 }
 
+// SetValue sets the value of the tracker and re-calculates if the tracker is
+// "done".
+func (t *Tracker) SetValue(value int64) {
+	t.Reset()
+	t.Increment(value)
+}
+
 func (t *Tracker) start() {
 	t.done = false
 	t.timeStart = time.Now()

diff --git a/progress/tracker_test.go b/progress/tracker_test.go
@@ -101,3 +101,17 @@ func TestTracker_Reset(t *testing.T) {
 	assert.Equal(t, time.Time{}, tracker.timeStop)
 	assert.Equal(t, int64(0), tracker.value)
 }
+
+func TestTracker_SetValue(t *testing.T) {
+	tracker := Tracker{Total: 100}
+	assert.Equal(t, int64(0), tracker.value)
+	assert.False(t, tracker.done)
+
+	tracker.SetValue(5)
+	assert.Equal(t, int64(5), tracker.value)
+	assert.False(t, tracker.done)
+
+	tracker.SetValue(tracker.Total)
+	assert.Equal(t, tracker.Total, tracker.value)
+	assert.True(t, tracker.done)
+}