1 /*
2 * Copyright (C) 2025 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "surface_decoder_adapter_unittest.h"
17
18 namespace OHOS {
19 namespace Media {
20 using namespace std;
21 using namespace testing;
22 using namespace testing::ext;
23
24 static const int32_t NUM_0 = 0;
25 static const int32_t NUM_1 = 1;
26
SetUpTestCase(void)27 void SurfaceDecoderAdapterUnitTest::SetUpTestCase(void) {}
28
TearDownTestCase(void)29 void SurfaceDecoderAdapterUnitTest::TearDownTestCase(void) {}
30
SetUp(void)31 void SurfaceDecoderAdapterUnitTest::SetUp(void)
32 {
33 adapter_ = std::make_shared<SurfaceDecoderAdapter>();
34 }
35
TearDown(void)36 void SurfaceDecoderAdapterUnitTest::TearDown(void)
37 {
38 adapter_ = nullptr;
39 }
40
41 /**
42 * @tc.name : Test SurfaceDecoderAdapterCallback OnError
43 * @tc.number: OnError_001
44 * @tc.desc : Test surfaceDecoderAdapter_.lock() == nullptr
45 * Test surfaceDecoderAdapter_.lock() != nullptr
46 * Test transCoderErrorCbOnce_ == false
47 */
48 HWTEST_F(SurfaceDecoderAdapterUnitTest, OnError_001, TestSize.Level0)
49 {
50 // Test surfaceDecoderAdapter_.lock() == nullptr
51 adapter_ = nullptr;
52 auto callback = std::make_shared<SurfaceDecoderAdapterCallback>(adapter_);
53 callback->OnError(MediaAVCodec::AVCodecErrorType::AVCODEC_ERROR_INTERNAL, NUM_1);
54
55 // Test surfaceDecoderAdapter_.lock() != nullptr
56 // Test transCoderErrorCbOnce_ == false
57 adapter_ = std::make_shared<SurfaceDecoderAdapter>();
58 ASSERT_NE(adapter_, nullptr);
59 auto decoderAdapterCallback = std::make_shared<MockDecoderAdapterCallback>();
60 adapter_->decoderAdapterCallback_ = decoderAdapterCallback;
61 callback = std::make_shared<SurfaceDecoderAdapterCallback>(adapter_);
62 EXPECT_CALL(*decoderAdapterCallback, OnError(_, _)).WillOnce(Return());
63 callback->OnError(MediaAVCodec::AVCodecErrorType::AVCODEC_ERROR_INTERNAL, NUM_1);
64 EXPECT_TRUE(callback->transCoderErrorCbOnce_);
65 }
66
67 /**
68 * @tc.name : Test SurfaceDecoderAdapterCallback OnOutputBufferAvailable
69 * @tc.number: SurfaceDecoderAdapterCallback_OnOutputBufferAvailable_001
70 * @tc.desc : Test surfaceDecoderAdapter_.lock() == nullptr
71 * Test surfaceDecoderAdapter_.lock() != nullptr
72 * Test (buffer->flag_ & BUFFER_IS_EOS) == 1
73 */
74 HWTEST_F(SurfaceDecoderAdapterUnitTest, SurfaceDecoderAdapterCallback_OnOutputBufferAvailable_001, TestSize.Level0)
75 {
76 // Test surfaceDecoderAdapter_.lock() == nullptr
77 adapter_ = nullptr;
78 auto buffer = std::make_shared<AVBuffer>();
79 auto callback = std::make_shared<SurfaceDecoderAdapterCallback>(adapter_);
80 callback->OnOutputBufferAvailable(NUM_1, buffer);
81
82 // Test surfaceDecoderAdapter_.lock() != nullptr
83 // Test (buffer->flag_ & BUFFER_IS_EOS) == 1
84 adapter_ = std::make_shared<SurfaceDecoderAdapter>();
85 ASSERT_NE(adapter_, nullptr);
86 auto decoderAdapterCallback = std::make_shared<MockDecoderAdapterCallback>();
87 adapter_->decoderAdapterCallback_ = decoderAdapterCallback;
88 callback = std::make_shared<SurfaceDecoderAdapterCallback>(adapter_);
89 buffer->flag_ = 1;
90 callback->OnOutputBufferAvailable(NUM_1, buffer);
91 auto surfaceDecoderAdapter = callback->surfaceDecoderAdapter_.lock();
92 ASSERT_NE(surfaceDecoderAdapter, nullptr);
93 uint32_t renderIndex = surfaceDecoderAdapter->indexs_[0];
94 EXPECT_EQ(renderIndex, NUM_1);
95 }
96
97 /**
98 * @tc.name : Test AVBufferAvailableListener OnBufferAvailable
99 * @tc.number: OnBufferAvailable_001
100 * @tc.desc : Test surfaceDecoderAdapter_.lock() == nullptr
101 */
102 HWTEST_F(SurfaceDecoderAdapterUnitTest, OnBufferAvailable_001, TestSize.Level0)
103 {
104 adapter_ = nullptr;
105 auto listener = std::make_shared<AVBufferAvailableListener>(adapter_);
106 listener->OnBufferAvailable();
107 EXPECT_EQ(listener->surfaceDecoderAdapter_.lock(), nullptr);
108 }
109
110 /**
111 * @tc.name : Test GetInputBufferQueue
112 * @tc.number: GetInputBufferQueue_001
113 * @tc.desc : Test inputBufferQueue_ != nullptr && inputBufferQueue_-> GetQueueSize() > 0
114 */
115 HWTEST_F(SurfaceDecoderAdapterUnitTest, GetInputBufferQueue_001, TestSize.Level0)
116 {
117 ASSERT_NE(adapter_, nullptr);
118 auto inputBufferQueue = std::make_shared<MockAVBufferQueue>();
119 adapter_->inputBufferQueue_ = inputBufferQueue;
120 EXPECT_CALL(*inputBufferQueue, GetQueueSize()).WillOnce(Return(NUM_1));
121 sptr<AVBufferQueueProducer> avBufferQueueProducer = new MockAVBufferQueueProducer();
122 adapter_->inputBufferQueueProducer_ = avBufferQueueProducer;
123 auto ret = adapter_->GetInputBufferQueue();
124 EXPECT_EQ(ret, avBufferQueueProducer);
125 }
126
127 /**
128 * @tc.name : Test SetDecoderAdapterCallback
129 * @tc.number: SetDecoderAdapterCallback_001
130 * @tc.desc : Test ret != 0
131 */
132 HWTEST_F(SurfaceDecoderAdapterUnitTest, SetDecoderAdapterCallback_001, TestSize.Level0)
133 {
134 ASSERT_NE(adapter_, nullptr);
135 auto decoderAdapterCallback = std::make_shared<MockDecoderAdapterCallback>();
136 auto codecServer = std::make_shared<MockAVCodecVideoDecoder>();
137 adapter_->codecServer_ = codecServer;
138 EXPECT_CALL(*codecServer, SetCallback(::testing::A<const std::shared_ptr<MediaCodecCallback> &>()))
139 .WillOnce(Return(NUM_1));
140 EXPECT_CALL(*codecServer, Release()).WillOnce(Return(NUM_1));
141 auto ret = adapter_->SetDecoderAdapterCallback(decoderAdapterCallback);
142 EXPECT_EQ(ret, Status::ERROR_UNKNOWN);
143 }
144
145 /**
146 * @tc.name : Test OnOutputBufferAvailable
147 * @tc.number: OnOutputBufferAvailable_001
148 * @tc.desc : Test (buffer->flag_ & BUFFER_IS_EOS) == 1 && buffer->pts_ <= lastBufferPts_.load()
149 */
150 HWTEST_F(SurfaceDecoderAdapterUnitTest, OnOutputBufferAvailable_001, TestSize.Level0)
151 {
152 ASSERT_NE(adapter_, nullptr);
153 auto buffer = std::make_shared<AVBuffer>();
154 buffer->flag_ = NUM_0;
155 buffer->pts_ = NUM_1;
156 adapter_->lastBufferPts_ = NUM_1;
157 adapter_->OnOutputBufferAvailable(NUM_1, buffer);
158 EXPECT_NE(adapter_->dropIndexs_.size(), 0);
159 }
160 } // namespace Media
161 } // namespace OHOS