From d0fb02f0a3b9761b31f5126b1ca77b677ca36acf Mon Sep 17 00:00:00 2001
From: MisterE <smart_e@126.com>
Date: Tue, 19 Nov 2024 20:00:15 +0800
Subject: [PATCH] codecheck

Signed-off-by: MisterE <smart_e@126.com>
---
 services/dtbschedmgr/src/continue/dsched_continue.cpp | 11 +++++------
 .../test/unittest/continue/dsched_continue_test.cpp   |  5 +++++
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/services/dtbschedmgr/src/continue/dsched_continue.cpp b/services/dtbschedmgr/src/continue/dsched_continue.cpp
index 181fee94..c6aa2ee0 100644
--- a/services/dtbschedmgr/src/continue/dsched_continue.cpp
+++ b/services/dtbschedmgr/src/continue/dsched_continue.cpp
@@ -434,7 +434,7 @@ int32_t DSchedContinue::UpdateElementInfo(std::shared_ptr<DSchedContinueDataCmd>
            srcContinueType.c_str());
     DmsBundleInfo distributedBundleInfo;
     if (!DmsBmStorage::GetInstance()->GetDistributedBundleInfo(
-            cmd->dstDeviceId_, cmd->dstBundleName_, distributedBundleInfo)) {
+        cmd->dstDeviceId_, cmd->dstBundleName_, distributedBundleInfo)) {
         HILOGE("UpdateElementInfo can not found bundle info for bundle name: %{public}s",
                cmd->dstBundleName_.c_str());
         return CAN_NOT_FOUND_MODULE_ERR;
@@ -1008,11 +1008,6 @@ int32_t DSchedContinue::ExecuteContinueData(std::shared_ptr<DSchedContinueDataCm
         return INVALID_PARAMETERS_ERR;
     }
 
-    if (UpdateElementInfo(cmd) != ERR_OK) {
-        HILOGE("ExecuteContinueData UpdateElementInfo failed.");
-        return CAN_NOT_FOUND_MODULE_ERR;
-    }
-
     DurationDumperBeforeStartAbility(cmd);
 
     std::string localDeviceId;
@@ -1022,6 +1017,10 @@ int32_t DSchedContinue::ExecuteContinueData(std::shared_ptr<DSchedContinueDataCm
         HILOGE("check deviceId failed");
         return INVALID_REMOTE_PARAMETERS_ERR;
     }
+    if (UpdateElementInfo(cmd) != ERR_OK) {
+        HILOGE("ExecuteContinueData UpdateElementInfo failed.");
+        return CAN_NOT_FOUND_MODULE_ERR;
+    }
     int32_t ret = CheckStartPermission(cmd);
     if (ret != ERR_OK) {
         HILOGE("ExecuteContinueData CheckTargetPermission failed!");
diff --git a/services/dtbschedmgr/test/unittest/continue/dsched_continue_test.cpp b/services/dtbschedmgr/test/unittest/continue/dsched_continue_test.cpp
index 6cbb0cbc..ae0e8300 100644
--- a/services/dtbschedmgr/test/unittest/continue/dsched_continue_test.cpp
+++ b/services/dtbschedmgr/test/unittest/continue/dsched_continue_test.cpp
@@ -653,6 +653,11 @@ HWTEST_F(DSchedContinueTest, DSchedContinueTest_0017_2, TestSize.Level0)
     cmd->continueType_ = CONTINUE_TYPE2;
     cmd->want_.SetElementName("", BUNDLEMAME_1, ABILITY_NAME_DIFF_AS_CONTINUE_TYPE, MODULE_NAME3);
     ret = conti->UpdateElementInfo(cmd);
+    EXPECT_EQ(ret, ERR_OK);
+    // has continueType, no same module
+    cmd->continueType_ = CONTINUE_TYPE2;
+    cmd->want_.SetElementName("", BUNDLEMAME_1, CONTINUE_TYPE1, MODULE_NAME3);
+    ret = conti->UpdateElementInfo(cmd);
     EXPECT_EQ(ret, CAN_NOT_FOUND_MODULE_ERR);
     DTEST_LOG << "DSchedContinueTest DSchedContinueTest_0017_2 end ret:" << ret << std::endl;
     usleep(WAITTIME);