Commit 38fe64fd03c56a2bd9e1fda310a78ddd6be47800 - golang-github-vbauerster-mpb

refactoring: TestWithContext Vladimir Bauer 3 years ago

1 changed file(s) with 9 addition(s) and 9 deletion(s). Raw diff Collapse all Expand all

-9

progress_test.go less more

92	92	func TestWithContext(t *testing.T) {
93	93	shutdown := make(chan struct{})
94	94	ctx, cancel := context.WithCancel(context.Background())
95		p := mpb.NewWithContext(ctx, mpb.WithShutdownNotifier(shutdown), mpb.WithOutput(io.Discard))
96
97		bar := p.AddBar(0) // never complete bar
	95	p := mpb.NewWithContext(ctx,
	96	mpb.WithShutdownNotifier(shutdown),
	97	mpb.WithOutput(io.Discard),
	98	)
	99	_ = p.AddBar(0) // never complete bar
	100	_ = p.AddBar(0) // never complete bar
98	101	go func() {
99		for !bar.Aborted() {
100		time.Sleep(randomDuration(100 * time.Millisecond))
101		cancel()
102		}
	102	time.Sleep(randomDuration(100 * time.Millisecond))
	103	cancel()
	104	p.Wait()
103	105	}()
104
105		go p.Wait()
106	106
107	107	select {
108	108	case <-shutdown: