check bar.Current() in TestBarEnableTriggerCompleteAndIncrementAfter

vbauerster · Oct 7, 2023 · fab4434 · fab4434
1 parent de4e3ab
commit fab4434
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/bar_test.go b/bar_test.go
@@ -102,11 +102,14 @@ func TestBarEnableTriggerCompleteAndIncrementAfter(t *testing.T) {
 	p := mpb.New(mpb.WithWidth(80), mpb.WithOutput(io.Discard))
 	bar := p.AddBar(0) // never complete bar
 
+	targetTotal := int64(80)
+
 	for _, f := range []func(){
 		func() { bar.SetTotal(40, false) },
 		func() { bar.IncrBy(60) },
-		func() { bar.SetTotal(80, false) },
-		func() { bar.EnableTriggerComplete() },
+		func() { bar.SetTotal(targetTotal, false) },
+		func() { bar.EnableTriggerComplete() }, // disables any next SetTotal
+		func() { bar.SetTotal(100, true) },     // nop
 	} {
 		f()
 		if bar.Completed() {
@@ -120,6 +123,10 @@ func TestBarEnableTriggerCompleteAndIncrementAfter(t *testing.T) {
 		t.Fail()
 	}
 
+	if current := bar.Current(); current != targetTotal {
+		t.Errorf("Expected current: %d, got: %d", targetTotal, current)
+	}
+
 	p.Wait()
 }