diff --git a/interfaces/innerkits/native_cpp/test/unittest/screensourcetest/src/dscreen_source_handler_test.cpp b/interfaces/innerkits/native_cpp/test/unittest/screensourcetest/src/dscreen_source_handler_test.cpp
index ed8856f2327ef757f6e7097ccd14a47c48bfc951..3c5e50ef255f2204eb50ad7ed11e95f14ca58443 100644
--- a/interfaces/innerkits/native_cpp/test/unittest/screensourcetest/src/dscreen_source_handler_test.cpp
+++ b/interfaces/innerkits/native_cpp/test/unittest/screensourcetest/src/dscreen_source_handler_test.cpp
@@ -357,26 +357,14 @@ HWTEST_F(DScreenSourceHandlerTest, OnRemoteDied_001, TestSize.Level1)
 {
     sptr<ISystemAbilityManager> samgr =
             SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
+    sptr<IRemoteObject> remoteObject1 = nullptr;
+    wptr<IRemoteObject> remote1(remoteObject1);
+    DScreenSourceHandler::GetInstance().sourceSvrRecipient_->OnRemoteDied(remote1);
 
-    sptr<IRemoteObject> remoteObject = samgr->GetSystemAbility(DISTRIBUTED_HARDWARE_SCREEN_SOURCE_SA_ID);
-    wptr<IRemoteObject> remote(remoteObject);
-    DScreenSourceHandler::GetInstance().sourceSvrRecipient_->OnRemoteDied(remote);
+    sptr<IRemoteObject> remoteObject2 = samgr->GetSystemAbility(DISTRIBUTED_HARDWARE_SCREEN_SOURCE_SA_ID);
+    wptr<IRemoteObject> remote2(remoteObject2);
+    DScreenSourceHandler::GetInstance().sourceSvrRecipient_->OnRemoteDied(remote2);
     EXPECT_EQ(nullptr, DScreenSourceHandler::GetInstance().dScreenSourceProxy_);
 }
-
-/**
- * @tc.name: OnRemoteDied_002
- * @tc.desc: Verify the OnRemoteDied function.
- * @tc.type: FUNC
- * @tc.require: Issue Number
- */
-HWTEST_F(DScreenSourceHandlerTest, OnRemoteDied_002, TestSize.Level1)
-{
-    sptr<ISystemAbilityManager> samgr = SystemAbilityManagerClient::GetInstance().GetSystemAbilityManager();
-    sptr<IRemoteObject> remoteObject = nullptr;
-    wptr<IRemoteObject> remote(remoteObject);
-    DScreenSourceHandler::GetInstance().sourceSvrRecipient_->OnRemoteDied(remote);
-    EXPECT_EQ(remote, remote);
-}
 }  // namespace DistributedHardware
 }  // namespace OHOS
