diff --git a/internal/streamingcoord/client/assignment/assignment_test.go b/internal/streamingcoord/client/assignment/assignment_test.go index 9e0be7a9fc339..0735293c013f7 100644 --- a/internal/streamingcoord/client/assignment/assignment_test.go +++ b/internal/streamingcoord/client/assignment/assignment_test.go @@ -12,7 +12,7 @@ import ( "github.com/milvus-io/milvus/internal/mocks/util/streamingutil/service/mock_lazygrpc" "github.com/milvus-io/milvus/internal/util/streamingutil/status" - "github.com/milvus-io/milvus/pkg/mocks/streaming/proto/mock_streamingpb" + "github.com/milvus-io/milvus/pkg/mocks/proto/mock_streamingpb" "github.com/milvus-io/milvus/pkg/proto/streamingpb" "github.com/milvus-io/milvus/pkg/streaming/util/types" "github.com/milvus-io/milvus/pkg/util/typeutil" diff --git a/internal/streamingcoord/server/service/discover/discover_server_test.go b/internal/streamingcoord/server/service/discover/discover_server_test.go index c4c4a94dde711..992ef29545431 100644 --- a/internal/streamingcoord/server/service/discover/discover_server_test.go +++ b/internal/streamingcoord/server/service/discover/discover_server_test.go @@ -9,7 +9,7 @@ import ( "github.com/milvus-io/milvus/internal/mocks/streamingcoord/server/mock_balancer" "github.com/milvus-io/milvus/internal/streamingcoord/server/resource" - "github.com/milvus-io/milvus/pkg/mocks/streaming/proto/mock_streamingpb" + "github.com/milvus-io/milvus/pkg/mocks/proto/mock_streamingpb" "github.com/milvus-io/milvus/pkg/proto/streamingpb" "github.com/milvus-io/milvus/pkg/streaming/util/types" "github.com/milvus-io/milvus/pkg/util/typeutil" diff --git a/internal/streamingnode/client/handler/consumer/consumer_test.go b/internal/streamingnode/client/handler/consumer/consumer_test.go index 86ce942d74120..a0bebccf96db3 100644 --- a/internal/streamingnode/client/handler/consumer/consumer_test.go +++ b/internal/streamingnode/client/handler/consumer/consumer_test.go @@ -10,7 +10,7 @@ import ( "github.com/stretchr/testify/mock" "github.com/milvus-io/milvus-proto/go-api/v2/msgpb" - "github.com/milvus-io/milvus/pkg/mocks/streaming/proto/mock_streamingpb" + "github.com/milvus-io/milvus/pkg/mocks/proto/mock_streamingpb" "github.com/milvus-io/milvus/pkg/proto/messagespb" "github.com/milvus-io/milvus/pkg/proto/streamingpb" "github.com/milvus-io/milvus/pkg/streaming/util/message" diff --git a/internal/streamingnode/client/handler/handler_client_test.go b/internal/streamingnode/client/handler/handler_client_test.go index b6afbbe1fc2f6..584a34e135632 100644 --- a/internal/streamingnode/client/handler/handler_client_test.go +++ b/internal/streamingnode/client/handler/handler_client_test.go @@ -16,7 +16,7 @@ import ( "github.com/milvus-io/milvus/internal/streamingnode/client/handler/consumer" "github.com/milvus-io/milvus/internal/streamingnode/client/handler/producer" "github.com/milvus-io/milvus/internal/util/streamingutil/status" - "github.com/milvus-io/milvus/pkg/mocks/streaming/proto/mock_streamingpb" + "github.com/milvus-io/milvus/pkg/mocks/proto/mock_streamingpb" "github.com/milvus-io/milvus/pkg/mocks/streaming/util/mock_types" "github.com/milvus-io/milvus/pkg/proto/streamingpb" "github.com/milvus-io/milvus/pkg/streaming/util/message" diff --git a/internal/streamingnode/client/handler/producer/producer_test.go b/internal/streamingnode/client/handler/producer/producer_test.go index 54e9fa394bfdb..c65c8d963ee1c 100644 --- a/internal/streamingnode/client/handler/producer/producer_test.go +++ b/internal/streamingnode/client/handler/producer/producer_test.go @@ -9,7 +9,7 @@ import ( "github.com/stretchr/testify/assert" "github.com/stretchr/testify/mock" - "github.com/milvus-io/milvus/pkg/mocks/streaming/proto/mock_streamingpb" + "github.com/milvus-io/milvus/pkg/mocks/proto/mock_streamingpb" "github.com/milvus-io/milvus/pkg/proto/messagespb" "github.com/milvus-io/milvus/pkg/proto/streamingpb" "github.com/milvus-io/milvus/pkg/streaming/util/message" diff --git a/internal/streamingnode/client/manager/manager_test.go b/internal/streamingnode/client/manager/manager_test.go index fb80a4597842d..d31f849ac17e9 100644 --- a/internal/streamingnode/client/manager/manager_test.go +++ b/internal/streamingnode/client/manager/manager_test.go @@ -17,7 +17,7 @@ import ( "github.com/milvus-io/milvus/internal/util/streamingutil/service/attributes" "github.com/milvus-io/milvus/internal/util/streamingutil/service/contextutil" "github.com/milvus-io/milvus/internal/util/streamingutil/service/discoverer" - "github.com/milvus-io/milvus/pkg/mocks/streaming/proto/mock_streamingpb" + "github.com/milvus-io/milvus/pkg/mocks/proto/mock_streamingpb" "github.com/milvus-io/milvus/pkg/proto/streamingpb" "github.com/milvus-io/milvus/pkg/streaming/util/types" "github.com/milvus-io/milvus/pkg/util/etcd" diff --git a/internal/streamingnode/server/service/handler/consumer/consume_server_test.go b/internal/streamingnode/server/service/handler/consumer/consume_server_test.go index e61b66dcb8a4c..2c6ab7affea40 100644 --- a/internal/streamingnode/server/service/handler/consumer/consume_server_test.go +++ b/internal/streamingnode/server/service/handler/consumer/consume_server_test.go @@ -17,7 +17,7 @@ import ( "github.com/milvus-io/milvus/internal/util/streamingutil/service/contextutil" "github.com/milvus-io/milvus/internal/util/streamingutil/status" "github.com/milvus-io/milvus/pkg/log" - "github.com/milvus-io/milvus/pkg/mocks/streaming/proto/mock_streamingpb" + "github.com/milvus-io/milvus/pkg/mocks/proto/mock_streamingpb" "github.com/milvus-io/milvus/pkg/mocks/streaming/util/mock_message" "github.com/milvus-io/milvus/pkg/proto/streamingpb" "github.com/milvus-io/milvus/pkg/streaming/util/message" diff --git a/internal/streamingnode/server/service/handler/producer/produce_server_test.go b/internal/streamingnode/server/service/handler/producer/produce_server_test.go index d517de56f7a52..0da35538c52ed 100644 --- a/internal/streamingnode/server/service/handler/producer/produce_server_test.go +++ b/internal/streamingnode/server/service/handler/producer/produce_server_test.go @@ -20,7 +20,7 @@ import ( "github.com/milvus-io/milvus/internal/streamingnode/server/walmanager" "github.com/milvus-io/milvus/internal/util/streamingutil/service/contextutil" "github.com/milvus-io/milvus/pkg/log" - "github.com/milvus-io/milvus/pkg/mocks/streaming/proto/mock_streamingpb" + "github.com/milvus-io/milvus/pkg/mocks/proto/mock_streamingpb" "github.com/milvus-io/milvus/pkg/proto/messagespb" "github.com/milvus-io/milvus/pkg/proto/streamingpb" "github.com/milvus-io/milvus/pkg/streaming/util/message" diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go similarity index 100% rename from pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go rename to pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentServiceClient.go diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go similarity index 100% rename from pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go rename to pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverClient.go diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go similarity index 100% rename from pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go rename to pkg/mocks/proto/mock_streamingpb/mock_StreamingCoordAssignmentService_AssignmentDiscoverServer.go diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go similarity index 100% rename from pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go rename to pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerServiceClient.go diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go similarity index 100% rename from pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go rename to pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeClient.go diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go similarity index 100% rename from pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go rename to pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ConsumeServer.go diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go similarity index 100% rename from pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go rename to pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceClient.go diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go similarity index 100% rename from pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go rename to pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeHandlerService_ProduceServer.go diff --git a/pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go b/pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go similarity index 100% rename from pkg/mocks/streaming/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go rename to pkg/mocks/proto/mock_streamingpb/mock_StreamingNodeManagerServiceClient.go