diff --git a/Makefile b/Makefile
index d1634e01ce..a86dc825c3 100644
--- a/Makefile
+++ b/Makefile
@@ -463,23 +463,13 @@ generate-mockery-proxy: getdeps
 	$(INSTALL_PATH)/mockery --name=channelsMgr --dir=$(PWD)/internal/proxy --output=$(PWD)/internal/proxy --filename=mock_channels_manager.go --structname=MockChannelsMgr --with-expecter --outpkg=proxy --inpackage
 
 generate-mockery-querycoord: getdeps
-	$(INSTALL_PATH)/mockery --name=QueryNodeServer --dir=$(PWD)/internal/proto/querypb/ --output=$(PWD)/internal/querycoordv2/mocks --filename=mock_querynode.go --with-expecter --structname=MockQueryNodeServer
-	$(INSTALL_PATH)/mockery --name=Broker --dir=$(PWD)/internal/querycoordv2/meta --output=$(PWD)/internal/querycoordv2/meta --filename=mock_broker.go --with-expecter --structname=MockBroker --outpkg=meta
-	$(INSTALL_PATH)/mockery --name=TargetManagerInterface --dir=$(PWD)/internal/querycoordv2/meta --output=$(PWD)/internal/querycoordv2/meta --filename=mock_target_manager.go --with-expecter --structname=MockTargetManager --inpackage
-	$(INSTALL_PATH)/mockery --name=Scheduler --dir=$(PWD)/internal/querycoordv2/task --output=$(PWD)/internal/querycoordv2/task --filename=mock_scheduler.go --with-expecter --structname=MockScheduler --outpkg=task --inpackage
-	$(INSTALL_PATH)/mockery --name=Cluster --dir=$(PWD)/internal/querycoordv2/session --output=$(PWD)/internal/querycoordv2/session --filename=mock_cluster.go --with-expecter --structname=MockCluster --outpkg=session --inpackage
-	$(INSTALL_PATH)/mockery --name=Balance --dir=$(PWD)/internal/querycoordv2/balance --output=$(PWD)/internal/querycoordv2/balance --filename=mock_balancer.go --with-expecter --structname=MockBalancer --outpkg=balance --inpackage
-	$(INSTALL_PATH)/mockery --name=Controller --dir=$(PWD)/internal/querycoordv2/dist --output=$(PWD)/internal/querycoordv2/dist --filename=mock_controller.go --with-expecter --structname=MockController --outpkg=dist --inpackage
+	$(INSTALL_PATH)/mockery --config $(PWD)/internal/querycoordv2/.mockery.yaml
 
-generate-mockery-querynode: getdeps build-cpp
-	@source $(PWD)/scripts/setenv.sh # setup PKG_CONFIG_PATH
-	$(INSTALL_PATH)/mockery --name=Manager --dir=$(PWD)/internal/querynodev2/cluster --output=$(PWD)/internal/querynodev2/cluster --filename=mock_manager.go --with-expecter --outpkg=cluster --structname=MockManager --inpackage
-	$(INSTALL_PATH)/mockery --name=SegmentManager --dir=$(PWD)/internal/querynodev2/segments --output=$(PWD)/internal/querynodev2/segments --filename=mock_segment_manager.go --with-expecter --outpkg=segments --structname=MockSegmentManager --inpackage
-	$(INSTALL_PATH)/mockery --name=CollectionManager --dir=$(PWD)/internal/querynodev2/segments --output=$(PWD)/internal/querynodev2/segments --filename=mock_collection_manager.go --with-expecter --outpkg=segments --structname=MockCollectionManager --inpackage
-	$(INSTALL_PATH)/mockery --name=Loader --dir=$(PWD)/internal/querynodev2/segments --output=$(PWD)/internal/querynodev2/segments --filename=mock_loader.go --with-expecter --outpkg=segments --structname=MockLoader --inpackage
-	$(INSTALL_PATH)/mockery --name=Segment --dir=$(PWD)/internal/querynodev2/segments --output=$(PWD)/internal/querynodev2/segments --filename=mock_segment.go --with-expecter --outpkg=segments --structname=MockSegment  --inpackage
-	$(INSTALL_PATH)/mockery --name=Worker --dir=$(PWD)/internal/querynodev2/cluster --output=$(PWD)/internal/querynodev2/cluster --filename=mock_worker.go --with-expecter --outpkg=worker --structname=MockWorker --inpackage
-	$(INSTALL_PATH)/mockery --name=ShardDelegator --dir=$(PWD)/internal/querynodev2/delegator/ --output=$(PWD)/internal/querynodev2/delegator/ --filename=mock_delegator.go --with-expecter --outpkg=delegator --structname=MockShardDelegator --inpackage
+generate-mockery-querynode-without-cpp:
+	@source $(PWD)/scripts/setenv.sh
+	$(INSTALL_PATH)/mockery --config $(PWD)/internal/querynodev2/.mockery.yaml
+
+generate-mockery-querynode: build-cpp generate-mockery-querynode-without-cpp
 
 generate-mockery-datacoord: getdeps
 	$(INSTALL_PATH)/mockery --name=compactionPlanContext --dir=internal/datacoord --filename=mock_compaction_plan_context.go --output=internal/datacoord  --structname=MockCompactionPlanContext --with-expecter --inpackage
