diff --git a/test/unittest/mock/datashare_helper_mock.h b/test/unittest/mock/datashare_helper_mock.h
index 6722ec39e2af5a3ed346f39a994a06d7f57746ff..1676d8d2d48a8ee160df516e740855b6a12d8862 100644
--- a/test/unittest/mock/datashare_helper_mock.h
+++ b/test/unittest/mock/datashare_helper_mock.h
@@ -86,6 +86,9 @@ public:
     MOCK_METHOD(std::vector<OperationResult>, UnsubscribePublishedData, (const std::vector<std::string> &, int64_t));
     MOCK_METHOD(std::vector<OperationResult>, EnablePubSubs, (const std::vector<std::string> &, int64_t));
     MOCK_METHOD(std::vector<OperationResult>, DisablePubSubs, (const std::vector<std::string> &, int64_t));
+    MOCK_METHOD(std::shared_ptr<DataShare::DataShareResultSet>, QueryTimeout, (Uri &,
+        const DataShare::DataSharePredicates &, std::vector<std::string> &,
+        DataShare::DataShareOption &, DatashareBusinessError *));
 private:
     DataShareHelperMock() = default;
     ~DataShareHelperMock() override = default;