25 lines
400 B
Go
25 lines
400 B
Go
|
package worker
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestWorker(t *testing.T) {
|
||
|
|
assert := assert.New(t)
|
||
|
|
|
||
|
|
runtime := 0
|
||
|
|
|
||
|
|
w := NewWorker(time.Duration(5)*time.Millisecond, func() {
|
||
|
|
runtime = runtime + 1
|
||
|
|
})
|
||
|
|
go w.Start()
|
||
|
|
time.Sleep(time.Duration(18) * time.Millisecond)
|
||
|
|
w.Close()
|
||
|
|
|
||
|
|
assert.Equal(3, runtime)
|
||
|
|
time.Sleep(time.Duration(8) * time.Millisecond)
|
||
|
|
}
|