Skip to content

Commit d1132ff

Browse files
committed
chore(testing): remove empty worker.
Signed-off-by: Bo-Yi Wu <[email protected]>
1 parent 3404d97 commit d1132ff

File tree

2 files changed

+9
-16
lines changed

2 files changed

+9
-16
lines changed

queue_test.go

+9-3
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,9 @@ import (
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

2527
func 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

4349
func TestShtdonwOnce(t *testing.T) {

worker_empty.go

-13
This file was deleted.

0 commit comments

Comments
 (0)