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 6a9754ce51392aa363933016059cb6451b2f9394..afa7cf0b4028c578092a11d650dd8f1147bba864 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 @@ -37,6 +37,7 @@ void ScreenDataChannelImplTest::TearDown(void) dataChannelImpl_ = nullptr; } + /** * @tc.name: CreateSession_001 * @tc.desc: Verify the CreateSession function. @@ -47,7 +48,6 @@ HWTEST_F(ScreenDataChannelImplTest, CreateSession_001, TestSize.Level1) { std::shared_ptr listener = nullptr; int32_t ret = dataChannelImpl_->CreateSession(listener); - EXPECT_EQ(ERR_DH_SCREEN_TRANS_NULL_VALUE, ret); } @@ -62,10 +62,22 @@ HWTEST_F(ScreenDataChannelImplTest, CreateSession_002, TestSize.Level1) std::shared_ptr listener = std::make_shared(); dataChannelImpl_->jpegSessionFlag_ = false; int32_t ret = dataChannelImpl_->CreateSession(listener); - 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 listener = std::make_shared(); + dataChannelImpl_->SetJpegSessionFlag(true); + EXPECT_EQ(ERR_DH_SCREEN_TRANS_ILLEGAL_OPERATION, dataChannelImpl_->ReleaseSession()); +} + /** * @tc.name: OpenSession_001 * @tc.desc: Verify the OpenSession function. @@ -75,7 +87,19 @@ HWTEST_F(ScreenDataChannelImplTest, CreateSession_002, TestSize.Level1) HWTEST_F(ScreenDataChannelImplTest, OpenSession_001, TestSize.Level1) { int32_t ret = dataChannelImpl_->OpenSession(); + 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); } @@ -89,7 +113,6 @@ HWTEST_F(ScreenDataChannelImplTest, SendFullData_001, TestSize.Level1) { std::shared_ptr screenData = nullptr; int32_t ret = dataChannelImpl_->SendFullData(screenData); - EXPECT_EQ(ERR_DH_SCREEN_TRANS_NULL_VALUE, ret); } @@ -104,7 +127,6 @@ HWTEST_F(ScreenDataChannelImplTest, SendFullData_002, TestSize.Level1) std::shared_ptr data = std::make_shared(10); dataChannelImpl_->sessionId_ = 1; int32_t ret = dataChannelImpl_->SendFullData(data); - EXPECT_EQ(ERR_DH_SCREEN_TRANS_ERROR, ret); } @@ -118,7 +140,6 @@ HWTEST_F(ScreenDataChannelImplTest, SendDirtyData_001, TestSize.Level1) { std::shared_ptr screenData = nullptr; int32_t ret = dataChannelImpl_->SendDirtyData(screenData); - EXPECT_EQ(ERR_DH_SCREEN_TRANS_NULL_VALUE, ret); } @@ -133,7 +154,6 @@ HWTEST_F(ScreenDataChannelImplTest, SendDirtyData_002, TestSize.Level1) std::shared_ptr data = std::make_shared(10); dataChannelImpl_->jpegSessionId_ = 1; int32_t ret = dataChannelImpl_->SendFullData(data); - EXPECT_EQ(ERR_DH_SCREEN_TRANS_ERROR, ret); } @@ -184,6 +204,32 @@ 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(); + + 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(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 +283,45 @@ 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 listener = nullptr; + dataChannelImpl_->channelListener_ = listener; + dataChannelImpl_->jpegSessionFlag_ = true; + int32_t sessionId = 0; + dataChannelImpl_->OnSessionOpened(sessionId, 10); + dataChannelImpl_->OnSessionClosed(sessionId); + std::shared_ptr data = std::make_shared(10); + data->SetDataType(VIDEO_FULL_SCREEN_DATA); + EXPECT_EQ(ERR_DH_SCREEN_TRANS_ERROR, dataChannelImpl_->SendData(data)); +} + +/** + * @tc.name: send_data_test_005 + * @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 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 data = std::make_shared(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.