forked from TrueCloudLab/restic
53 lines
1 KiB
Go
53 lines
1 KiB
Go
|
package progress_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/restic/restic/internal/test"
|
||
|
|
"github.com/restic/restic/internal/ui/progress"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestUpdater(t *testing.T) {
|
||
|
|
finalSeen := false
|
||
|
|
var ncalls int
|
||
|
|
|
||
|
|
report := func(d time.Duration, final bool) {
|
||
|
|
if final {
|
||
|
|
finalSeen = true
|
||
|
|
}
|
||
|
|
ncalls++
|
||
|
|
}
|
||
|
|
c := progress.NewUpdater(10*time.Millisecond, report)
|
||
|
|
time.Sleep(100 * time.Millisecond)
|
||
|
|
c.Done()
|
||
|
|
|
||
|
|
test.Assert(t, finalSeen, "final call did not happen")
|
||
|
|
test.Assert(t, ncalls > 0, "no progress was reported")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUpdaterStopTwice(t *testing.T) {
|
||
|
|
c := progress.NewUpdater(0, func(runtime time.Duration, final bool) {})
|
||
|
|
c.Done()
|
||
|
|
c.Done()
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUpdaterNoTick(t *testing.T) {
|
||
|
|
finalSeen := false
|
||
|
|
otherSeen := false
|
||
|
|
|
||
|
|
report := func(d time.Duration, final bool) {
|
||
|
|
if final {
|
||
|
|
finalSeen = true
|
||
|
|
} else {
|
||
|
|
otherSeen = true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
c := progress.NewUpdater(0, report)
|
||
|
|
time.Sleep(time.Millisecond)
|
||
|
|
c.Done()
|
||
|
|
|
||
|
|
test.Assert(t, finalSeen, "final call did not happen")
|
||
|
|
test.Assert(t, !otherSeen, "unexpected status update")
|
||
|
|
}
|