diff --git a/services/screenclient/test/unittest/src/screen_client_window_adapter_test.cpp b/services/screenclient/test/unittest/src/screen_client_window_adapter_test.cpp
index 1f24b82e975f71b35bbdda5099f1e4a575a3b924..e13587782a9585e16d4f7dc9597c1af9b7939cbd 100644
--- a/services/screenclient/test/unittest/src/screen_client_window_adapter_test.cpp
+++ b/services/screenclient/test/unittest/src/screen_client_window_adapter_test.cpp
@@ -254,6 +254,10 @@ HWTEST_F(ScreenClientWindowAdapterTest, ScreenClientInputEventListener_OnInputEv
     std::shared_ptr<MMI::PointerEvent> pointerEvent = nullptr;
     std::shared_ptr<ScreenClientInputEventListener> listener = std::make_shared<ScreenClientInputEventListener>();
     EXPECT_FALSE(listener->OnInputEvent(pointerEvent));
+
+    pointerEvent = MMI::PointerEvent::Create();
+    EXPECT_TRUE(pointerEvent != nullptr);
+    EXPECT_TRUE(listener->OnInputEvent(pointerEvent));
 }
 
 /**
@@ -267,6 +271,10 @@ HWTEST_F(ScreenClientWindowAdapterTest, ScreenClientInputEventListener_OnInputEv
     std::shared_ptr<MMI::KeyEvent> keyEvent = nullptr;
     std::shared_ptr<ScreenClientInputEventListener> listener = std::make_shared<ScreenClientInputEventListener>();
     EXPECT_FALSE(listener->OnInputEvent(keyEvent));
+
+    keyEvent = MMI::KeyEvent::Create();
+    EXPECT_TRUE(keyEvent != nullptr);
+    EXPECT_TRUE(listener->OnInputEvent(keyEvent));
 }
 
 /**
@@ -280,6 +288,10 @@ HWTEST_F(ScreenClientWindowAdapterTest, ScreenClientInputEventListener_OnInputEv
     std::shared_ptr<MMI::AxisEvent> axisEvent = nullptr;
     std::shared_ptr<ScreenClientInputEventListener> listener = std::make_shared<ScreenClientInputEventListener>();
     EXPECT_FALSE(listener->OnInputEvent(axisEvent));
+
+    axisEvent = MMI::AxisEvent::Create();
+    EXPECT_TRUE(axisEvent != nullptr);
+    EXPECT_TRUE(listener->OnInputEvent(axisEvent));
 }
 } // DistributedHardware
 } // OHOS
\ No newline at end of file
diff --git a/services/screentransport/test/unittest/screendatachannel/src/screen_data_channel_impl_test.cpp b/services/screentransport/test/unittest/screendatachannel/src/screen_data_channel_impl_test.cpp
index 7f4d6ab2cedb2157e9a15ab5459c42b0f1f3052b..109e9c8821105e9f27e6565a526d0f053fcfedd1 100644
--- a/services/screentransport/test/unittest/screendatachannel/src/screen_data_channel_impl_test.cpp
+++ b/services/screentransport/test/unittest/screendatachannel/src/screen_data_channel_impl_test.cpp
@@ -396,5 +396,21 @@ HWTEST_F(ScreenDataChannelImplTest, JsonToDirtyJson_001, TestSize.Level1)
     EXPECT_EQ(nullptr, dataChannelImpl_->channelListener_.lock());
 }
 
+/**
+ * @tc.name: SetJpegSessionFlag_001
+ * @tc.desc: Verify the SetJpegSessionFlag function.
+ * @tc.type: FUNC
+ * @tc.require: Issue Number
+ */
+HWTEST_F(ScreenDataChannelImplTest, SetJpegSessionFlag_001, TestSize.Level1)
+{
+    bool flag = true;
+    dataChannelImpl_->SetJpegSessionFlag(flag);
+    EXPECT_EQ(dataChannelImpl_->jpegSessionFlag_, flag);
+
+    flag = false;
+    dataChannelImpl_->SetJpegSessionFlag(flag);
+    EXPECT_EQ(dataChannelImpl_->jpegSessionFlag_, flag);
+}
 } // DistributedHardware
 } // OHOS
\ No newline at end of file
diff --git a/services/screentransport/test/unittest/screensinkprocessor/src/image_sink_processor_test.cpp b/services/screentransport/test/unittest/screensinkprocessor/src/image_sink_processor_test.cpp
index 85ffa88213f702a79bb8faaee81e66bf235e4d44..1953f6484969625b34342475ce7e9809ce15745c 100644
--- a/services/screentransport/test/unittest/screensinkprocessor/src/image_sink_processor_test.cpp
+++ b/services/screentransport/test/unittest/screensinkprocessor/src/image_sink_processor_test.cpp
@@ -189,7 +189,9 @@ HWTEST_F(ImageSinkProcessorTest, set_image_surface_test_002, TestSize.Level1)
  */
 HWTEST_F(ImageSinkProcessorTest, processimage_test_001, TestSize.Level1)
 {
-    std::shared_ptr<DataBuffer> data = std::make_shared<DataBuffer>(DATA_LEN);
+    std::shared_ptr<DataBuffer> data = nullptr;
+    EXPECT_EQ(ERR_DH_SCREEN_TRANS_NULL_VALUE, processor_->ProcessImage(data));
+    data = std::make_shared<DataBuffer>(DATA_LEN);
     EXPECT_EQ(DH_SUCCESS, processor_->ProcessImage(data));
 }