diff --git a/services/sink/sinkmanager/test/sinkmanagerunittest/BUILD.gn b/services/sink/sinkmanager/test/sinkmanagerunittest/BUILD.gn
index ce95bae..7a71503 100755
--- a/services/sink/sinkmanager/test/sinkmanagerunittest/BUILD.gn
+++ b/services/sink/sinkmanager/test/sinkmanagerunittest/BUILD.gn
@@ -62,6 +62,7 @@ ohos_unittest("distributed_input_sinkmanager_test") {
   sources = [
     "${common_path}/include/input_hub.cpp",
     "${common_path}/include/white_list_util.cpp",
+    "${common_path}/test/mock/socket_mock.cpp",
     "${common_path}/test/mock/softbus_bus_center_mock.cpp",
     "${ipc_path}/src/distributed_input_sink_stub.cpp",
     "${ipc_path}/src/get_sink_screen_infos_call_back_stub.cpp",
diff --git a/services/sink/transport/test/sinktransunittest/BUILD.gn b/services/sink/transport/test/sinktransunittest/BUILD.gn
index 2886428..a0972ee 100755
--- a/services/sink/transport/test/sinktransunittest/BUILD.gn
+++ b/services/sink/transport/test/sinktransunittest/BUILD.gn
@@ -54,6 +54,7 @@ ohos_unittest("distributed_input_sinktrans_test") {
   ]
 
   sources = [
+    "${common_path}/test/mock/socket_mock.cpp",
     "${common_path}/test/mock/softbus_bus_center_mock.cpp",
     "${distributedinput_path}/services/transportbase/src/distributed_input_transport_base.cpp",
     "${services_sink_path}/sinkmanager/src/distributed_input_sink_manager.cpp",
diff --git a/services/source/inputinject/test/sourceinjectunittest/BUILD.gn b/services/source/inputinject/test/sourceinjectunittest/BUILD.gn
index 335e137..e9af480 100755
--- a/services/source/inputinject/test/sourceinjectunittest/BUILD.gn
+++ b/services/source/inputinject/test/sourceinjectunittest/BUILD.gn
@@ -50,6 +50,7 @@ ohos_unittest("distributed_input_inner_sourceinject_test") {
 
   sources = [
     "${common_path}/include/input_hub.cpp",
+    "${common_path}/test/mock/socket_mock.cpp",
     "${common_path}/test/mock/softbus_bus_center_mock.cpp",
     "${distributedinput_path}/inputdevicehandler/src/distributed_input_handler.cpp",
     "${services_source_path}/inputinject/src/distributed_input_inject.cpp",
diff --git a/services/source/sourcemanager/test/sourcemanagerunittest/BUILD.gn b/services/source/sourcemanager/test/sourcemanagerunittest/BUILD.gn
index eec55eb..1b5e06b 100755
--- a/services/source/sourcemanager/test/sourcemanagerunittest/BUILD.gn
+++ b/services/source/sourcemanager/test/sourcemanagerunittest/BUILD.gn
@@ -56,6 +56,7 @@ ohos_unittest("distributed_input_sourcemanager_test") {
 
   sources = [
     "${common_path}/include/input_hub.cpp",
+    "${common_path}/test/mock/socket_mock.cpp",
     "${common_path}/test/mock/softbus_bus_center_mock.cpp",
     "${distributedinput_path}/services/transportbase/src/distributed_input_transport_base.cpp",
     "${distributedinput_path}/utils/src/dinput_context.cpp",
diff --git a/services/source/transport/test/sourcetransunittest/BUILD.gn b/services/source/transport/test/sourcetransunittest/BUILD.gn
index 6efbbcd..d5ffdbe 100755
--- a/services/source/transport/test/sourcetransunittest/BUILD.gn
+++ b/services/source/transport/test/sourcetransunittest/BUILD.gn
@@ -52,6 +52,7 @@ ohos_unittest("distributed_input_sourcetrans_test") {
   ]
 
   sources = [
+    "${common_path}/test/mock/socket_mock.cpp",
     "${common_path}/test/mock/softbus_bus_center_mock.cpp",
     "${distributedinput_path}/services/transportbase/src/distributed_input_transport_base.cpp",
     "${services_source_path}/transport/src/distributed_input_source_transport.cpp",
diff --git a/utils/test/unittest/BUILD.gn b/utils/test/unittest/BUILD.gn
index 6322f5f..ae01e55 100644
--- a/utils/test/unittest/BUILD.gn
+++ b/utils/test/unittest/BUILD.gn
@@ -43,6 +43,7 @@ ohos_unittest("distributed_input_utils_test") {
   ]
 
   sources = [
+    "${common_path}/test/mock/socket_mock.cpp",
     "${common_path}/test/mock/softbus_bus_center_mock.cpp",
     "${distributedinput_path}/utils/src/dinput_context.cpp",
     "${distributedinput_path}/utils/src/dinput_log.cpp",