add mock file and gomock. start writing test for bucket

Author: changsongl

bucket/bucket_test.go

@@ -0,0 +1,32 @@
+package bucket
+
+import (
+	"errors"
+	lockmock "github.com/changsongl/delay-queue/test/mock/pkg/lock"
+	storemock "github.com/changsongl/delay-queue/test/mock/store"
+	"github.com/golang/mock/gomock"
+	"github.com/stretchr/testify/require"
+	"testing"
+)
+
+func TestBucketCreateJob(t *testing.T){
+	ctrl := gomock.NewController(t)
+	defer ctrl.Finish()
+
+	sm := storemock.NewMockStore(ctrl)
+	lockMk := lockmock.NewMockLocker(ctrl)
+
+	b := New(sm, 10, "test_bucket")
+
+	// case1: no error
+	sm.EXPECT().GetLock(gomock.All()).Return(lockMk).AnyTimes()
+	sm.EXPECT().CreateJobInBucket(gomock.Eq("test_bucket_1"), gomock.All(), gomock.All()).Return(nil)
+	err := b.CreateJob(nil, true)
+	require.NoError(t, err, "first create should no error")
+
+	// case2: has error
+	expectErr := errors.New("expect error")
+	sm.EXPECT().CreateJobInBucket(gomock.Eq("test_bucket_2"), gomock.All(), gomock.All()).Return(expectErr)
+	err = b.CreateJob(nil, true)
+	require.Equal(t, expectErr, err, "second create should be expect error")
+}

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/go-playground/validator/v10 v10.4.1 // indirect
 	github.com/go-redis/redis/v8 v8.4.4
 	github.com/go-redsync/redsync/v4 v4.0.4
+	github.com/golang/mock v1.5.0
 	github.com/gomodule/redigo v2.0.0+incompatible // indirect
 	github.com/leodido/go-urn v1.2.1 // indirect
 	github.com/prometheus/client_golang v1.9.0