From 22927f1721e2469a0c8ace7db32d2dd9447c2f35 Mon Sep 17 00:00:00 2001
From: Jeam_wang <jeam.wangwei@huawei.com>
Date: Tue, 18 Mar 2025 09:44:30 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A1=A5=E5=85=85=E5=88=86=E6=94=AF=E8=A6=86?=
 =?UTF-8?q?=E7=9B=96=E7=94=A8=E4=BE=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Jeam_wang <jeam.wangwei@huawei.com>
---
 .../service/test/cloud_data_test.cpp          | 33 +++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/services/distributeddataservice/service/test/cloud_data_test.cpp b/services/distributeddataservice/service/test/cloud_data_test.cpp
index 107a81cf9..7670e3ae5 100644
--- a/services/distributeddataservice/service/test/cloud_data_test.cpp
+++ b/services/distributeddataservice/service/test/cloud_data_test.cpp
@@ -190,6 +190,21 @@ std::pair<int32_t, SchemaMeta> CloudServerMock::GetAppSchema(int32_t userId, con
     return { E_OK, CloudDataTest::schemaMeta_ };
 }
 
+class MockQuery : public GenQuery {
+public:
+    static constexpr uint64_t TYPE_ID = 1;
+
+    std::vector<std::string> GetTables() override
+    {
+        return { TEST_CLOUD_TABLE };
+    }
+    
+    bool IsEqual(uint64_t tid) override
+    {
+        return tid == TYPE_ID;
+    }
+ };
+
 std::shared_ptr<DBStoreMock> CloudDataTest::dbStoreMock_ = std::make_shared<DBStoreMock>();
 SchemaMeta CloudDataTest::schemaMeta_;
 StoreMetaData CloudDataTest::metaData_;
@@ -2703,5 +2718,23 @@ HWTEST_F(CloudDataTest, GetPrepareTraceId, TestSize.Level0)
     EXPECT_EQ(evt.GetUser(), EVT_USER);
     EXPECT_EQ(evt.GetPrepareTraceId(), TEST_TRACE_ID);
 }
+
+/**
+* @tc.name: GenerateQuery
+* @tc.desc: Test GenerateQuery function
+* @tc.type: FUNC
+* @tc.require:
+*/
+HWTEST_F(CloudDataTest, GenerateQuery, TestSize.Level0)
+{
+    int32_t user = 100;
+    CloudData::SyncManager::SyncInfo::Param param = { user, TEST_CLOUD_BUNDLE, TEST_CLOUD_STORE, {}, 0, "" };
+    CloudData::SyncManager::SyncInfo info(param);
+    EXPECT_EQ(info.tables_.size(), 1);
+    auto query = std::make_shared<MockQuery>();
+    info.SetQuery(query);
+    info.GenerateQuery(TEST_CLOUD_BUNDLE, {});
+    EXPECT_TRUE(info.query_ != nullptr);
+}
 } // namespace DistributedDataTest
 } // namespace OHOS::Test
\ No newline at end of file
-- 
Gitee