blob: cb56658d17b58f292b5886a7ad006c7c66a80fee (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
package up
import (
"sync"
"sync/atomic"
"testing"
"time"
)
func TestUp(t *testing.T) {
pr := New()
wg := sync.WaitGroup{}
hits := int32(0)
upfunc := func(s string) bool {
atomic.AddInt32(&hits, 1)
// Sleep tiny amount so that our other pr.Do() calls hit the lock.
time.Sleep(3 * time.Millisecond)
wg.Done()
return true
}
pr.Start("nonexistent", 5*time.Millisecond)
defer pr.Stop()
// These functions AddInt32 to the same hits variable, but we only want to wait when
// upfunc finishes, as that only calls Done() on the waitgroup.
upfuncNoWg := func(s string) bool { atomic.AddInt32(&hits, 1); return true }
wg.Add(1)
pr.Do(upfunc)
pr.Do(upfuncNoWg)
pr.Do(upfuncNoWg)
wg.Wait()
h := atomic.LoadInt32(&hits)
if h != 1 {
t.Errorf("Expected hits to be %d, got %d", 1, h)
}
}
|
