diff --git a/test/mock/mock_network_search.h b/test/mock/mock_network_search.h
index 08357ce793032eaaf4e657edf1ccd5f40142e436..c503604717ce76a0a0de1552c1866a0cf2be762e 100644
--- a/test/mock/mock_network_search.h
+++ b/test/mock/mock_network_search.h
@@ -99,6 +99,7 @@ public:
     MOCK_METHOD2(IsGsm, int32_t(int32_t slotId, bool &isGsm));
     MOCK_METHOD2(IsCdma, int32_t(int32_t slotId, bool &isCdma));
     MOCK_METHOD1(UpdateOperatorName, int32_t(int32_t slotId));
+    MOCK_METHOD3(UpdateDeviceState, void(int32_t slotId, bool isEnterStrMode, bool isNeedUpdateNetworkState));
 };
 
 }  // namespace Telephony