File tree Expand file tree Collapse file tree 2 files changed +9
-16
lines changed Expand file tree Collapse file tree 2 files changed +9
-16
lines changed Original file line number Diff line number Diff line change 55 "time"
66
77 "github.com/golang-queue/queue/core"
8+ "github.com/golang-queue/queue/mocks"
89
10+ "github.com/golang/mock/gomock"
911 "github.com/stretchr/testify/assert"
1012 "go.uber.org/goleak"
1113)
@@ -23,11 +25,16 @@ func (m mockMessage) Bytes() []byte {
2325}
2426
2527func TestNewQueue (t * testing.T ) {
28+ controller := gomock .NewController (t )
29+ defer controller .Finish ()
30+
2631 q , err := NewQueue ()
2732 assert .Error (t , err )
2833 assert .Nil (t , q )
2934
30- w := & emptyWorker {}
35+ w := mocks .NewMockWorker (controller )
36+ w .EXPECT ().Shutdown ().Return (nil )
37+ w .EXPECT ().Request ().Return (nil , nil )
3138 q , err = NewQueue (
3239 WithWorker (w ),
3340 )
@@ -36,8 +43,7 @@ func TestNewQueue(t *testing.T) {
3643
3744 q .Start ()
3845 assert .Equal (t , 0 , q .BusyWorkers ())
39- q .Shutdown ()
40- q .Wait ()
46+ q .Release ()
4147}
4248
4349func TestShtdonwOnce (t * testing.T ) {
Load Diff This file was deleted.
You can’t perform that action at this time.
0 commit comments