parent
3a02c2003c
commit
b695b2c73f
@ -0,0 +1,80 @@ |
|||||||
|
package thread |
||||||
|
|
||||||
|
import ( |
||||||
|
"context" |
||||||
|
"github.com/stretchr/testify/assert" |
||||||
|
"strconv" |
||||||
|
"sync" |
||||||
|
"sync/atomic" |
||||||
|
"testing" |
||||||
|
"time" |
||||||
|
) |
||||||
|
|
||||||
|
type report struct { |
||||||
|
rw sync.Mutex |
||||||
|
text string |
||||||
|
} |
||||||
|
|
||||||
|
func (r *report) log(text string) { |
||||||
|
r.rw.Lock() |
||||||
|
defer r.rw.Unlock() |
||||||
|
r.text += text |
||||||
|
} |
||||||
|
|
||||||
|
func (r *report) String() string { |
||||||
|
return r.text |
||||||
|
} |
||||||
|
|
||||||
|
func Test_checkManager(t *testing.T) { |
||||||
|
var count atomic.Int32 |
||||||
|
var r report |
||||||
|
var wg sync.WaitGroup |
||||||
|
|
||||||
|
ctx, cancel := context.WithCancel(context.Background()) |
||||||
|
|
||||||
|
wg.Add(1) |
||||||
|
go func() { |
||||||
|
defer wg.Done() |
||||||
|
|
||||||
|
r.log("Start manger\n") |
||||||
|
|
||||||
|
manager := Manager{Count: 3} |
||||||
|
go manager.Start(func() { |
||||||
|
number := count.Add(1) |
||||||
|
r.log("Start " + strconv.Itoa(int(number)) + "\n") |
||||||
|
switch number { |
||||||
|
case 1: |
||||||
|
<-ctx.Done() |
||||||
|
time.Sleep(100 * time.Millisecond) |
||||||
|
case 2: |
||||||
|
time.Sleep(100 * time.Millisecond) |
||||||
|
case 3: |
||||||
|
time.Sleep(200 * time.Millisecond) |
||||||
|
case 4: |
||||||
|
time.Sleep(200 * time.Millisecond) |
||||||
|
case 5: |
||||||
|
<-ctx.Done() |
||||||
|
time.Sleep(200 * time.Millisecond) |
||||||
|
case 6: |
||||||
|
<-ctx.Done() |
||||||
|
time.Sleep(300 * time.Millisecond) |
||||||
|
default: |
||||||
|
<-ctx.Done() |
||||||
|
time.Sleep(400 * time.Millisecond) |
||||||
|
} |
||||||
|
r.log("Finish " + strconv.Itoa(int(number)) + "\n") |
||||||
|
}) |
||||||
|
|
||||||
|
r.log("Wait cancel\n") |
||||||
|
manager.Wait(ctx) |
||||||
|
r.log("Finish manger\n") |
||||||
|
}() |
||||||
|
|
||||||
|
time.Sleep(1 * time.Second) |
||||||
|
cancel() |
||||||
|
|
||||||
|
wg.Wait() |
||||||
|
|
||||||
|
assert.Equal(t, int32(6), count.Load()) |
||||||
|
assert.Equal(t, "Start manger\nWait cancel\nStart 1\nStart 2\nStart 3\nFinish 2\nStart 4\nFinish 3\nStart 5\nFinish 4\nStart 6\nFinish 1\nFinish 5\nFinish 6\nFinish manger\n", r.String()) |
||||||
|
} |
Loading…
Reference in new issue