72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package packer
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cheggaaa/pb"
|
|
)
|
|
|
|
func speedyProgressBar(bar *pb.ProgressBar) {
|
|
bar.SetUnits(pb.U_BYTES)
|
|
bar.SetRefreshRate(1 * time.Millisecond)
|
|
bar.NotPrint = true
|
|
bar.Format("[\x00=\x00>\x00-\x00]")
|
|
}
|
|
|
|
func TestStackableProgressBar_race(t *testing.T) {
|
|
bar := &StackableProgressBar{
|
|
ConfigProgressbarFN: speedyProgressBar,
|
|
}
|
|
|
|
start42Fn := func() { bar.Start(42) }
|
|
finishFn := func() { bar.Finish() }
|
|
add21 := func() { bar.Add(21) }
|
|
// prefix := func() { bar.prefix() }
|
|
|
|
type fields struct {
|
|
pre func()
|
|
calls []func()
|
|
post func()
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
iterations int
|
|
}{
|
|
{"all public", fields{nil, []func(){start42Fn, finishFn, add21, add21}, finishFn}, 300},
|
|
{"add", fields{start42Fn, []func(){add21}, finishFn}, 300},
|
|
{"add start", fields{start42Fn, []func(){start42Fn, add21}, finishFn}, 300},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
tt := tt
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
for i := 0; i < tt.iterations; i++ {
|
|
if tt.fields.pre != nil {
|
|
tt.fields.pre()
|
|
}
|
|
var startWg, endWg sync.WaitGroup
|
|
startWg.Add(1)
|
|
endWg.Add(len(tt.fields.calls))
|
|
for _, call := range tt.fields.calls {
|
|
call := call
|
|
go func() {
|
|
defer endWg.Done()
|
|
startWg.Wait()
|
|
call()
|
|
}()
|
|
}
|
|
startWg.Done() // everyone is initialized, let's unlock everyone at the same time.
|
|
// ....
|
|
endWg.Wait() // wait for all calls to return.
|
|
if tt.fields.post != nil {
|
|
tt.fields.post()
|
|
}
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|