diff --git a/internal/querycoordv2/.mockery.yaml b/internal/querycoordv2/.mockery.yaml
new file mode 100644
index 0000000000..11302e8643
--- /dev/null
+++ b/internal/querycoordv2/.mockery.yaml
@@ -0,0 +1,39 @@
+quiet: False
+with-expecter: True
+inpackage: True
+filename: "mock_{{.InterfaceNameSnake}}.go"
+mockname: "Mock{{.InterfaceName}}"
+outpkg: "{{.PackageName}}"
+dir: "{{.InterfaceDir}}"
+packages:
+  github.com/milvus-io/milvus/internal/querycoordv2/balance:
+    interfaces:
+      Balance:
+        config:
+          filename: "mock_balancer.go"
+          mockname: "MockBalancer"
+  github.com/milvus-io/milvus/internal/querycoordv2/dist:
+    interfaces:
+      Controller:
+  github.com/milvus-io/milvus/internal/querycoordv2/meta:
+    interfaces:
+      Broker:
+      TargetManagerInterface:
+        config:
+          filename: "mock_target_manager.go"
+          mockname: "MockTargetManager"
+  github.com/milvus-io/milvus/internal/querycoordv2/session:
+    interfaces:
+      Cluster:
+  github.com/milvus-io/milvus/internal/querycoordv2/task:
+    interfaces:
+      Scheduler:
+  github.com/milvus-io/milvus/internal/proto/querypb:
+    interfaces:
+      QueryNodeServer:
+        config:
+          dir: "internal/querycoordv2/mocks"
+          inpackage: False
+          outpkg: "mocks"
+          filename: "mock_querynode.go"
+          mockname: "MockQueryNodeServer"
diff --git a/internal/querynodev2/.mockery.yaml b/internal/querynodev2/.mockery.yaml
new file mode 100644
index 0000000000..10fb65e54a
--- /dev/null
+++ b/internal/querynodev2/.mockery.yaml
@@ -0,0 +1,23 @@
+quiet: False
+with-expecter: True
+inpackage: True
+filename: "mock_{{.InterfaceNameSnake}}.go"
+mockname: "Mock{{.InterfaceName}}"
+outpkg: "{{.PackageName}}"
+dir: "{{.InterfaceDir}}"
+packages:
+  github.com/milvus-io/milvus/internal/querynodev2/cluster:
+    interfaces:
+      Manager:
+      Worker:
+  github.com/milvus-io/milvus/internal/querynodev2/segments:
+    interfaces:
+      SegmentManager:
+      CollectionManager:
+      Loader:
+      Segment:
+  github.com/milvus-io/milvus/internal/querynodev2/delegator:
+    interfaces:
+      ShardDelegator:
+        config:
+          filename: "mock_delegator.go"