From f9ef13735b0f6f7ee1af0054a2167b55b9173266 Mon Sep 17 00:00:00 2001
From: zhanglei <zhanglei730@huawei.com>
Date: Fri, 15 Sep 2023 09:23:47 +0800
Subject: [PATCH] UT add

---
 .../src/screen_data_channel_impl_test.cpp     | 109 ++++++++++++++++++
 1 file changed, 109 insertions(+)

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 6a9754ce..ce15c7d4 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
@@ -66,6 +66,25 @@ HWTEST_F(ScreenDataChannelImplTest, CreateSession_002, TestSize.Level1)
     EXPECT_EQ(-1, ret);
 }
 
+
+/**
+ * @tc.name: ReleaseSession_001
+ * @tc.desc: Verify the CreateSession function.
+ * @tc.type: FUNC
+ * @tc.require: Issue Number
+ */
+
+HWTEST_F(ScreenDataChannelImplTest, ReleaseSession_001, TestSize.Level1)
+{
+    std::shared_ptr<IScreenChannelListener> listener = std::make_shared<MockIScreenChannelListener>();
+    dataChannelImpl_->SetJpegSessionFlag(true);
+    EXPECT_EQ(ERR_DH_SCREEN_TRANS_ILLEGAL_OPERATION, dataChannelImpl_->ReleaseSession());
+    
+}
+
+
+
+
 /**
  * @tc.name: OpenSession_001
  * @tc.desc: Verify the OpenSession function.
@@ -79,6 +98,27 @@ HWTEST_F(ScreenDataChannelImplTest, OpenSession_001, TestSize.Level1)
     EXPECT_EQ(ERR_DH_SCREEN_TRANS_ERROR, ret);
 }
 
+
+
+/**
+ * @tc.name: OpenSession_002
+ * @tc.desc: Verify the OpenSession function.
+ * @tc.type: FUNC
+ * @tc.require: Issue Number
+ */
+HWTEST_F(ScreenDataChannelImplTest, OpenSession_002, TestSize.Level1)
+{
+    dataChannelImpl_->SetJpegSessionFlag(true);
+
+    int32_t ret = dataChannelImpl_->OpenSession();
+
+    EXPECT_EQ(ERR_DH_SCREEN_TRANS_ERROR, ret);
+}
+
+
+
+
+
 /**
  * @tc.name: SendFullData_001
  * @tc.desc: Verify the SendFullData function.
@@ -184,6 +224,34 @@ HWTEST_F(ScreenDataChannelImplTest, close_session_test_002, TestSize.Level1)
     EXPECT_EQ(ERR_DH_SCREEN_TRANS_SESSION_NOT_OPEN, dataChannelImpl_->CloseSession());
 }
 
+/**
+ * @tc.name: close_session_test_003
+ * @tc.desc: Verify the CloseSession function.
+ * @tc.type: FUNC
+ * @tc.require: Issue Number
+ */
+HWTEST_F(ScreenDataChannelImplTest, close_session_test_003, TestSize.Level1)
+{
+    dataChannelImpl_->channelListener_ = std::make_shared<MockIScreenChannelListener>();
+
+    StreamData ext = {0};
+    StreamFrameInfo *param = nullptr;
+
+    int32_t sessionId = 0;
+    StreamData data;
+    data.buf = new char[DATA_LEN];
+    data.bufLen = DATA_LEN;
+
+    auto dataBuffer = std::make_shared<DataBuffer>(10);
+    dataChannelImpl_->ProcessDirtyData(&data, dataBuffer, &ext);
+    dataChannelImpl_->OnStreamReceived(sessionId, &data, &ext, param);
+    delete[] data.buf;
+    dataChannelImpl_->sessionId_ = 0;
+    EXPECT_EQ(ERR_DH_SCREEN_TRANS_SESSION_NOT_OPEN, dataChannelImpl_->CloseSession());
+}
+
+
+
 /**
  * @tc.name: send_data_test_001
  * @tc.desc: Verify the SendData function.
@@ -237,6 +305,47 @@ HWTEST_F(ScreenDataChannelImplTest, send_data_test_003, TestSize.Level1)
     EXPECT_EQ(ERR_DH_SCREEN_TRANS_ERROR, dataChannelImpl_->SendData(data));
 }
 
+/**
+ * @tc.name: send_data_test_004
+ * @tc.desc: Verify the SendData function.
+ * @tc.type: FUNC
+ * @tc.require: Issue Number
+ */
+HWTEST_F(ScreenDataChannelImplTest, send_data_test_004, TestSize.Level1)
+{
+    std::shared_ptr<IScreenChannelListener> listener = nullptr;
+    dataChannelImpl_->channelListener_ = listener;
+    dataChannelImpl_->jpegSessionFlag_ = true;
+    int32_t sessionId = 0;
+    dataChannelImpl_->OnSessionOpened(sessionId, 10);
+    dataChannelImpl_->OnSessionClosed(sessionId);
+    std::shared_ptr<DataBuffer> data = std::make_shared<DataBuffer>(10);
+    data->SetDataType(VIDEO_FULL_SCREEN_DATA);
+    EXPECT_EQ(ERR_DH_SCREEN_TRANS_ERROR, dataChannelImpl_->SendData(data));
+}
+
+/**
+ * @tc.name: send_data_test_004
+ * @tc.desc: Verify the SendData function.
+ * @tc.type: FUNC
+ * @tc.require: Issue Number
+ */
+HWTEST_F(ScreenDataChannelImplTest, send_data_test_005, TestSize.Level1)
+{
+    std::shared_ptr<IScreenChannelListener> listener = nullptr;
+    dataChannelImpl_->channelListener_ = listener;
+    dataChannelImpl_->jpegSessionFlag_ = true;
+    dataChannelImpl_->sessionId_ = 1;
+    int32_t sessionId = 0;
+    dataChannelImpl_->OnSessionOpened(sessionId, 10);
+    dataChannelImpl_->OnSessionClosed(sessionId);
+    std::shared_ptr<DataBuffer> data = std::make_shared<DataBuffer>(10);
+    data->SetDataType(VIDEO_FULL_SCREEN_DATA);
+    EXPECT_EQ(ERR_DH_SCREEN_TRANS_ERROR, dataChannelImpl_->SendData(data));
+}
+
+
+
 /**
  * @tc.name: on_session_opened_test_001
  * @tc.desc: Verify the OnSessionOpened function.
-- 
Gitee