File tree 2 files changed +9
-16
lines changed
2 files changed +9
-16
lines changed Original file line number Diff line number Diff line change 5
5
"time"
6
6
7
7
"github.com/golang-queue/queue/core"
8
+ "github.com/golang-queue/queue/mocks"
8
9
10
+ "github.com/golang/mock/gomock"
9
11
"github.com/stretchr/testify/assert"
10
12
"go.uber.org/goleak"
11
13
)
@@ -23,11 +25,16 @@ func (m mockMessage) Bytes() []byte {
23
25
}
24
26
25
27
func TestNewQueue (t * testing.T ) {
28
+ controller := gomock .NewController (t )
29
+ defer controller .Finish ()
30
+
26
31
q , err := NewQueue ()
27
32
assert .Error (t , err )
28
33
assert .Nil (t , q )
29
34
30
- w := & emptyWorker {}
35
+ w := mocks .NewMockWorker (controller )
36
+ w .EXPECT ().Shutdown ().Return (nil )
37
+ w .EXPECT ().Request ().Return (nil , nil )
31
38
q , err = NewQueue (
32
39
WithWorker (w ),
33
40
)
@@ -36,8 +43,7 @@ func TestNewQueue(t *testing.T) {
36
43
37
44
q .Start ()
38
45
assert .Equal (t , 0 , q .BusyWorkers ())
39
- q .Shutdown ()
40
- q .Wait ()
46
+ q .Release ()
41
47
}
42
48
43
49
func TestShtdonwOnce (t * testing.T ) {
Load Diff This file was deleted.
You can’t perform that action at this time.
0 commit comments