test: add unit test for /arana/pkg/sequence/... package

arana-db · Sep 16, 2023 · 58cf3ba · 58cf3ba
1 parent 4ebc401
commit 58cf3ba
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 136 deletions.
diff --git a/pkg/runtime/runtime.go b/pkg/runtime/runtime.go
@@ -15,7 +15,7 @@
  * limitations under the License.
  */
 
-//go:generate mockgen -destination=../../testdata/mock_rt_runtime.go -package=testdata . Runtime
+//go:generate mockgen -destination=./mock_runtime.go -package=runtime . Runtime
 package runtime
 
 import (

diff --git a/pkg/sequence/group/group_test.go b/pkg/sequence/group/group_test.go
@@ -77,7 +77,7 @@ func Test_groupSequence_Acquire(t *testing.T) {
 	mockTx.EXPECT().Rollback(gomock.Any()).Return(mockRes, uint16(0), nil).AnyTimes()
 
 	// build mock runtime
-	mockRt := testdata.NewMockRuntime(ctrl)
+	mockRt := runtime.NewMockRuntime(ctrl)
 	mockRt.EXPECT().Begin(gomock.Any()).Return(mockTx, nil).AnyTimes()
 
 	runtime.Register(tenant, schema, mockRt)

diff --git a/pkg/sequence/snowflake/snowflake_test.go b/pkg/sequence/snowflake/snowflake_test.go
@@ -33,6 +33,7 @@ import (
 
 import (
 	"github.com/arana-db/arana/pkg/proto"
+	"github.com/arana-db/arana/pkg/runtime"
 	"github.com/arana-db/arana/testdata"
 )
 
@@ -68,7 +69,7 @@ func Test_snowflakeSequence_Acquire(t *testing.T) {
 	mockTx.EXPECT().Rollback(gomock.Any()).Return(mockRes, uint16(0), nil).AnyTimes()
 
 	// build mock runtime
-	mockRt := testdata.NewMockRuntime(ctrl)
+	mockRt := runtime.NewMockRuntime(ctrl)
 	mockRt.EXPECT().Begin(gomock.Any()).Return(mockTx, nil).AnyTimes()
 
 	ctx := context.WithValue(context.Background(), proto.RuntimeCtxKey{}, mockRt)

diff --git a/testdata/mock_rt_runtime.go b/testdata/mock_rt_runtime.go